Question

假设我有两个python脚本A.py和B.py。我正在寻找一种从A中运行B的方法：

B认为它是__main__（因此B中if __name__=="__main__"块中的代码将会运行）
B实际上不是__main__（因此它不会覆盖sys.modules中的"__main__"条目）
B中引发的异常传播到A（即，可能会被A中的except子句捕获。
如果未捕获这些异常，则会在B中生成引用行号的正确回溯。

我尝试了各种技术，但似乎都没有满足我的所有要求。

使用子进程模块中的工具意味着B中的异常不会传播到A.
execfile("B.py", {})运行B，但它不认为它是主要的。
execfile("B.py", {'__name__': '__main__'})使得B.py认为它是主要的，但它似乎也搞砸了异常回溯打印，因此回溯指的是A中的行（即真正的__main__）。 / LI>
使用imp.load_source和__main__作为名称几乎可以工作，除了它实际修改了sys.modules，因此踩到__main__

有没有办法得到我想要的东西？

（我这样做的原因是因为我正在对现有的库进行一些清理。这个库没有真正的测试套件，只有一组产生特定输出的“示例”脚本。我正在尝试利用这些作为测试来确保我的清理不会影响库执行这些示例的能力，因此我想从我的测试套件中运行每个示例脚本。我希望能够在这些脚本中查看这些脚本的异常。测试脚本，以便测试脚本可以报告失败的类型，而不是仅在示例脚本引发某些异常时报告通用SubprocessError。）

Answer 1

您的用例是有道理的，但我仍然认为您最好不要重构测试，以便可以在外部运行。

你测试脚本有这样的东西吗？

def test():
    pass

if __name__ == '__main__':
    test()

如果没有，也许您应该将测试转换为调用test等函数。然后，从主测试脚本中，您可以：

import test1
test1.test()
import test2
test2.test()

提供运行测试的通用接口，测试本身使用。在__main__检查中拥有一大块代码并不是一件好事。

很抱歉，我没有回答您提出的问题，但我认为这是正确的解决方案，而且远离原始测试代码。

Answer 2

回答我自己的问题，因为结果很有趣，可能对其他人有用：

事实证明我错了：execfile("B.py", {'__name__': '__main__'}是最重要的方法。正确生成回溯。我看到错误的行号不是例外而是警告。这些警告是使用warnings.warn("blah", stacklevel=2)生成的。 stacklevel=2参数应该允许在使用弃用事物的地方引发弃用警告，而不是在警告调用时（参见the documentation）。

但是，似乎execfile-d文件不会为此目的计为“堆栈级别”，并且对于stacklevel目的而言是不可见的。因此，如果execfile-d模块顶层的代码导致stacklevel 2的警告，则不会在execfile -d源文件中的右行号处引发警告;相反，它是在运行execfile的文件的相应行号处引发的。

这很不幸，但我可以忍受它，因为这些只是警告，所以它们不会影响测试的实际性能。（起初我没有注意到只有那些受到行号不匹配影响的警告，因为在测试输出中混有很多警告和异常。）

从另一个python程序运行python程序（有一定的要求）

2 个答案: