Python解析命令行读取参数--argparse模块使用方法
在多个文件或者不同语言协同的项目中,python脚本经常需要从命令行直接读取参数。万能的python就自带了argprase包使得这一工作变得简单而规范。PS:optparse包是类似的功能,只不过写起来更麻烦一些。 如果脚本很简单或临时使用,没有多个复杂的参数选项,可以直接利用 import sys print "Input argument is %s" %(sys.argv[0]) 在shell脚本中运行python test.py help可以得到Input argument is help的结果。 1) 一般形式 但是大多数情况下,脚本很可能需要多个参数,而且每次参数的类型用处各不相同,那么这个时候在参数前添加标签表明参数的类型和用途便十分有用,而利用argparse模块可以很方便得实现这一目的。 同样用名为test.py的脚本举个栗子: import argparse parser = argparse.ArgumentParser(description="your script description") # description参数可以用于插入描述脚本用途的信息,可以为空 parser.add_argument('--verbose','-v',action='store_true',help='verbose mode') # 添加--verbose标签,标签别名可以为-v,这里action的意思是当读取的参数中出现--verbose/-v的时候 # 参数字典的verbose建对应的值为True,而help参数用于描述--verbose参数的用途或意义。 args = parser.parse_args() # 将变量以标签-值的字典形式存入args字典 if args.verbose: print "Verbose mode on!" else: print "Verbose mode off!" 运行 $ python test.py --help usage: test.py [-h] [--verbose] your script description optional arguments: -h,--help show this help message and exit --verbose,-v verbose mode 2) 必需参数 这种模式用于确保某些必需的参数有输入。 parser.add_argument('--verbose',required=True,type=int) required标签就是说--verbose参数是必需的,并且类型为int,输入别的类型会报错。 3)位置参数(positional arguments) 位置参数与sys.argv调用比较像,参数没有显式的--xxx或者-xxx标签,因此调用属性也与sys.argv相同。 parser.add_argument('filename') # 输入的第一个参数赋予名为filename的键 args = parser.parse_args() print "Read in %s" %(args.filename) 输入 此外,可以用nargs参数来限定输入的位置参数的个数,默认为1。当然nargs参数也可用于普通带标签的参数。 比如用: parser.add_argument('filename') parser.add_argument('num',nargs='*) 就可以运行python test.py text.txt 1 2 由于没有标签,所以用位置参数的时候需要比较小心。 4)输入类型 之前已经提到了用type参数就可以指定输入的参数类型。而这个type类型还可以表示文件操作的类型从而直接进行文件的读写操作。 parser.add_argument('file',type=argparser.FileType('r')) # 读取文件 args = parser.parse_args() for line in args.file: print line.strip() 5)参数默认值 一般情况下会设置一些默认参数从而不需要每次输入某些不需要变动的参数,利用default参数即可实现。
这个时候至直接运行python text.py就能得到Read in text.txt而不需要输入文件名了。 6)候选参数选择 表示该参数能接受的值只能来自某几个值候选值中,除此以外会报错,用choices参数即可。比如:
参考: https://mkaz.tech/python-argparse-cookbook.html https://docs.python.org/2/howto/argparse.html 以下是其他网友的补充: 一、简介: argparse是python用于解析命令行参数和选项的标准模块,用于代替已经过时的optparse模块。argparse模块的作用是用于解析命令行参数,例如python parseTest.py input.txt output.txt --user=name --port=8080。 二、使用步骤: 1:import argparse 2:parser = argparse.ArgumentParser() 3:parser.add_argument() 4:parser.parse_args() 解释:首先导入该模块;然后创建一个解析对象;然后向该对象中添加你要关注的命令行参数和选项,每一个add_argument方法对应一个你要关注的参数或选项;最后调用parse_args()方法进行解析;解析成功之后即可使用,下面简单说明一下步骤2和3。 三、方法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) 这些参数都有默认值,当调用parser.print_help()或者运行程序时由于参数不正确(此时python解释器其实也是调用了pring_help()方法)时,会打印这些描述信息,一般只需要传递description参数,如上。 四、方法add_argument(name or flags...[,action][,nargs][,const][,default][,type][,choices][,required][,help][,metavar][,dest]) 其中: name or flags:命令行参数名或者选项,如上面的address或者-p,--port.其中命令行参数如果没给定,且没有设置defualt,则出错。但是如果是选项的话,则设置为None nargs:命令行参数的个数,一般使用通配符表示,其中,'?'表示只用一个,'*'表示0到多个,'+'表示至少一个 default:默认值 type:参数的类型,默认是字符串string类型,还有float、int等类型 help:和ArgumentParser方法中的参数作用相似,出现的场合也一致 最常用的地方就是这些,其他的可以参考官方文档。下面给出一个例子,基本包括了常见的情形: import argparse def parse_args(): description = usage: %prog [options] poetry-file This is the Slow Poetry Server,blocking edition. Run it like this: python slowpoetry.py <path-to-poetry-file> If you are in the base directory of the twisted-intro package,you could run it like this: python blocking-server/slowpoetry.py poetry/ecstasy.txt to serve up John Donne's Ecstasy,which I know you want to do. parser = argparse.ArgumentParser(description = description) help = The addresses to connect. parser.add_argument('addresses',nargs = '*',help = help) help = The filename to operate on.Default is poetry/ecstasy.txt parser.add_argument('filename',help=help) help = The port to listen on. Default to a random available port. parser.add_argument('-p',--port',type=int,help=help) help = The interface to listen on. Default is localhost. parser.add_argument('--iface',help=help,default='localhost') help = The number of seconds between sending bytes. parser.add_argument('--delay',type=float,default=.7) help = The number of bytes to send at a time. parser.add_argument('--bytes',default=10) args = parser.parse_args(); return args if __name__ == '__main__': args = parse_args() for address in args.addresses: print 'The address is : %s .' % address print 'The filename is : %s .' % args.filename print 'The port is : %d.' % args.port print 'The interface is : %s.' % args.iface print 'The number of seconds between sending bytes : %f'% args.delay print 'The number of bytes to send at a time : %d.' % args.bytes</path-to-poetry-file> 运行该脚本:python test.py --port 10000 --delay 1.2 127.0.0.1 172.16.55.67 poetry/ecstasy.txt 输出为: The address is : 127.0.0.1 . (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |