非致命地终止模块的导入？

Question

我想使用if __name__ != '__main__':，然后在导入脚本时结束脚本的执行，而不是像往常一样使用if __name__ == '__main__': 并缩进文件中的所有其余代码。

但是，我无法确定是什么原因只会导致这种情况发生并且不会产生其他不良副作用。 sys.exit() 停止整个解释器，我尝试过的所有其他事情要么引发某种异常，要么是非法的。

更新：

我选择了@trutheality 的答案，因为它实现了我想要的并且非常容易开始使用。也就是说，我认为其他几个答案非常有趣和/或聪明——感谢所有回复的人——并计划在时间允许的情况下进一步调查其中的一些。我不知道做我想做的事会如此投入。

Answer 1

另一个黑客：

# code

if __name__ == "__main__": exec("""

# main code

#""")

所以...你已经失去了缩进，还有语法高亮和你正在使用的编辑器的任何其他功能，除非你在每次编辑时注释掉 if 行。

Answer 2

与我最初的答案非常相似，但有点吓人

部分导入.py

class PartialImport(Exception):
    def __init__(self, locals, msg = ""):
        self.module = locals

main.py

from partialimport import PartialImport
try:
   import foo
except PartialImport, e:
    #Note e.module and therefore foo will be a Dict and not a module instance!
    foo = e.module

foo.py

from partialimport import PartialImport

class Boo:
    pass

if __name__ != "__main__":    
    raise PartialImport(locals())

class Foo:
    pass

免责声明

这是一个非常可怕的黑客，会增加你的同伴谋杀你的可能性，但这确实有效。

Answer 3

main.py

try:
   import foo
except:
    print "Failed to import foo"

foo.py

文件顶部

if __name__ != "__main__":
   raise RunTimeError("foo must be run as main, not as a module.")

class foo(Object):
   pass

由于python是按顺序处理文件的，所以不会定义类Foo。

另一个想法是通过PEP 302 重载导入逻辑本身

Answer 4

IMO 最佳解决方案：有两个文件。

module_main.py

import actual_module.py

if __name__ != '__main__':
    raise RunTimeError("You should be importing actual_module.py, not module_main.py")

# Your "main" code

actual_module.py

# Actual module code (classes, functions, etc)

这是“干净”的，因为只有在实际出现问题时才会引发异常——任何人都不应该导入module_main.py，他们应该导入actual_module.py。

愚人节解决方案

如果您使用的是 python 2.3，则有来自 entrian 的 goto 模块，这显然有效！它是作为愚人节玩笑制作的，永远不应使用，（如果您查看源代码，您会明白原因：它增加了很多开销）但作为概念证明，它似乎就像我能找到的以任何简洁方式完成你想要的唯一方法。

from goto import goto, label

# Code that should always be imported (classes etc.)

if __name__ != "__main__":
    goto .end

# Stuff to be executed when this is main, NOT indented

label .end