导航:起始页 > Dive Into Python > 脚本和流 > 处理命令行参数 | << >> | ||||
深入 Python :Dive Into Python 中文版Python 从新手到专家 [Dip_5.4b_CPyUG_Release] |
Python 完全支持创建在命令行运行的程序,也支持通过命令行参数和短长样式来指定各种选项。这些并非是 XML 特定的,但是这样的脚本可以充分使用命令行处理,看来是时候提一下它了。
如果不理解命令行参数如何暴露给你的 Python 程序,讨论命令行处理是很困难的,所以让我们先写个简单点的程序来看一下。
如果您还没有下载本书附带的样例程序, 可以 下载本程序和其他样例程序。
#argecho.py import sys for arg in sys.argv: print arg
[you@localhost py]$ python argecho.py argecho.py [you@localhost py]$ python argecho.py abc def argecho.py abc def [you@localhost py]$ python argecho.py --help argecho.py --help [you@localhost py]$ python argecho.py -m kant.xml argecho.py -m kant.xml
关于 sys.argv 需要了解的第一件事情就是:它包含了你正在调用的脚本的名称。你后面会实际使用这个知识,在第 16 章 函数编程中。现在不用担心。 | |
命令行参数通过空格进行分隔。在 sys.argv 列表中,每个参数都是一个独立的元素。 | |
命令行标志,像 --help,在 sys.argv 列表中还保存了它们自己的元素。 | |
为了让事情更有趣,有些命令行标志本身就接收参数。比如,这里有一个标记 (-m) 接收一个参数 (kant.xml)。标记自身和标记参数只是 sys.argv 列表中的一串元素。并没有试图将元素与其它元素进行关联;所有你得到的是一个列表。 |
所以正如你所看到的,你确实拥有了命令行传入的所有信息,但是接下来要实际使用它似乎不那么容易。对于只是接收单个参数或者没有标记的简单程序,你可以简单地使用 sys.argv[1] 来访问参数。这没有什么羞耻的;我一直都是这样做的。对更复杂的程序,你需要 getopt 模块。
def main(argv): grammar = "kant.xml" try: opts, args = getopt.getopt(argv, "hg:d", ["help", "grammar="]) except getopt.GetoptError: usage() sys.exit(2) ... if __name__ == "__main__": main(sys.argv[1:])
那么你传给 getopt 函数的参数是什么呢?好的,第一个只不过是一个命令行标志和参数的原始列表 (不包括第一个元素――脚本名称,你在调用 main 函数之前就已经将它砍掉了)。第二个是脚本接收的短命令行标志的一个列表。
第一个标志和第三个标志是简单的独立标志;你选择是否指定它们,它们做某些事情 (打印帮助) 或者改变状态 (打开调试)。但是,第二个标志 (-g) 必须 跟随一个参数――进行读取的语法文件的名称。实际上,它可以是一个文件名或者一个 web 地址,这时还不知道 (后面会确定),但是你要知道必须要有些东西。所以,你可以通过在 getopt 函数的第二个参数的 g 后面放一个冒号,来向 getopt 说明这一点。
更复杂的是,这个脚本既接收短标志 (像 -h),也接受长标志 (像 --help),并且你要它们做相同的事。这就是 getopt 第三个参数存在的原因:它是指定长标志的一个列表,其中的长标志是和第二个参数中指定的短标志相对应的。
这里有三点要注意:
被搞昏没?让我们看一下真实的代码,看看它在上下文中是否起作用。
def main(argv): grammar = "kant.xml" try: opts, args = getopt.getopt(argv, "hg:d", ["help", "grammar="]) except getopt.GetoptError: usage() sys.exit(2) for opt, arg in opts: if opt in ("-h", "--help"): usage() sys.exit() elif opt == '-d': global _debug _debug = 1 elif opt in ("-g", "--grammar"): grammar = arg source = "".join(args) k = KantGenerator(grammar, source) print k.output()
<< 根据节点类型创建不同的处理器 |
| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | |
全部放在一起 >> |