argparse模块

argparse模块
argparse是python用于解析命令行参数和选项的标准模块，用于代替已经过时的optparse模块。argparse模块的作用是用于解析命令行参数。

import argparse


def get_parameters():
    # 创建参数解析对象
    parser = argparse.ArgumentParser(description='命令行参数解析测试')
    # 前两个参数指定命令行选项，type指定参数类型，default指定默认值,required指定参数是否为必需，help帮助信息
    # 获取参数值时，参数名为第二个参数去掉前面的--后的name
    parser.add_argument('-n', '--name', type=str, default=None, required=False, help='名称')
    # 也可以没有第二个参数，此时，参数名为p
    parser.add_argument('-p', type=int, default=80, required=False, help='端口')

    # 参数为一个数组，则加nargs参数，argparse.ONE_OR_MORE选项表示传需要一个或多个值
    parser.add_argument('-pl', type=int, nargs=argparse.ONE_OR_MORE, default=80, required=False, help='端口')
    # 解析获取参数
    return parser.parse_args()


if __name__ == '__main__':
    # 调用解析方法，解析获取命令行参数
    args = get_parameters()
    print(args) # Namespace(name='12', p=90)
    print(args.name, args.p)    # 12 90
    print(args.pl)  # [1, 3, 4]

 

命令行输入：

python demo.py -n abc -p 90 -pl 1 3 4

控制台输出：

abc 90
[1, 3, 4]