python - 静音打印导入的 Python 脚本

Question

我想将一个 python 脚本导入另一个脚本。

$ cat py1.py
test=("hi", "hello")

print test[0]

$ cat py2.py
from py1 import test

print test

如果我执行 py2.py:

$ python py2.py 
hi
('hi', 'hello')

我是否可以将来自 from py1 import test 的第一个 print 静音？

我无法在 py1 中注释 print，因为它正在其他地方使用。

Answer 1

py1.py 使用 if __name__=="__main__":

所以你的 py1.py 应该是这样的:

def main():
    test=("hi", "hello")

    print test[0]

if __name__=="__main__":
    main()  

这将允许您仍然正常使用 py1.py，但是当您导入它时，它不会运行 main() 函数，除非您调用它。

This explains what's going on