首页 > 编程 > Python > 正文

python获取命令行输入参数列表的实例代码

2020-02-15 21:59:31
字体:
来源:转载
供稿:网友

(一)单一独立的参数

如果命令行输入的参数都是各自单一独立的,直接用个循环把所有参数逐一读出来就行了。sys模块里面直接用args = sys.argv就可以获取到所有参数了(返回值args是一个list),args0]是执行的python文件路径名,args[1:]后面的才是真正命令行输入的参数,用个for循环就可以一一读取args列表,示例代码如下:

import sysdef 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 sysimport getoptdef 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)            
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表