我有一个我正在构建的Python程序,可以以两种方式之一运行:第一种是调用Python main.py,它会以友好的方式提示用户输入,然后通过程序运行用户输入。另一种方法是调用python batch.py -file-它将传递所有友好的输入收集,并通过程序一次性运行整个文件的输入值。
问题是,当我运行batch.py时,它从main.py中导入了一些变量/方法/等,当它运行以下代码时:
import main
在程序的第一行,它立即出错,因为它试图运行main.py中的代码。
如何阻止Python运行我正在导入的主模块中包含的代码?
由于Python的工作方式,当它导入模块时,有必要运行它们。
为了防止模块中的代码在导入时被执行,但只在直接运行时执行,你可以使用以下if来保护它:
if __name__ == "__main__":
# this won't be run when imported
您可能希望将此代码放在main()方法中,以便可以直接执行文件,或者导入模块并调用main()。例如,假设这是在文件foo.py中。
def main():
print "Hello World"
if __name__ == "__main__":
main()
这个程序可以通过python foo.py运行,也可以通过另一个python脚本运行:
import foo
...
foo.main()