python - 我们如何改变当前的 __main__

Question

假设我们执行script_1。因此，script_1 是 __main__。但是，script_1 从 script_2 导入一些类。有没有办法，当我们输入 script_2 保存旧的 __main__ 上下文时，使 script_2 成为新的 __main__，在返回到 script_1 之前，恢复旧的上下文？

Answer 1

重要说明:我同意这些评论。您需要这样做非常可疑

对于这项运动，这里有一种使用 importlib 来完成您所要求的事情的方法:

script_2.py:

if __name__ == '__main__':
    print('script_2 is the main!')

script_1.py:

import importlib.util
spec = importlib.util.spec_from_file_location('__main__', 'script_2.py')
mod = importlib.util.module_from_spec(spec)
spec.loader.exec_module(mod)
if __name__ == '__main__':
    print('script_3 is the main!')

终于

python script_1.py
> script_2 is the main!
> script_1 is the main!