python命令行参数

Python 提供了 getopt 模块来获取命令行参数。

$ python test.py arg1 arg2 arg3

Python 中也可以所用 sys 的 sys.argv 来获取命令行参数：

• sys.argv 是命令行参数列表。

• len(sys.argv) 是命令行参数个数。

注：sys.argv[0] 表示脚本名，sys.argv[1：] 表示后面跟的所有的参数

 

我们看到通过 sys.argv 我们可以获得运行 Python 程序中所有的命令行参数。

getopt模块

getopt模块是专门处理命令行参数的模块，用于获取命令行选项和参数，也就是sys.argv。命令行选项使得程序的参数更加灵活。支持短选项模式（-）和长选项模式（--）。

该模块提供了两个方法及一个异常处理来解析命令行参数。

getopt.getopt 方法

getopt.getopt 方法用于解析命令行参数列表，语法格式如下：

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

方法参数说明：

• args: 要解析的命令行参数列表。

• options: 以列表的格式定义，options后的冒号(:)表示该选项必须有附加的参数，不带冒号表示该选项不附加参数。

• long_options: 以字符串的格式定义，long_options 后的等号(=)表示如果设置该选项，必须有附加的参数，否则就不附加参数。

• 该方法返回值由两个元素组成: 第一个是 (option, value) 元组的列表。 第二个是参数列表，包含那些没有'-'或'--'的参数。

Exception getopt.GetoptError

在没有找到参数列表，或选项的需要的参数为空时会触发该异常。

异常的参数是一个字符串，表示错误的原因。属性 msg 和 opt 为相关选项的错误信息。

#!/usr/bin/env python
# encoding: utf-8
import sys, getopt

def main(argv):
    inputfile = ""
    outputfile = ""
    packagefile=""
    try:
        # 这里的 h 就表示该选项无参数，i:表示 i 选项后需要有参数
        opts, args = getopt.getopt(argv, "hi:o:p:",["infile=", "outfile=","package="])
    except getopt.GetoptError:   #当输入的有参数的时间，但不是指定的参数
        print 'xx.py -i  -o  -p '
        print 'or: xx.py --infile= --outfile= --package= '
        sys.exit(2)

    for opt, arg in opts:
        if opt == "-h":
            print u'-h     help帮助方法：'
            print u'-i     输入文件'
            print u'-o     输出文件：'
            print u'-p     package 包名'
            print 'eg:'
            print "*"*50
            print 'xx.py -i  -o  -p '
            print 'or: xx.py --infile= --outfile= --package= '
            print "*" * 50
            sys.exit()
        elif opt in ("-i", "--infile"):
            inputfile = arg
        elif opt in ("-p", "--package"):
            packagefile = arg
        elif opt in ("-o", "--outfile"):
            outputfile = arg
    print u'参数个数为:', len(sys.argv), u'个参数。'
    print "argv :{}".format(sys.argv)
    if len(sys.argv)>1:
        print 'Input file : ', inputfile
        print 'Package file : ', packagefile
        print 'Output file: ', outputfile
    else:
        print u"无参数哦！-h 查看所有参数列表!"

if __name__ == "__main__":
    main(sys.argv[1:])

执行 ：

参数：