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)