(一)单一独立的参数
如果命令行输入的参数都是各自单一独立的,直接用个循环把所有参数逐一读出来就行了。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函数,并且要注意的是三双引号里面的内容不用缩进,直接从最开头开始写,否则会导致出错。