Python的exec和__name__

Question

當我運行：

exec "print __name__" 

它打印__main__。

但是當我運行：

exec "print __name__" in {} 

它打印__builtin__。

如何使第二個示例也打印__main__？

我試圖實現的是使用exec運行一段代碼，以便從它的角度來看，它看起來像是從命令行運行。

我想用乾淨的範圍來代碼，但第二個例子打破了依賴if __name__ == "__main__"的代碼。如何解決這個問題？

Answer 1

你可以使用imp.load_module代替：

import imp 

with open(mainfile) as src: 
    imp.load_module('__main__', src, mainfile, (".py", "r", imp.PY_SOURCE)) 

此文件導入爲__main__模塊，執行它。

請注意，當類型設置爲imp.PY_SOURCE時，它需要一個實際的文件對象，所以如果您的源代碼來自文件以外的其他位置，您需要創建一個臨時文件才能工作。

否則，可以隨時設置__name__手動：

>>> src = '''\ 
... if __name__ == '__main__': print 'Main!' 
... else: print 'Damn', __name__ 
... ''' 
>>> exec src 
Main! 
>>> exec src in {} 
Damn __builtin__ 
>>> exec src in {'__name__':'__main__'} 
Main! 

Answer 2

一種解決方案是提供__name__明確在你exec ution字典：

exec "print __name__" in {'__name__': '__main__'}