python获取命令行输入参数列表的实例代码
(一)单一独立的参数
如果命令行输入的参数都是各自单一独立的,直接用个循环把所有参数逐一读出来就行了。sys模块里面直接用args = sys.argv就可以获取到所有参数了(返回值args是一个list),args0]是执行的python文件路径名,args[1:]后面的才是真正命令行输入的参数,用个for循环就可以一一读取args列表,示例代码如下:
import sys def usage(): print """ This is usage. This is usage. This is usage. """ def main(argv): for arg in argv[1:]: if arg == '-t': pass elif arg == '-h' or arg == '--help': usage() sys.exit() elif arg == '-p': pass else: print "Error: invalid parameters" sys.exit() if __name__ == '__main__': main(sys.argv)
(二)参数结合
如果要读取:-t mytest 这样的组合参数,上面的方法就行不通了,在shell里面有shift可以直接移到下一个参数,但在python没有shift的功能,所以得换种方式。
一种方法是用getopt模块(官方文档关于getopt的解释:点击打开链接):getopt模块里面的getopt函数可以指定参数可能的类型并且获取所有参数。
用法:
getopt.getopt(args, options[, long_options])
options参数指定短式输入,也就是单横杠"-",例如"-h",long_options是可选的,用于指定长式输入,也就是双横杠"--",例如"--help"。如果是单一参数输入,options和long_options都只需要直接写参数名,例如下面这句代码指定两个单一参数:-h/--help -t/--test:
opts, args = getopt.getopt(sys.argv[1:], 'ht', ['help', 'test''])
如果是后面还要继续接参数的,options要在参数名后面加冒号":",long_options要在参数名后面加等号"=",例如下面这句代码指定三个参数:-h/--help -t/--test -r path / --root path:
opts, args = getopt.getopt(sys.argv[1:], 'htr:', ['help', 'test','root='])
getopt函数的返回值有两个,一般只用第一个,如下面的代码,第一个返回值opts :" is a list of (option, value) pairs",也就是一个 -t mytest 这样的元素对,当然如果-t后面不需要再接参数的话,那就没有value值了。利用这个pairs返回值,我们可以用个for循环获取参数,不管是单一独立的参数,还是结合的参数,都可以,示例代码如下:
import sys import getopt def usage(): print """ This is usage. This is usage. This is usage. """ def main(argv): try: opts, args = getopt.getopt(sys.argv[1:], 'm:p:h', ['miner=', 'params=', 'help']) except getopt.GetoptError: usage() sys.exit() for opt, arg in opts: if opt in ['-h', '--help']: pass elif opt in ['-t', '--test']: pass elif opt in ['-p', '--path']: param = arg print param sys.exit() else: print "Error: invalid parameters" usage() sys.exit() if __name__ == '__main__': main(sys.argv)
另外,还有个知识点就是,如果想要print一大段string怎么办,用三个双引号包住即可,如上面的代码中usage函数,并且要注意的是三双引号里面的内容不用缩进,直接从最开头开始写,否则会导致出错。