Python中如何从嵌套调用的check函数直接返回至explore函数？

问题背景 #

给定以下代码：

def check():
    if (cond):
        # 希望在这里直接回到explore()
        pass
    else:
        return 1 # here return to f2

def cannot_be_modified():
    try:
        check() # inserted
        print("hello from cannot_be_modified")
        return 2
    except Exception as e:
        print(e)

def explore():
    cannot_be_modified()
    print("hello from explore")
    return 0

explore()

调用栈为：explore() -> cannot_be_modified() -> check()

需求：当check函数满足特定条件时，退出check和cannot_be_modified函数，直接回到explore函数继续执行，但无法修改cannot_be_modified函数，且常规异常会被该函数的except Exception捕获。

核心思路是利用Python的异常层级特性：自定义一个继承自BaseException的异常，这类异常不会被except Exception语句捕获，从而直接冒泡到explore()中处理，完美绕过cannot_be_modified()的异常拦截逻辑。

具体实现代码 #

1. 定义专属跳转异常

class JumpBackToExplore(BaseException):
    pass

2. 修改check()函数，满足条件时抛出该异常

def check():
    cond = True  # 替换为你的实际判断条件
    if cond:
        # 抛出异常直接跳回explore函数
        raise JumpBackToExplore()
    else:
        return 1

3. 修改explore()函数，捕获自定义异常

def explore():
    try:
        cannot_be_modified()
    except JumpBackToExplore:
        # 捕获到跳转信号后，直接跳过后续处理，继续执行explore的代码
        pass
    print("hello from explore")
    return 0

方案原理 #

Python的异常体系中，Exception是大部分业务异常的父类，但BaseException是所有异常的根类（包含Exception和系统级异常如SystemExit）。cannot_be_modified()中的except Exception只会捕获继承自Exception的异常，我们自定义的JumpBackToExplore继承自BaseException，因此不会被它拦截，会直接传递到explore()的except块中。

这样当check()触发条件时，cannot_be_modified()中check()调用后的print("hello from cannot_be_modified")和return 2都不会执行，直接回到explore()执行后续代码。

注意事项 #

• 不要滥用BaseException子类，它会绕过常规的Exception捕获逻辑，这里是因为无法修改cannot_be_modified()才采用这个方案。
• 务必在explore()中明确捕获自定义异常，避免异常扩散到更上层调用栈。

内容的提问来源于stack exchange，提问作者Xuan Linh Ha