在Python脚本的开头设置输入变量

时间:2014-08-27 16:40:30

标签: python

我正在使用Red Hat中的python脚本连接到SQL数据库并在特定行上运行查询。我试图做的是将我想要查询的行设置为一个输入变量,当我启动它时它将被传递到脚本中。例如,当我启动脚本时,我想运行命令:

./SQLQuery.py Row_To_Query = Row_Name

对于我在VBScript中的类似脚本,我会像这样运行wscript命令:

wscript SQLQuery.vbs /name:Row_Name

然后行名称将被传递到脚本中,并且用作脚本的相关数据将继续进行。

我的最终目标是每10分钟左右将python脚本作为一个任务运行并从特定行中提取相关数据,但也可以指定运行SQL查询的哪一行而无需编辑脚本每次我想运行它。

1 个答案:

答案 0 :(得分:1)

您可以使用argparse来解析以类似unix的标准格式提供的命令行参数:

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--row-to-query', type=str, help='row to query')
namespace = parser.parse_args()

row_name = namespace.row_to_query
...

然后你可以运行这样的程序:

./SQLQuery.py --row-to-query=some_row_name

./SQLQuery.py --row-to-query some_row_name

这两个调用应该是等效的。