实现Python脚本模拟正常执行结束的无干扰退出函数

需要实现Python 3.12脚本的退出行为，使其在以下所有环境中与正常执行完毕完全一致：

• 以python script.py运行；
• 以python -i script.py运行；
• 在Thonny（或其他IDE）的shell中运行。

示例脚本如下（实际场景中退出函数可能嵌套在分支、循环等结构中）：

x = 1
print("HELLO")

magic_exit_function()

print("won't see this")
y = 1

要求magic_exit_function满足：
(a) 确保输出HELLO，但不输出won't see this；
(b) 交互模式下脚本结束后可查看x的值，但无法查看y的值；
(c) 不在stdout或stderr产生冗余输出。

已尝试的方案：

1. 使用exit/quit/sys.exit：普通模式有效，python -i下产生SystemExit回溯，Thonny中终止子进程无法查x。
2. 自定义继承Exception的异常：有冗余输出，但Thonny中可查x。
3. 自定义Exception并设sys.excepthook忽略：普通和交互模式有效，但Thonny仍有冗余输出。
4. 自定义BaseException并设sys.excepthook：Thonny无冗余输出，但终止子进程无法查x。
5. 用sys.settrace修改帧f_lineno：几乎全场景有效，但会执行模块最后一行导致y=1执行，违反需求。

询问：除修改magic_exit_function外，是否有其他可行方案？

各尝试方案的实现代码如下：

def magic_exit_function1():
    import sys
    sys.exit()  # or exit() or quit() or raise SystemExit()


def magic_exit_function2():
    class MyExit(Exception):
        pass

    raise MyExit()


def magic_exit_function3():
    import sys

    class MyExit(Exception):
        pass

    def hook(exc_type, exc, tb):
        if exc_type is MyExit:
            return None
        return sys.__excepthook__(exc_type, exc, tb)

    sys.excepthook = hook
    raise MyExit()


def magic_exit_function4():
    import sys

    class MyExit(BaseException):
        pass

    def hook(exc_type, exc, tb):
        if exc_type is MyExit:
            return None
        return sys.__excepthook__(exc_type, exc, tb)

    sys.excepthook = hook
    raise MyExit()


def magic_exit_function5():
    import sys
    import inspect

    def trace(frame, event, arg):
        if event != 'line':
            return trace

        frame.f_lineno = last_line
        return trace

    sys.settrace(lambda *args, **kwargs: None)
    frame = inspect.currentframe().f_back
    _, _, last_line = list(frame.f_code.co_lines())[-1]
    frame.f_trace = trace

方案1：全局捕获SystemExit并屏蔽回溯 #

在脚本开头添加以下代码，重写异常处理钩子，让SystemExit不会产生回溯输出，同时保留交互环境的上下文：

import sys

def silent_system_exit_hook(exc_type, exc_val, exc_tb):
    if exc_type is SystemExit:
        return  # 屏蔽SystemExit的回溯输出
    sys.__excepthook__(exc_type, exc_val, exc_tb)  # 其他异常正常处理

sys.excepthook = silent_system_exit_hook

此时magic_exit_function仍使用sys.exit()即可，该方案效果：

• python script.py：正常退出，无冗余输出，不执行后续代码；
• python -i script.py：退出脚本后进入交互模式，可查看x，无回溯输出；
• Thonny环境：脚本执行到sys.exit()后停止，保留shell上下文，可查看x的值，无冗余输出。

方案2：利用代码结构控制执行流程 #

如果允许调整脚本的整体结构，可以将后续需要跳过的代码放在单独代码块中，通过标志位控制是否执行：

x = 1
print("HELLO")

continue_execution = True
try:
    magic_exit_function()
except:
    continue_execution = False

if continue_execution:
    print("won't see this")
    y = 1

这种方式无需修改magic_exit_function的实现，只要它能抛出异常（或通过其他方式触发标志位），即可跳过后续代码。缺点是需要调整脚本结构，对于嵌套在复杂逻辑中的退出点需额外处理。

方案3：使用解释器启动脚本时的自定义配置 #

如果允许在启动脚本时添加额外参数，可以利用Python的-c参数先设置异常钩子，再执行脚本：

python -c "import sys; sys.excepthook = lambda t,v,tb: None if t is SystemExit else sys.__excepthook__(t,v,tb); exec(open('script.py').read())"

这种方式无需修改脚本内容，通过启动命令预先配置异常处理，同样可以实现无回溯的退出，且交互模式下保留上下文。

内容的提问来源于stack exchange，提问作者Nikola Benes