假设我有两个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__).
>使用imp.load_source和__main__作为名称几乎可以工作,除了它实际修改sys.modules,因此踩踏__main__的现有值
有什么方法可以得到我想要的东西吗？
(我这样做的原因是因为我正在对现有的库进行一些清理.这个库没有真正的测试套件,只有一组产生特定输出的“示例”脚本.我试图利用它们作为测试以确保我的清理不会影响库执行这些示例的能力,因此我想从我的测试套件中运行每个示例脚本.我希望能够在测试脚本中看到这些脚本的异常,以便测试脚本可以报告失败的类型,而不是仅在示例脚本引发某些异常时报告通用的SubprocessError.)
最佳答案