Python命令行参数

测试积点老人

Python 获得命令行参数的方法

需要模块：sys
参数个数：len(sys.argv)
脚本名：    sys.argv[0]
参数1：     sys.argv[1]
参数2：     sys.argv[2]

解析命令行参数

Python提供了一个getopt模块，可用于解析命令行选项和参数。

该模块提供了两个功能和异常，以启用命令行参数解析。

getopt.getopt方法此方法解析命令行选项和参数列表。以下是此方法的简单语法

1. -getopt.getopt(args, options, [long_options])

复制代码

Pythongetopt.GetoptError异常

当在参数列表中有一个无法识别的选项，或者当需要一个参数的选项不为任何参数时，会引发这个异常。
异常的参数是一个字符串，指示错误的原因。 属性msg和opt给出错误消息和相关选项。

1. import sys, getopt
   
2. 
   
3. def main(argv):
   
4.     inputfile = ''
   
5.     outputfile = ''
   
6. 
   
7.     try:
   
8.         opts, args = getopt.getopt(argv, "hi:o:g:", ['help', 'ifile=', "ofile="])
   
9.         # 长选项'='相当于短选项的'：'
   
10.         print(opts)  # 打印参数与值 [('-i', 'in'), ('-o', 'out')]
    
11. 
    
12.     except getopt.GetoptError as e:
    
13.         print(e)
    
14.         sys.exit(-2)
    
15. 
    
16.     for opt, arg in opts:
    
17.         if opt in ('-h', '--help'):
    
18.             print('usage: command_line_usage.py -i <inputfile> -o <outputfile>')
    
19.             sys.exit()
    
20.         elif opt in ("-i", "--ifile"):
    
21.             inputfile = arg
    
22.         elif opt in ("-o", "--ofile"):
    
23.             outputfile = arg
    
24. 
    
25.     print('Input file is "', inputfile)
    
26.     print('Output file is "', outputfile)
    
27. 
    
28. if __name__ == "__main__":
    
29.     main(sys.argv[1:])  # 要处理的参数列表，sys.argv[0]为脚本名，所以用sys.argv[1:]过滤掉脚本名。
    

复制代码

代码解释：
a) sys.argv[1:]为要处理的参数列表，sys.argv[0]为脚本名，所以用sys.argv[1:]过滤掉脚本名。
b) "hi: o :": 当一个选项只是表示开关状态时，即后面不带附加参数时，在分析串中写入选项字符。当选项后面是带一个附加参数时，在分析串中写入选项字符同时后面加一个":"号。所以"hi : o:"就表示"h"是一个开关选项；"i :"和" o : "则表示后面应该带一个参数。
c) 调用getopt函数。函数返回两个列表：opts和args。opts为分析出的格式信息。args为不属于格式信息的剩余的命令行参数。opts是一个两元组的列表。每个元素为：(选项串,附加参数)。如果没有附加参数则为空串''。

getopt函数的第三个参数[, long_options]为可选的长选项参数，上面例子中的都为短选项(如-i -o)

长选项格式举例:

• --version
• --file=error.txt
  

让一个脚本同时支持短选项和长选项
getopt.getopt(sys.argv[1:], "hi: o:", ["version", "file="])

1. D:\TestFPython>python 1_1_getopt.py -i in -o out
   
2. [('-i', 'in'), ('-o', 'out')]
   
3. Input file is " in
   
4. Output file is " out

复制代码

Miss_love

支持分享