所有,
我最近发现了PYTHONSTARTUP环境变量,并期待设置一些实用程序功能以自动加载到我的解释器中。但是,我想做的一件事是使用相同的脚本来设置环境变量本身。
我的问题是确定文件何时作为脚本运行。我的想法是使用if __name__ == "__main__":
技巧来确定文件何时以脚本运行,但是测试表明,当通过PYTHONSTARTUP加载文件时,名称显示为"__main__"
。
有谁知道一种方法来识别文件何时作为脚本运行以及何时通过PYTHONSTARTUP加载?
最佳答案
您可以检查PYTHONSTARTUP环境变量是否设置为当前文件名(通过__file__
)。
import os
if os.environ.get('PYTHONSTARTUP') == __file__:
print "Used as startup!"
对我来说很好。