编写 Python 脚本以使用控制台命令执行

Question

研究在底部，在-1'ing之前阅读...谢谢。

我必须编写一个运行 SQL 查询的 Python 脚本。我做了一个主类并调用 SQLQuery。每个 SQLQuery 实例代表一个查询。脚本的结构必须如下：

class SQLQuery(object):

     def __init___(self, string_myQuery)...

instance1 = SQLQuery(SQLQuery1)...

instance2 = SQLQuery(SQLQuery2)...

作为用户要求，实例必须与类在同一个文件中（所以我不能只将每个实例设为主文件并单独执行该文件），并且每个实例 必须使用 Linux 控制台命令执行。我可以使用简单的 python SQLQuery.py 执行整个脚本，但我需要分别执行每个实例。查询将每天自动执行，因此我不需要终端 UI 树。它应该使用类似这样的命令来执行：

python SQLQuery.py -inst1

将执行instance1。

python SQLQuery.py -inst2

将执行instance2。

我研究了如何使用 Linux 命令执行 Python 脚本，大部分文章都是关于从 Python 脚本调用命令的。但是，我从 Python 文档中找到了this article。它建议添加 -m，所以：

python SQLQuery.py -m inst1

这可以让我使用控制台命令设置我的 main，但它不起作用，因为实例不是模块。而且由于实例必须与类在同一个文件中，所以当我使用控制台命令执行 SQLQuery.py 时，我不能将它们作为模块导入。

Answer 1

忽略所有无关紧要的事情，听起来你的问题是你有一堆名为instance1、instance2、instance3 等的全局对象，并且你想基于其中一个调用一些方法在命令行参数上，其值将与实例名称相似但不完全相同。

这可能不是一个好主意……但也没那么难：

if __name__ == '__main__':
    inst = sys.argv[1] # will be 'inst1', 'inst13', etc.
    inst_number = inst[5:] # so '1', '13', etc.
    inst_name = 'instance' + inst_number
    instance = globals()[inst_name]
    instance.execute()

---

做同样事情的更好方法是将 instance 全局变量放入可以索引的列表或字典中。

例如，假设你有一个 instances 字典，而不是 instance1、instance2 等，还有 instances['1']、instances[2] 等。现在代替这个：

inst_name = 'instance' + inst_number
instance = globals()[inst_name]
instance.execute()

……你只需这样做：

instances[inst_number].execute()

---

另外，与其想出一个命令行参数，其中包含额外的东西，你必须解析和丢弃，而且对人类读者来说没有比对你的代码更有意义的东西，为什么不只取一个数字?

python myscript.py 12

或者，或者，使用argparse 创建一个可以以所有明显方式使用的参数：

python myscript.py --instance=12
python myscript.py --instance 12
python myscript.py -i12
python myscript.py -i 12

无论哪种方式，您的代码都会获取字符串'12'，然后可以使用该字符串来查找函数，如上所述。

Answer 2

-m 选项的语法有误。假设您有以下名为foo.py 的文件：

import sys
print 'First arg is: ', sys.argv[1]

那么你可以这样称呼它：

$ python -m foo bar
First arg is:  bar

请注意，“.py”扩展名被省略。然后，您可以使用命令行参数来决定使用哪个对象或使用argparse 或optparse 模块来处理参数。