10.6. 处理命令行参数

Python 完全支持创建在命令行运行的程序，也支持通过命令行参数和短长样式来指定各种选项。这些并非是 XML 特定的，但是这样的脚本可以充分使用命令行处理，看来是时候提一下它了。

如果不理解命令行参数如何暴露给你的 Python 程序，讨论命令行处理是很困难的，所以让我们先写个简单点的程序来看一下。

例 10.20. sys.argv 介绍

如果您还没有下载本书附带的样例程序, 可以 下载本程序和其他样例程序。

#argecho.py
import sys
for arg in sys.argv: 
    print arg

例 10.21. sys.argv 的内容

[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[1] 来访问参数。这没有什么羞耻的；我一直都是这样做的。对更复杂的程序，你需要 getopt 模块。

例 10.22. 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 函数之前就已经将它砍掉了)。第二个是脚本接收的短命令行标志的一个列表。

"hg:d"

-h

打印用法概要

-g ...

使用给定的语法文件或 URL

-d

在解析时显示调试信息

第一个标志和第三个标志是简单的独立标志；你选择是否指定它们，它们做某些事情 (打印帮助) 或者改变状态 (打开调试)。但是，第二个标志 (-g) 必须 跟随一个参数――进行读取的语法文件的名称。实际上，它可以是一个文件名或者一个 web 地址，这时还不知道 (后面会确定)，但是你要知道必须要有些东西。所以，你可以通过在 getopt 函数的第二个参数的 g 后面放一个冒号，来向 getopt 说明这一点。

更复杂的是，这个脚本既接收短标志 (像 -h)，也接受长标志 (像 --help)，并且你要它们做相同的事。这就是 getopt 第三个参数存在的原因：它是指定长标志的一个列表，其中的长标志是和第二个参数中指定的短标志相对应的。

["help", "grammar="]

--help

打印用法概要

--grammar ...

使用给定的语法文件或 URL

这里有三点要注意：

1. 所有命令行中的长标志以两个短划线开始，但是在调用 getopt 时，你不用包含这两个短划线。它们是能够被理解的。
2. --grammar 标志的后面必须跟着另一个参数，就像 -g 标志一样。通过等于号标识出来："grammar="。
3. 长标志列表比短标志列表更短一些，因为 -d 标志没有相应的长标志。这很好；只有 -d 才会打开调试。但是短标志和长标志的顺序必须是相同的，你应该先指定有长标志的短标志，然后才是剩下的短标志。

被搞昏没？让我们看一下真实的代码，看看它在上下文中是否起作用。

例 10.23. 在 kgp.py 中处理命令行参数

 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()