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

我想将一个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')

我能把第一个来自print的from py1 import test静音吗？
我不能评论print中的py1，因为它正在其他地方使用。

最佳答案

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