Chinaunix首页 | 论坛 | 博客
  • 博客访问: 418584
  • 博文数量: 72
  • 博客积分: 2541
  • 博客等级: 少校
  • 技术积分: 756
  • 用 户 组: 普通用户
  • 注册时间: 2006-09-20 16:08
文章分类

全部博文(72)

文章存档

2013年(13)

2009年(1)

2008年(7)

2007年(45)

2006年(6)

我的朋友

分类: Python/Ruby

2013-06-27 18:37:58

python中使用getopt处理命令行参数,本文主要对getopt进行介绍。getopt的调用主要分三步:
1、导入sys和getopt模块;
2、分析命令行参数;
3、处理结果;

第一步很简单,只需要:

点击(此处)折叠或打开

  1. import sys
  2. import getopt

第二步处理方法如下:

点击(此处)折叠或打开

  1. try:
  2.     opts, args = getopt.getopt(sys.argv[1:], "ho:", ["help", "output="])
  3. except getopt.GetoptError:
  4.     # 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']

第三步主要是对分析出的参数进行判断,并进行下一步的处理。主要处理模式如下:

点击(此处)折叠或打开

  1. for o, a in opts:
  2.     if o in ("-h", "--help"):
  3.         usage()
  4.         sys.exit()
  5.     if o in ("-o", "--output"):
  6.         output = a

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

实践篇1:

点击(此处)折叠或打开

  1. #!/usr/bin/env python

  2. import sys;
  3. import getopt;

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


  6. if "__main__" == __name__:
  7.     #lsArgs = [""];
  8.     
  9.     try:
  10.         opts,args = getopt.getopt(sys.argv[1:], "ao:c", ["help", "output="]);
  11.     
  12.         print("============ opts ==================");
  13.         print(opts);
  14.     
  15.         print("============ args ==================");
  16.         print(args);
  17.         
  18.         #check all param
  19.         for opt,arg in opts:
  20.             if opt in ("-h", "--help"):
  21.                 usage();
  22.                 sys.exit(1);
  23.             elif opt in ("-t", "--test"):
  24.                 print("for test option");
  25.             else:
  26.                 print("%s ==> %s" %(opt, arg));
  27.         
  28.     except getopt.GetoptError:
  29.         print("getopt error!");
  30.         usage();
  31.         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

实践篇2:

点击(此处)折叠或打开

  1. import sys, getopt

  2. def usage():
  3.     print \
  4.     '''
  5. Usage:convert.py [<-i input_file>
  6.                   <-o output_file>
  7.                   [-h or --help]
  8.                   ]

  9. Options:
  10. -i: input_file
  11. -o: output_file
  12. -h or --help: help

  13. Example:convert.py -i test1 -o test2
  14. '''


  15. shortargs = 'hi:o:'
  16. longargs = 'help'
  17. opts, args = getopt.getopt(sys.argv[1:], shortargs,[longargs])
  18. print opts
  19. print args

  20. input_file = ''
  21. output_file = ''

  22. for op , value in opts:
  23.     if op == '-i':
  24.         input_file = value
  25.     elif op == '-o':
  26.         output_file = value
  27.     elif op in ('-h','--help'):
  28.         usage()
  29.         sys.exit()
本实践中,定义shortargs时,有多个参数,假如后面没有longargs,“o”后面的":"必须带上,表示其为opts,而不是args;

阅读(3371) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~