通过argparse模块,可方便解析
sys.argv 中定义的参数,从而轻松编写用户友好的命令行界面。
参数解析
通过argparse进行参数解析,需要以下几个步骤:
- 导入模块:import argparse
- 获取解析器对象:argparse.ArgumentParser
- 添加参数解析规则:add_argument
- 获取解析结果对象:parse_args
- 从解析结果对象中获取参数值
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-o', '--out', action='store_true') // 对应参数有时为True,否则为False
parser.add_argument('--data', required=True, dest='path') // data为必传参数
parser.add_argument('--seed', type=int)
args = parser.parse_args()
print('out:', args.out, type(args.out))
print('path:', args.path, type(args.path))
print('seed:', args.seed, type(args.seed))
bool类型参数,action要设为store_true或store_false:
python main.py -o args.out为Truepython main.py args.out为False
解析器
通过ArgumentParser() 函数的调用可获取了解析器对象ArgumentParser ,解析器说明:
名字 | 默认值 | 功能 |
---|
prog | sys.argv[0] | -h 时显示的程序名 | usage | - | usage字段描述 | description | None | description字段描述 | formatter_class | None | 定制说明文本的显示风格 | prefix_class | - | 定制前缀字符,例如前缀**“-b"改为“+b”** | add_help | True | 是否使能显示参数 -h --help | allow_abbrev | True | 是否支持长参数 |
添加规则
通过解析器的add_argument可方便添加解析规则:
parser.add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])
关键字 | 简介 |
---|
name or flags | 参数名或者"-/–"开头的选项,例如foo 或者-f, --foo | action | 匹配到选项后的行为 | nargs | 选项跟随的参数个数 | const | 在某些action 和nargs 下,使用的固定值 | default | 默认值,默认为None; nargs为? 或者* 时,命令行没有对应参数时,使用此默认值; 若default=argparse.SUPPRESS 命令行没对应参数时,不会添加对象到解析结果中; | type | 参数类型;默认为字符串str | choices | 可选的参数值范围 | required | 是否为必选项 | help | 参数描述 | metavar | 使用说明中显示选项对应的值示例 | dest | 选项最终在解析结果对象中的属性名(必须是合法的变量标识),不设定,则使用选项本身: flags模式时,会去掉前缀- (且优先使用长参数); 若选项中间有- 符号,会替换为_ ; |
name or flags:
- 不以"-/–"开始的为位置参数,必须赋值;其顺序与add添加的顺序相同;
- 以"-/–"开始的为选项参数;
- 标识短参数;-- 标识长参数;- 可只给出一种,也可两种同时给出;
- 没有dest时,对应解析的属性名为name或去掉前缀的flags
action参数
参数action用于控制命令行选项的行为
值 | 含义 |
---|
store | 保存参数值(默认) | store_const | 与const配合使用;有对应选项时,赋值const值给选项;且此时不能有type参数 | store_true | 保存值为True;对应选项有时为True,否则为False | store_false | 保存值为False;对应选项有时为False,否则为True | append | 追加选项值到列表中;允许选项出现多次 | append_const | 与const配合使用,追加const的值到列表中 | count | 把选项出现的次数赋值给对应选项 | help | 效果等效于-h 和 --help | version | 打印版本 |
append参数:
parser = argparse.ArgumentParser()
parser.add_argument('-a', '-append', action='append', dest='files')
parser.add_argument('-A', '--add', action='append_const', const='Add-param', dest='dirs')
args = parser.parse_args()
print('files:', args.files)
print('dirs:', args.dirs)
// python .\main.py -a f1 -a f2 -A -A
files: ['f1', 'f2']
dirs: ['Add-param', 'Add-param']
store_const参数:
parser = argparse.ArgumentParser()
parser.add_argument('--seed', const=12, action='store_const')
args = parser.parse_args()
print('seed:', args.seed, type(args.seed))
// python .\main.py --seed
seed: 12 <class 'int'>
nargs参数
nargs参数是number argumemts的缩写,表示选项有多少个参数
值 | 含义 |
---|
N (an integer) | 收集N个参数到列表 | ‘?’ | 无参数或单个参数 | ‘*’ | 大于等于0个参数 | ‘+’ | 大于等于1个参数 | argparse.REMAINDER | 只收集不解析 |
没有指定时,其取值取决于参数action:
action = "store" 时,默认获取1个参数;action = "store_true"/"store_false" 时,选项不需要带参数。
choice参数
choices限定了参数值的可选列表,如果命令行提供的参数值不在列表中,则会报错;而且choice中的元素类型,需要与type对应。
parser = argparse.ArgumentParser()
parser.add_argument('--move', choices=['rock', 'paper', 'scissors'])
args = parser.parse_args()
print('move:', args.move)
// python .\main.py --move rock
move: rock
示例
常用参数选项示例:
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('-o', '--out', action='store_true')
parser.add_argument('--revout', action='store_false')
parser.add_argument('--data', required=True, dest='path')
parser.add_argument('--seed', const=12, action='store_const')
parser.add_argument('-v', type=str, help="add verb")
parser.add_argument('-a', '--append-files', action='append')
parser.add_argument('-A', '--add', action='append_const', const='Add-param', dest='file_const')
parser.add_argument('--move', choices=['rock', 'paper', 'scissors'])
args = parser.parse_args()
print('append-files:', args.append_files)
print('file-const:', args.file_const)
print('move:', args.move)
print('out:', args.out, type(args.out))
print('revout:', args.revout)
print("verb:", args.v, type(args.v))
print('seed:', args.seed, type(args.seed))
// python .\main.py -v sd --data pdata -a f1 -a f2 -AAA --seed
append-files: ['f1', 'f2']
file-const: ['Add-param', 'Add-param', 'Add-param']
move: None
out: False <class 'bool'>
revout: True
verb: sd <class 'str'>
seed: 12 <class 'int'>
|