python内置库--argparse
Posted MyRecords
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python内置库--argparse相关的知识,希望对你有一定的参考价值。
1 关于argparse
从命令行工具运行python时,argparse 可以解析命令行工具输入的各种数据,通过argparse提供的函数或者属性,我们可以获得它解析到的数据
通过argparse,我们也可以自定义命令行选项,比如pytest -s -v ,-s -v就是pytest定义的命令行选项,通过argparse,我们也可以定义自己的命令行选项
下面是一个例子
命令行执行 python argparse_a.py a b
可以看到在命令行执行python文件时输入的参数 a b,通过argparse,我们得到了这2个参数
现在执行 python argparse_a.py -o ad a b
然后再,在我们执行命令的目录下面,多了一个ad文件
这些都是argparse解析命令行数据的功劳
2 argparse的使用
argparse的核心功能就3步
第一步 生成一个ArgumentParser这个类的实例对象parser
上面使用了prog等参数字段,具体意思后面会说到,这里也可以像之前的例子不加任何参数
第二步 在parser对象上添加参数名,这些参数就是我们命令行执行时要用到的
总的来说,添加参数名时,有3种类型的参数,positional arguments, options that accept values, and on/off flags
第三步 从parser对象上提取具体的参数数据,这些参数数据大部分是我们在命令行输入的数据
如上,通过parse_args()方法返回的对象加上参数名,我们就可以得到具体的参数数据
class argparse.ArgumentParser()
执行ArgumentParser()时,括号里接受下面这些参数来创建一个对象
-
prog - The name of the program (default: os.path.basename(sys.argv[0]))
-
usage - The string describing the program usage (default: generated from arguments added to parser)
-
description - Text to display before the argument help (by default, no text)
-
epilog - Text to display after the argument help (by default, no text)
-
parents - A list of ArgumentParser objects whose arguments should also be included
-
formatter_class - A class for customizing the help output
-
prefix_chars - The set of characters that prefix optional arguments (default: ‘-‘)
-
fromfile_prefix_chars - The set of characters that prefix files from which additional arguments should be read (default: None)
-
argument_default - The global default value for arguments (default: None)
-
conflict_handler - The strategy for resolving conflicting optionals (usually unnecessary)
-
add_help - Add a -h/--help option to the parser (default: True)
-
allow_abbrev - Allows long options to be abbreviated if the abbreviation is unambiguous. (default: True)
-
exit_on_error - Determines whether or not ArgumentParser exits with error info when an error occurs. (default: True)
prog
这个字段会影响到program name的值,默认情况下,它是通过sys.argv[0]来取值的,即在命令行python后面紧邻的那一段文本中最后一部分,比如py文件的文件名
默认情况下是像这样
现在加上pro参数
usage
默认情况下 显示自定义的参数字段
加上usage后
description
添加该参数
如上,可看出默认情况下并没有相关数据的显示
还有很多其他参数,可具体查看官网
add_argument()
该方法也可以在括号中添加多个参数,如下
-
name or flags - Either a name or a list of option strings, e.g. foo or -f, --foo.
-
action - The basic type of action to be taken when this argument is encountered at the command line.
-
nargs - The number of command-line arguments that should be consumed.
-
const - A constant value required by some action and nargs selections.
-
default - The value produced if the argument is absent from the command line and if it is absent from the namespace object.
-
type - The type to which the command-line argument should be converted.
-
choices - A sequence of the allowable values for the argument.
-
required - Whether or not the command-line option may be omitted (optionals only).
-
help - A brief description of what the argument does.
-
metavar - A name for the argument in usage messages.
-
dest - The name of the attribute to be added to the object returned by parse_args()
action
action参数值表示,我们在命令行中的参数数据应该被怎样处理,该参数默认值是store
action=\'store. 这是默认的处理方式,可以不用故意写出来,默认就是这样
action=\'store_const\'
如上,未输入p2参数,p2的值被设为const的值,sore_const就是该参数的值为const的值
注意,这种情况下我们不能在命令行指定p2的值
\'store_true\' and \'store_false\'
和action=\'store_const\'类似
append
\'count\'
没有看懂到底有啥作用,好像就是统计参数字段出现的次数
未完待续
Python命令行解析库argparse(转)
原文:http://www.cnblogs.com/linxiyue/p/3908623.html
2.7之后python不再对optparse模块进行扩展,python标准库推荐使用argparse模块对命令行进行解析。
1.example
有一道面试题:编写一个脚本main.py,使用方式如下:
main.py -u http://www.sohu.com -d ‘a=1,b=2,c=3‘ -o /tmp/index.html
功能要求:打开-u指定的页面,将页面中所有的链接后面增加参数a=1&b=2&c=3(需要考虑链接中已经存在指定的参数的问题), 然后保存到-o指定的文件中。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
import argparse import urllib from pyquery import PyQuery as pq def getArgs(): parse = argparse.ArgumentParser() parse.add_argument( ‘-u‘ , type = str ) parse.add_argument( ‘-d‘ , type = str ) parse.add_argument( ‘-o‘ , type = str ) args = parse.parse_args() return vars (args) def urlAddQuery(url,query): query = query.replace( ‘,‘ , ‘&‘ ) if ‘?‘ in url: return url.replace( ‘?‘ , ‘?‘ + query + ‘&‘ ) else : return url + ‘?‘ + query def getHref(): args = getArgs() url = args[ ‘u‘ ] query = args[ ‘d‘ ].strip( "\‘" ) fileName = args[ ‘o‘ ] doc = pq(url = url) with open (fileName, ‘w‘ ) as f: for a in doc( ‘a‘ ): a = pq(a) href = a.attr( ‘href‘ ) if href: newurl = urlAddQuery(href,query) f.write(newurl + ‘\n‘ ) if __name__ = = ‘__main__‘ : getHref() |
2.创建解析器
1
2
|
import argparse parser = argparse.ArgumentParser() |
class ArgumentParser(prog=None, usage=None, description=None, epilog=None, parents=[], formatter_class=argparse.HelpFormatter, prefix_chars=‘-‘, fromfile_prefix_chars=None, argument_default=None, conflict_handler=‘error‘, add_help=True)
创建一个ArgumentParser实例对象,ArgumentParser对象的参数都为关键字参数。
prog:程序的名字,默认为sys.argv[0],用来在help信息中描述程序的名称。
1
2
3
4
5
6
|
>>> parser = argparse.ArgumentParser(prog= ‘myprogram‘ ) >>> parser.print_help() usage: myprogram [-h] optional arguments: -h, --help show this help message and exit |
usage:描述程序用途的字符串
1
2
3
4
5
6
7
8
9
10
11
12
|
>>> parser = argparse.ArgumentParser(prog= ‘PROG‘ , usage= ‘%(prog)s [options]‘ ) >>> parser.add_argument( ‘--foo‘ , nargs= ‘?‘ , help= ‘foo help‘ ) >>> parser.add_argument( ‘bar‘ , nargs= ‘+‘ , help= ‘bar help‘ ) >>> parser.print_help() usage: PROG [options] positional arguments: bar bar help optional arguments: -h, --help show this help message and exit --foo [FOO] foo help |
description:help信息前的文字。
epilog:help信息之后的信息
1
2
3
4
5
6
7
8
9
10
11
12
|
>>> parser = argparse.ArgumentParser( ... description= ‘A foo that bars‘ , ... epilog= "And that‘s how you‘d foo a bar" ) >>> parser.print_help() usage: argparse.py [-h] A foo that bars optional arguments: -h, --help show this help message and exit And that ‘s how you‘ d foo a bar |
parents:由ArgumentParser对象组成的列表,它们的arguments选项会被包含到新ArgumentParser对象中。
1
2
3
4
5
6
7
|
>>> parent_parser = argparse.ArgumentParser(add_help=False) >>> parent_parser.add_argument( ‘--parent‘ , type= int ) >>> foo_parser = argparse.ArgumentParser(parents=[parent_parser]) >>> foo_parser.add_argument( ‘foo‘ ) >>> foo_parser.parse_args([ ‘--parent‘ , ‘2‘ , ‘XXX‘ ]) Namespace(foo= ‘XXX‘ , parent=2) |
formatter_class:help信息输出的格式,
prefix_chars:参数前缀,默认为‘-‘
1
2
3
4
5
|
>>> parser = argparse.ArgumentParser(prog= ‘PROG‘ , prefix_chars= ‘-+‘ ) >>> parser.add_argument( ‘+f‘ ) >>> parser.add_argument( ‘++bar‘ ) >>> parser.parse_args( ‘+f X ++bar Y‘ .split()) Namespace(bar= ‘Y‘ , f= ‘X‘ ) |
fromfile_prefix_chars:前缀字符,放在文件名之前
1
2
3
4
5
6
|
>>> with open( ‘args.txt‘ , ‘w‘ ) as fp: ... fp.write( ‘-f\nbar‘ ) >>> parser = argparse.ArgumentParser(fromfile_prefix_chars= ‘@‘ ) >>> parser.add_argument( ‘-f‘ ) >>> parser.parse_args([ ‘-f‘ , ‘foo‘ , ‘@args.txt‘ ]) Namespace(f= ‘bar‘ ) |
当参数过多时,可以将参数放到文件中读取,例子中parser.parse_args([‘-f‘, ‘foo‘, ‘@args.txt‘])解析时会从文件args.txt读取,相当于[‘-f‘, ‘foo‘, ‘-f‘, ‘bar‘]。
argument_default:参数的全局默认值。例如,要禁止parse_args时的参数默认添加,我们可以:
1
2
3
4
5
6
7
|
>>> parser = argparse.ArgumentParser(argument_default=argparse.SUPPRESS) >>> parser.add_argument( ‘--foo‘ ) >>> parser.add_argument( ‘bar‘ , nargs= ‘?‘ ) >>> parser.parse_args([ ‘--foo‘ , ‘1‘ , ‘BAR‘ ]) Namespace(bar= ‘BAR‘ , foo= ‘1‘ ) >>> parser.parse_args() Namespace() |
当parser.parse_args()时不会自动解析foo和bar了。
conflict_handler:解决冲突的策略,默认情况下冲突会发生错误:
1
2
3
4
5
6
|
>>> parser = argparse.ArgumentParser(prog= ‘PROG‘ ) >>> parser.add_argument( ‘-f‘ , ‘--foo‘ , help= ‘old foo help‘ ) >>> parser.add_argument( ‘--foo‘ , help= ‘new foo help‘ ) Traceback (most recent call last): .. ArgumentError: argument --foo: conflicting option string (s): --foo |
我们可以设定冲突解决策略:
1
2
3
4
5
6
7
8
9
10
|
>>> parser = argparse.ArgumentParser(prog= ‘PROG‘ , conflict_handler= ‘resolve‘ ) >>> parser.add_argument( ‘-f‘ , ‘--foo‘ , help= ‘old foo help‘ ) >>> parser.add_argument( ‘--foo‘ , help= ‘new foo help‘ ) >>> parser.print_help() usage: PROG [-h] [-f FOO] [--foo FOO] optional arguments: -h, --help show this help message and exit -f FOO old foo help --foo FOO new foo help |
add_help:设为False时,help信息里面不再显示-h --help信息。
3.添加参数选项
1
2
3
4
5
|
>>> parser.add_argument( ‘integers‘ , metavar= ‘N‘ , type= int , nargs= ‘+‘ , ... help= ‘an integer for the accumulator‘ ) >>> parser.add_argument( ‘--sum‘ , dest= ‘accumulate‘ , action= ‘store_const‘ , ... const =sum, default =max, ... help= ‘sum the integers (default: find the max)‘ ) |
add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])
name or flags:参数有两种,可选参数和位置参数。
添加可选参数:
1
|
>>> parser.add_argument( ‘-f‘ , ‘--foo‘ ) |
添加位置参数:
1
|
>>> parser.add_argument( ‘bar‘ ) |
parse_args()运行时,会用‘-‘来认证可选参数,剩下的即为位置参数。
1
2
3
4
5
6
7
8
9
10
|
>>> parser = argparse.ArgumentParser(prog= ‘PROG‘ ) >>> parser.add_argument( ‘-f‘ , ‘--foo‘ ) >>> parser.add_argument( ‘bar‘ ) >>> parser.parse_args([ ‘BAR‘ ]) Namespace(bar= ‘BAR‘ , foo=None) >>> parser.parse_args([ ‘BAR‘ , ‘--foo‘ , ‘FOO‘ ]) Namespace(bar= ‘BAR‘ , foo= ‘FOO‘ ) >>> parser.parse_args([ ‘--foo‘ , ‘FOO‘ ]) usage: PROG [-h] [-f FOO] bar PROG: error: too few arguments |
解析时没有位置参数就会报错了。
action:默认为store
store_const,值存放在const中:
1
2
3
4
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘--foo‘ , action= ‘store_const‘ , const =42) >>> parser.parse_args( ‘--foo‘ .split()) Namespace(foo=42) |
store_true和store_false,值存为True或False
1
2
3
4
5
6
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘--foo‘ , action= ‘store_true‘ ) >>> parser.add_argument( ‘--bar‘ , action= ‘store_false‘ ) >>> parser.add_argument( ‘--baz‘ , action= ‘store_false‘ ) >>> parser.parse_args( ‘--foo --bar‘ .split()) Namespace(bar=False, baz=True, foo=True) |
append:存为列表
1
2
3
4
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘--foo‘ , action= ‘append‘ ) >>> parser.parse_args( ‘--foo 1 --foo 2‘ .split()) Namespace(foo=[ ‘1‘ , ‘2‘ ]) |
append_const:存为列表,会根据const关键参数进行添加:
1
2
3
4
5
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘--str‘ , dest= ‘types‘ , action= ‘append_const‘ , const =str) >>> parser.add_argument( ‘--int‘ , dest= ‘types‘ , action= ‘append_const‘ , const = int ) >>> parser.parse_args( ‘--str --int‘ .split()) Namespace(types=[<type ‘str‘ >, <type ‘int‘ >]) |
count:统计参数出现的次数
1
2
3
4
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘--verbose‘ , ‘-v‘ , action= ‘count‘ ) >>> parser.parse_args( ‘-vvv‘ .split()) Namespace(verbose=3) |
help:help信息
version:版本
1
2
3
4
5
|
>>> import argparse >>> parser = argparse.ArgumentParser(prog= ‘PROG‘ ) >>> parser.add_argument( ‘--version‘ , action= ‘version‘ , version= ‘%(prog)s 2.0‘ ) >>> parser.parse_args([ ‘--version‘ ]) PROG 2.0 |
nargs:参数的数量
值可以为整数N(N个),*(任意多个),+(一个或更多)
1
2
3
4
5
6
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘--foo‘ , nargs= ‘*‘ ) >>> parser.add_argument( ‘--bar‘ , nargs= ‘*‘ ) >>> parser.add_argument( ‘baz‘ , nargs= ‘*‘ ) >>> parser.parse_args( ‘a b --foo x y --bar 1 2‘ .split()) Namespace(bar=[ ‘1‘ , ‘2‘ ], baz=[ ‘a‘ , ‘b‘ ], foo=[ ‘x‘ , ‘y‘ ]) |
值为?时,首先从命令行获得参数,若没有则从const获得,然后从default获得:
1
2
3
4
5
6
7
8
9
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘--foo‘ , nargs= ‘?‘ , const = ‘c‘ , default = ‘d‘ ) >>> parser.add_argument( ‘bar‘ , nargs= ‘?‘ , default = ‘d‘ ) >>> parser.parse_args( ‘XX --foo YY‘ .split()) Namespace(bar= ‘XX‘ , foo= ‘YY‘ ) >>> parser.parse_args( ‘XX --foo‘ .split()) Namespace(bar= ‘XX‘ , foo= ‘c‘ ) >>> parser.parse_args( ‘‘ .split()) Namespace(bar= ‘d‘ , foo= ‘d‘ ) |
更常用的情况是允许参数为文件
1
2
3
4
5
6
7
8
9
10
11
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘infile‘ , nargs= ‘?‘ , type=argparse.FileType( ‘r‘ ), ... default =sys.stdin) >>> parser.add_argument( ‘outfile‘ , nargs= ‘?‘ , type=argparse.FileType( ‘w‘ ), ... default =sys.stdout) >>> parser.parse_args([ ‘input.txt‘ , ‘output.txt‘ ]) Namespace(infile=<open file ‘input.txt‘ , mode ‘r‘ at 0x...>, outfile=<open file ‘output.txt‘ , mode ‘w‘ at 0x...>) >>> parser.parse_args([]) Namespace(infile=<open file ‘<stdin>‘ , mode ‘r‘ at 0x...>, outfile=<open file ‘<stdout>‘ , mode ‘w‘ at 0x...>) |
const:保存一个常量
default:默认值
type:参数类型
choices:可供选择的值
1
2
3
4
5
6
7
|
>>> parser = argparse.ArgumentParser(prog= ‘doors.py‘ ) >>> parser.add_argument( ‘door‘ , type= int , choices=range(1, 4)) >>> print(parser.parse_args([ ‘3‘ ])) Namespace(door=3) >>> parser.parse_args([ ‘4‘ ]) usage: doors.py [-h] {1,2,3} doors.py: error: argument door: invalid choice: 4 (choose from 1, 2, 3) |
required:是否必选
desk:可作为参数名
1
2
3
4
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘--foo‘ , dest= ‘bar‘ ) >>> parser.parse_args( ‘--foo XXX‘ .split()) Namespace(bar= ‘XXX‘ ) |
4.解析参数
参数有几种写法:
最常见的空格分开:
1
2
3
4
5
6
7
|
>>> parser = argparse.ArgumentParser(prog= ‘PROG‘ ) >>> parser.add_argument( ‘-x‘ ) >>> parser.add_argument( ‘--foo‘ ) >>> parser.parse_args( ‘-x X‘ .split()) Namespace(foo=None, x= ‘X‘ ) >>> parser.parse_args( ‘--foo FOO‘ .split()) Namespace(foo= ‘FOO‘ , x=None) |
长选项用=分开
1
2
|
>>> parser.parse_args( ‘--foo=FOO‘ .split()) Namespace(foo= ‘FOO‘ , x=None) |
短选项可以写在一起:
1
2
|
>>> parser.parse_args( ‘-xX‘ .split()) Namespace(foo=None, x= ‘X‘ ) |
parse_args()方法的返回值为namespace,可以用vars()内建函数化为字典:
1
2
3
4
5
|
>>> parser = argparse.ArgumentParser() >>> parser.add_argument( ‘--foo‘ ) >>> args = parser.parse_args([ ‘--foo‘ , ‘BAR‘ ]) >>> vars(args) { ‘foo‘ : ‘BAR‘ } |
以上是关于python内置库--argparse的主要内容,如果未能解决你的问题,请参考以下文章