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

在编程世界中，某些函数经过装饰器的处理后，可能会呈现出令人意想不到的属性变化。就像人类中的某些名人通过精心的打扮让人难以分辨性别一样，Python中的函数也能通过装饰器获得新的属性。然而，这种装饰后的函数往往会失去原有的属性，这在某些情况下并非我们所期望的。接下来，我们将探讨如何在使用装饰器时保留函数的原始属性。

首先，让我们定义一个简单的函数，并观察其基本属性：

def iron_egg():    """我是铁蛋儿~"""    return

执行上述代码，可以看到函数的名称和文档字符串：

iron_egg我是铁蛋儿~

接下来，我们定义一个装饰器函数：

def decorator(func):    def spring_blossom(*args, **kwargs):        """我是春花儿~"""        return func(*args, **kwargs)    return spring_blossom

将上述装饰器应用到iron_egg函数上：

iron_egg = decorator(iron_egg)

此时，函数的属性发生了变化：

iron_egg.__name__spring_blossomiron_egg.__doc__我是春花儿~

显然，装饰后的函数失去了原有的名称和文档字符串，这在大多数情况下并非我们想要的。为了保留函数的原始属性，我们可以使用functools模块提供的wraps装饰器。wraps装饰器能够将装饰器函数的属性（如名称和文档字符串）复制到原函数上，从而保留函数的原始属性。

修改后的装饰器代码如下：

from functools import wrapsdef decorator(func):    @wraps(func)    def spring_blossom(*args, **kwargs):        """我是春花儿~"""        return func(*args, **kwargs)    return spring_blossom

应用wraps装饰器后，函数的属性将保留原样：

iron_egg = decorator(iron_egg)iron_egg.__name__iron_eggiron_egg.__doc__我是铁蛋儿~

通过使用functools.wraps，我们可以确保装饰器不影响函数的原始属性。这使得函数在使用装饰器后依然保持其原始的名称和文档字符串，从而避免了不必要的混淆。

functools.wraps背后的实现机制涉及到functools.partial函数，这是一个强大的工具，可以用于部分应用函数。虽然partial在本例中并未直接使用，但它为wraps实现属性复制提供了底层支持。

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