getopt在Python中的使用(转)
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