Python中复用通用异常处理逻辑：动态构建except语句方案

问题描述 #

我有多个执行不同操作的函数，它们的错误处理逻辑大体相似但存在细微差异。所有函数都需要处理ErrorA和ErrorB，我希望重构代码，避免在每个函数中重复编写这两个异常的except子句。请问Python中有没有可行的实现方法？要求不改变原有代码行为，也不使用嵌套try-except块。

原代码示例：

def func_a():
   try:
       do_action_a()
   except ErrorA:
       handle_error_a()
   except ErrorB:
       handle_error_b()
   except ErrorC:
       handle_error_c()

def func_b():
   try:
       do_action_b()
   except ErrorA:
       handle_error_a()
   except ErrorB:
       handle_error_b()
   except ErrorD:
       handle_error_d()

def func_c():
   try:
       do_action_c()
   except ErrorA:
       handle_error_a()
   except ErrorB:
       handle_error_b()
   except Exception:
       handle_general_exception()

解决方案：使用装饰器抽离通用逻辑 #

装饰器是Python中抽离重复代码的常用方案，这里可以把ErrorA和ErrorB的处理逻辑封装到装饰器中，每个函数只保留自身特有的异常处理逻辑，完全符合需求：

1. 定义通用异常处理装饰器 #

def handle_common_errors(func):
    def wrapper(*args, **kwargs):
        try:
            return func(*args, **kwargs)
        except ErrorA:
            handle_error_a()
        except ErrorB:
            handle_error_b()
    return wrapper

2. 修改原有函数，保留特有异常处理 #

给每个函数加上装饰器，同时移除重复的ErrorA/ErrorB处理代码，只保留各自特有的异常分支：

@handle_common_errors
def func_a():
    try:
        do_action_a()
    except ErrorC:
        handle_error_c()

@handle_common_errors
def func_b():
    try:
        do_action_b()
    except ErrorD:
        handle_error_d()

@handle_common_errors
def func_c():
    try:
        do_action_c()
    except Exception:
        handle_general_exception()

逻辑说明 #

• 当函数内部操作抛出ErrorA或ErrorB时，会跳过函数内部的try-except（因为内部没有对应分支），直接被装饰器的try块捕获处理，和原代码行为完全一致。
• 若抛出的是函数特有的异常（如ErrorC/ErrorD），则由函数内部的except分支处理，不会触发装饰器的通用逻辑。
• 整个实现没有嵌套try-except块，完全保留了原代码的异常处理优先级和行为。

替代方案：上下文管理器 #

如果你更倾向于上下文管理器的写法，也可以实现同样的效果：

1. 定义上下文管理器类 #

class CommonErrorHandler:
    def __enter__(self):
        pass
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_type == ErrorA:
            handle_error_a()
            return True  # 阻止异常继续传播
        elif exc_type == ErrorB:
            handle_error_b()
            return True
        return False  # 其他异常交给后续逻辑处理

2. 改造原有函数 #

def func_a():
    with CommonErrorHandler():
        try:
            do_action_a()
        except ErrorC:
            handle_error_c()

def func_b():
    with CommonErrorHandler():
        try:
            do_action_b()
        except ErrorD:
            handle_error_d()

def func_c():
    with CommonErrorHandler():
        try:
            do_action_c()
        except Exception:
            handle_general_exception()

这种方式通过上下文管理器的__exit__方法处理通用异常，同样满足不嵌套try-except、不改变原行为的要求。

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