我正在使用Python 2.7,我有以下简单的脚本,它需要一个命令行参数:
#!/usr/bin/env python
import sys
if (len(sys.argv) == 2):
print "Thanks for passing ", sys.argv[1]
else:
print "Oops."
我可以从命令行执行以下操作:
My-Box:~/$ ./useArg.py asdfkjlasdjfdsa
Thanks for passing asdfkjlasdjfdsa
或者这个:
My-Box:~/$ ./useArg.py
Oops.
我想从交互式编辑器中做类似的事情:
>>> import useArg asdfasdf
File "<stdin>", line 1
import useArg asdfasdf
^
SyntaxError: invalid syntax
但我不知道怎么做。如何在交互式编辑器中传递参数以导入/重新加载?
答案 0 :(得分:6)
你做不到。将代码包装在函数
中#!/usr/bin/env python
import sys
def main(args):
if (len(args) == 2):
print "Thanks for passing ", args[1]
else:
print "Oops."
if __name__ == '__main__':
main(sys.argv)
如果你从命令行执行你的脚本,你可以像以前一样,如果你想从解释器中使用它:
import useArg
useArg.main(['foo', 'bar'])
在这种情况下,您必须在列表的第一个位置使用一些虚拟值,因此大多数时候更好的解决方案是使用argparse library。您还可以在调用main函数之前检查命令行参数的数量:
import sys
def main(arg):
print(arg)
if __name__ == '__main__':
if len(sys.argv) == 2:
main(sys.argv[1])
else:
main('Oops')
您可以在此处执行if __name__ == '__main__':
时找到正确的解释:What does if __name__ == "__main__": do?