getopt

 python 之 分割参数getopt

 

os下有个方法walk，非常的好用，用来生成一个generator。每次可以得到一个三元tupple，其中第一个为起始路径，第二个为起始路径下的文件夹，第三个是起始路径下的文件。 
 1. 导入getopt, sys 模块 
2. 分析命令行参数 
3. 处理结果 

第一步很简单，只需要： 
import getopt, sys

第二步处理方法如下（以Python 手册上的例子为例）： 

try:
    opts, args = getopt.getopt(sys.argv[1:], "ho:", ["help", "output="])
except getopt.GetoptError:
    # print help information and exit:

 

1. 处理所使用的函数叫getopt() ，因为是直接使用import 导入的getopt 模块，所以要加上限定getopt 才可以。 
2. 使用sys.argv[1:] 过滤掉第一个参数（它是执行脚本的名字，不应算作参数的一部分）。 
3. 使用短格式分析串"ho:" 。当一个选项只是表示开关状态时，即后面不带附加参数时，在分析串中写入选项字符。当选项后面是带一个附加参数时，在分析串中写入选项字符同时后面加一个":" 号 。所以"ho:" 就表示"h" 是一个开关选项；"o:" 则表示后面应该带一个参数。 
4. 使用长格式分析串列表：["help", "output="] 。长格式串也可以有开关状态，即后面不跟"=" 号。如果跟一个等号则表示后面还应有一个参数 。这个长格式表示"help" 是一个开关选项；"output=" 则表示后面应该带一个参数。 
5. 调用getopt 函数。函数返回两个列表：opts 和args 。opts 为分析出的格式信息。args 为不属于格式信息的剩余的命令行参数。opts 是一个两元组的列表。每个元素为：( 选项串, 附加参数) 。如果没有附加参数则为空串'' 。
6. 整个过程使用异常来包含，这样当分析出错时，就可以打印出使用信息来通知用户如何使用这个程序。 

如上面解释的一个命令行例子为： 
'-h -o file --help --output=out file1 file2'

在分析完成后，opts 应该是： 
[('-h', ''), ('-o', 'file'), ('--help', ''), ('--output', 'out')]

而args 则为： 
['file1', 'file2']
第三步主要是对分析出的参数进行判断是否存在，然后再进一步处理。主要的处理模式为： 

for o, a in opts:
    if o in ("-h", "--help"):
        usage()
        sys.exit()
    if o in ("-o", "--output"):
        output = a

 

使用一个循环，每次从opts 中取出一个两元组，赋给两个变量。o 保存选项参数，a 为附加参数。接着对取出的选项参数进行处理。（例子也采用手册的例子）

 

实践篇：

#!/usr/bin/env python

import sys;
import getopt;

def usage():
    print("Usage:%s [-a|-o|-c] [--help|--output] args...." %Dsys.argv[0]);


if "__main__" == __name__:
    #lsArgs = [""];

    try:
        opts,args = getopt.getopt(sys.argv[1:], "ao:c", ["help", "output="]);

        print("============ opts ==================");
        print(opts);

        print("============ args ==================");
        print(args);

        #check all param
        for opt,arg in opts:
            if opt in ("-h", "--help"):
                usage();
                sys.exit(1);
            elif opt in ("-t", "--test"):
                print("for test option");
            else:
                print("%s  ==> %s" %(opt, arg));

    except getopt.GetoptError:
        print("getopt error!");
        usage();
        sys.exit(1);

 

运行结果：

$ ./test_getopt.py  -a -oaaa -caa --output=out file1 t file2 -d
============ opts ==================
[('-a', ''), ('-o', 'aaa'), ('-c', ''), ('-a', ''), ('-a', ''), ('--output', 'out')]
============ args ==================
['file1', 't', 'file2', '-d']
-a  ==> 
-o  ==> aaa
-c  ==> 
-a  ==> 
-a  ==> 
--output  ==> out