使用 functools.wraps 装饰器尝试类的每个函数

Question

我试图定义一个装饰器来执行一个类方法，首先尝试，如果检测到错误，则提出它并提及失败的方法，以便用户可以看到哪个方法出现错误。

这里我展示了我的代码的MRE（最小的、可重现的示例）。

from functools import wraps

def trier(func):
    """Decorator for trying A-class methods"""
    @wraps(func)
    def inner_func(self, name, *args):
        
        try:
            func(self, *args)
        
        except:
            print(f"An error apeared while {name}")
    
    return inner_func
    
class A:
    def __init__(self):
        self._animals = 2
        self._humans = 5
    
    @trier('getting animals')
    def animals(self, num):
        return self._animals + num
    
    @trier('getting humans')
    def humans(self):
        return self._humans

A().animals

出现许多错误，例如：

类型错误：inner_func() 缺少 1 个必需的位置参数：“名称”

或误解 self 类与 self 函数。

Answer 1

作为Stefan答案的替代方案，下面简单地使用@trier而不带任何参数来装饰函数，然后在打印错误消息时我们可以使用func.__name__。

from functools import wraps

def trier(func):
    """Decorator for trying A-class methods"""
    @wraps(func)
    def inner_func(self, *args, **kwargs):

        try:
            return func(self, *args, **kwargs)

        except:
            print(f"An error apeared in {func.__name__}")

    return inner_func

class A:
    def __init__(self):
        self._animals = 2
        self._humans = 5

    @trier
    def animals(self, num):
        return self._animals + num

    @trier
    def humans(self):
        return self._humans

print(A().animals(1))

我还修复了代码中的几个错误：在 trier 的 try 和 except 中，调用 func 的结果从未返回，并且您需要包含 *除了 *args 之外，还有 *kwargs，以便您可以使用命名参数。即 A().animals(num=1) 仅在处理 kwargs 时才有效。

Answer 2

对于带有参数的装饰器，您需要多一层嵌套：

from functools import wraps


def trier(name):
    def wrapper(func):
        @wraps(func)
        def inner(*args, **kwargs):
            try:
                return func(*args, **kwargs)
            except:
                print(f"An error apeared while executing {name!r}")
        return inner
    return wrapper


class A:

    def __init__(self):
        self._animals = 2
        self._humans = 5

    @trier('getting animals')
    def animals(self, num):
        return self._animals + num

    @trier('getting humans')
    def humans(self):
        return self._hoomans # wrong attribute name


a = A()
a.humans() # An error apeared while executing 'getting humans'

Answer 3

我会这样做，希望它有帮助。

from functools import wraps
import sys


def trier(func):
    """Decorator for trying A-class methods"""
    @wraps(func)
    def inner_func(self, *args, **kwargs):
        print(f'Calling {func.__name__}')
        try:
            func(self, *args, **kwargs)

        except:
            print(f"An error apeared on function : {func.__name__}")
            e = sys.exc_info()[2]
            raise Exception(f"Exception occured on line: {e.tb_next.tb_lineno}")

    return inner_func


class A:
    def __init__(self):
        self._animals = 2
        self._humans = 5

    @trier
    def get_animals(self, num):
        return self._animals + num

    @trier
    def get_humans(self):
        return self._humans

    @trier
    def function_raising_exception(self):
        raise Exception('This is some exception')


if __name__ == "__main__":
    a = A()
    a.get_animals(2)
    a.function_raising_exception()

使用e = sys.exc_info()[2]，您还可以获得回溯消息并指向发生异常的行。

值得注意的是，当编写装饰器时，它应该适用于您在应用程序中使用的其他功能。