一详解Python装饰器，真香！

1 函数基础

函数体内可以定义函数 和 函数可以被引用。有基础的朋友可以略过，直接看下一小节。

例1.  函数体内定义函数

def func1():
    def func2():
        return "in func2"

    res = func2()

    return res

在 func1 函数体内， 定义了一个 func2 函数并调用， 然后返回调用的结果。其实很多静态编程语言，如 C、Java，不支持在函数体内定义函数。但在 Python 中是支持的。

例2. 函数可以被引用

new_func = func1
print(new_func())

直接修改 my_decorator 业务就可以了，不会对其他业务产生副作用。

理解了上面两个问题，我们也就知道了装饰器的作用。如果不使用装饰器（如：例3 ）要么会产生代码冗余，要么就业务耦合在一起改一个业务影响到另外的业务。

4 装饰器的进阶

接下来，来看看装饰器高级一些的用法。

例5.  装饰器的参数

def cai_factory(cai_type='清炒'):
    def my_decorator(func):
        def wrapper():
            func()
            print(f'{cai_type}')
            print('出锅')

        return wrapper
    return my_decorator


@cai_factory('凉拌')
def tu_dou_si():
    print('清洗土豆')
    print('去皮，切丝')

定义了一个 cai_factory 函数（即：菜工厂），它里面包含了之前定义的 my_decorator 装饰器。cai_factory 函数接收 cai_type 参数用来指定如果制作食材。

定义 tu_dou_si 函数时，使用新的装饰器，并且传入参数即可。

例6. 被装饰函数的参数

def cai_factory(cai_type='清炒'):
    def my_decorator(func):
        def wrapper(cai_name):
            func(cai_name)
            print(f'{cai_type}')
            print('出锅')

        return wrapper
    return my_decorator


@cai_factory()
def cai(cai_name='土豆'):
    print(f'清洗{cai_name}')
    print('去皮，切丝')

定义一个用 cai_factory 修饰的函数 cai，接收 cai_name 参数来指定制作的食材。同时，在 cai_factory 装饰器中的 wrapper 函数中增加 cai_name 参数，并在调用 func 时，传入该参数。

例7. 被装饰函数的不定参数

def cai_factory(cai_type='清炒'):
    def my_decorator(func):
        def wrapper(*args, **kwargs):
            func(*args, **kwargs)
            print(f'{cai_type}')
            print('出锅')

        return wrapper
    return my_decorator


@cai_factory()
def cai(cai_name, cai_cnt):
    print(f'清洗{cai_name} {cai_cnt}')
    print('去皮，切丝')

cai('山药', '两个')

例6 只能支持1个参数，本例中的 wrapper 函数的参数改为 args 和 *kwargs，就可以支持任意参数，把装饰器变得更通用。

5 装饰类

类装饰器的用法跟函数是一样的。这次我们不做菜了，开始玩游戏了。游戏里面皮肤和英雄的关系就完美地表现了装饰类与装饰类的关系。下面我们就定义一个英雄类，然后定义一个皮肤类装饰它。

def pi_fu(cls):
    class PiFuClass:
        def __init__(self, name, pi_fu_name):
            self.wrapped = cls(name, pi_fu_name)
            self.pi_fu_name = pi_fu_name

        def display(self):
            self.wrapped.display()
            print(f'展示皮肤{self.pi_fu_name}')

    return PiFuClass


@pi_fu
class YingXiong:
    def __init__(self, name, pi_fu_name):
        self.name = name
        self.pi_fu_name = pi_fu_name

    def display(self):
        print(f'展示英雄{self.name}')


ya_se = YingXiong('亚瑟', '死亡骑士')
ya_se.display()

写法和用法跟函数装饰器类似，这里就不再赘述了。

6 内置装饰器

Python 内置了一些装饰器，这里列几个我们经常会用到的一些：

• @property：修饰类方法，使得我们可以像访问属性一样来获取一个函数的返回值
• @staticmethod：修饰类方法，表示该方法是一个静态方法，可以直接被调用无需实例化
• @wraps：函数来进行装饰，可以让我们在装饰器里面访问在装饰之前的函数的属性

简单总结一下，Python 装饰器支持修饰方法和类，可能增加被修饰方法和方法的功能。同时又能起到业务之间解耦、增加代码复用性和简化代码的作用。