本文共 2686 字，大约阅读时间需要 8 分钟。

Python 装饰器入门指南

装饰器是Python 中一项强大的功能，它允许开发者在不修改原始函数代码的情况下，通过增加额外的功能或改变函数行为来增强函数的功能性。这对于日志记录、性能测试、事务处理等场景尤为实用。

装饰器的基本概念

装饰器本质上是一个接受函数作为参数的高阶函数，并返回一个新的函数。通过装饰器，可以在函数执行前后添加自定义逻辑。例如：

def my_decorator(func):    def wrapper():        print("在函数执行前做一些事情")        func()        print("在函数执行后做一些事情")    return wrapper@my_decoratordef say_hello():    print("Hello!")#say_hello() 会被替换为 wrapper 函数的调用，执行上述 print 语句

在这个示例中，my_decorator 接收目标函数 func 作为参数，并返回一个新的函数 wrapper。当使用 @my_decorator 装饰 say_hello 函数时，实际上是在调用 my_decorator(say_hello)，并将返回的 wrapper 函数赋值给 say_hello。这样一来，执行 say_hello() 时会先输出 "在函数执行前做一些事情"，然后执行 func()，最后输出 "在函数执行后做一些事情"。

装饰器支持参数

有些情况下，我们需要将参数传递给装饰器本身。这可以通过再包裹一层函数来实现。例如：

def repeat(num_times):    def decorator(func):        def wrapper():            for _ in range(num_times):                func()        return wrapper    return decorator@repeat(3)def greet():    print("你好！")#执行 greet() 会输出 "你好！" 三次

在这个示例中，repeat 是一个接受参数 num_times 的函数，它返回一个真正的装饰器 decorator。decorator 接收目标函数 func 作为参数，并返回一个新的函数 wrapper。wrapper 函数会在调用 func 时重复执行 num_times 次。

装饰带有参数的函数

如果被装饰的函数本身带有参数，我们需要在 wrapper 函数中传递这些参数。例如：

def log_function_call(func):    def wrapper(*args, **kwargs):        print(f"调用函数 {func.__name__}，参数为：{args}, {kwargs}")        result = func(*args, **kwargs)        print(f"函数 {func.__name__} 返回值为：{result}")        return result    return wrapper@log_function_calldef add(a, b):    return a + bresult = add(3, 4)

在这个示例中，log_function_call 装饰器会在调用 add 函数前后打印相关信息。wrapper 函数使用 *args 和 **kwargs 接收任意数量的位置参数和关键字参数，并将它们传递给 func。

Python 内置装饰器

Python 提供了一些内置的装饰器，如 @staticmethod、@classmethod 和 @property，这些装饰器用于修改类方法的行为。例如：

class MyClass:    @staticmethod    def static_method():        print("这是一个静态方法")        @classmethod    def class_method(cls):        print(f"这是一个类方法，类名是 {cls.__name__}")        @property    def prop(self):        return "这是一个属性"MyClass.static_method()MyClass.class_method()obj = MyClass()print(obj.prop)

装饰器在日常开发中的使用建议

from functools import wrapsdef my_decorator(func):    @wraps(func)    def wrapper(*args, **kwargs):        print("在函数执行前做一些事情")        result = func(*args, **kwargs)        print("在函数执行后做一些事情")        return result    return wrapper@my_decoratordef say_hello():    """这是一个打招呼的函数"""    print("Hello!")print(say_hello.__name__)  # 输出: say_helloprint(say_hello.__doc__)   # 输出: 这是一个打招呼的函数

3. 避免过度使用装饰器：虽然装饰器非常强大，但过度使用会导致代码难以理解。只有在确实需要增强或修改函数行为时才使用装饰器。

4. 测试装饰器：像测试普通函数一样测试装饰器，确保它们按预期工作。可以使用单元测试框架（如 unittest 或 pytest）来编写测试用例。

5. 考虑性能影响：装饰器可能会引入额外的开销，特别是在高频率调用的场景中。如果性能敏感，需要仔细评估装饰器的影响。

通过遵循这些建议，开发者可以更高效地利用装饰器来提升代码的可读性和功能性，同时避免因过度使用装饰器带来的潜在问题。

转载地址：http://lbofk.baihongyu.com/