main中的python命令行参数,跳过脚本名称
问题内容:
这是我的剧本
def main(argv):
if len(sys.argv)>1:
for x in sys.argv:
build(x)
if __name__ == "__main__":
main(sys.argv)
所以从命令行我写 python myscript.py commandlineargument
我希望它跳过myscript.py
和简单的运行commandlineargument
通过commandlineargument(n)
所以我知道我的for循环并不能解决这个问题,但是我如何使它做到这一点呢?
问题答案:
由于sys.argv是一个列表,因此可以使用slicing
sys.argv[1:]
:
def main(argv):
for x in argv[1:]:
build(x)
if __name__ == "__main__":
main(sys.argv)
但是,如果您只能有一个脚本参数,则只需通过index:获得它sys.argv[1]
。但是,您应该检查的长度sys.argv
是否大于1,如果不超过则抛出错误,例如:
def main(argv):
if len(argv) == 1:
print "Not enough arguments"
return
else:
build(argv[1])
if __name__ == "__main__":
main(sys.argv)