如何更改当前__main__？导入脚本时能否切换并恢复__main__上下文？

这是个相当巧妙的需求！在Python里，__main__本质上就是模块的__name__属性——当你直接运行一个脚本时，它的__name__会被自动设为'__main__'；而当它被其他脚本导入时，__name__就是它的模块名（比如script_2）。要实现你说的「保存旧__main__上下文、临时切换到新__main__、返回时恢复」的效果，我们可以通过手动修改模块的__name__和sys.modules来实现，配合上下文管理器能让这个过程更安全、更优雅。

实现思路 #

核心逻辑是：

1. 进入切换上下文前，保存当前的__main__模块（也就是script_1）及其__name__属性
2. 将目标模块（script_2）替换为sys.modules['__main__']，并把它的__name__改成'__main__'
3. 执行需要在script_2作为__main__时的逻辑
4. 退出上下文时，恢复最初的__main__模块和所有属性，避免影响后续代码

具体代码示例 #

第一步：在script_2.py中定义上下文管理器

import sys
from contextlib import contextmanager

@contextmanager
def switch_to_main():
    # 保存当前的__main__模块和它的原始__name__
    old_main_module = sys.modules['__main__']
    old_main_name = old_main_module.__name__
    
    try:
        # 把当前模块（script_2）设为新的__main__
        sys.modules['__main__'] = sys.modules[__name__]
        # 修改script_2的__name__为'__main__'
        sys.modules[__name__].__name__ = '__main__'
        yield  # 这里是执行用户逻辑的地方
    finally:
        # 恢复原来的__main__上下文
        sys.modules['__main__'] = old_main_module
        old_main_module.__name__ = old_main_name
        # 把script_2的__name__改回原始模块名
        sys.modules[__name__].__name__ = __name__

# 用来验证当前__main__状态的测试函数
def verify_main_context():
    print(f"当前模块的__name__: {__name__}")
    print(f"sys.modules['__main__']指向的模块: {sys.modules['__main__'].__name__}")

# 如果script_2本身有直接运行时的逻辑，可以封装成函数
def run_as_main():
    if __name__ == '__main__':
        print("\n我是script_2的__main__逻辑！")

第二步：在script_1.py中调用切换逻辑

import sys
from script_2 import switch_to_main, verify_main_context, run_as_main

print("=== 初始状态（script_1作为__main__） ===")
verify_main_context()
print(f"script_1的__name__: {__name__}")

print("\n=== 临时切换到script_2作为__main__ ===")
with switch_to_main():
    verify_main_context()
    # 在这里可以执行需要script_2作为__main__的逻辑
    run_as_main()

print("\n=== 回到script_1，恢复原始上下文 ===")
verify_main_context()
print(f"script_1的__name__: {__name__}")

运行结果 #

执行script_1.py后，你会看到类似这样的输出：

=== 初始状态（script_1作为__main__） ===
当前模块的__name__: __main__
sys.modules['__main__']指向的模块: __main__
script_1的__name__: __main__

=== 临时切换到script_2作为__main__ ===
当前模块的__name__: __main__
sys.modules['__main__']指向的模块: __main__

我是script_2的__main__逻辑！

=== 回到script_1，恢复原始上下文 ===
当前模块的__name__: __main__
sys.modules['__main__']指向的模块: __main__
script_1的__name__: __main__

注意事项 #

• 这种方法属于「非常规操作」，Python的__main__设计初衷是区分脚本的运行方式（直接运行/导入），强行切换可能会导致依赖__main__的第三方库出现异常（比如日志、配置类库）
• 如果script_2中有if __name__ == '__main__':块，导入时不会自动执行，需要像示例中那样封装成函数，在上下文内手动调用
• 上下文管理器的finally块确保了即使切换过程中抛出异常，也能恢复原始的__main__上下文，避免程序陷入混乱

内容的提问来源于stack exchange，提问作者Toothpick Anemone