python中装饰器的写法

装饰器本质上是一个Python函数（返回值是一个函数对象），它可以让其他函数在不需要任何代码变动的前提下增加额外功能。它经常用于有切面需求的场景，比如：插入日志、性能测试、事务处理、缓存、权限校验等场景。有了装饰器，我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。

第一种：不带参数的函数装饰器

写一个最普通的装饰器，这种装饰器是不能传参的，只能对被装饰函数，执行固定逻辑。

def logger(func):   # func是被装饰的函数
    def wrapper(*args, **kw):
        print('我准备开始执行:{}函数了'.format(func.__name__))

        # 被装饰的函数真正执行
        func(*args, **kw)

        print('函数执行完啦')
    return wrapper

@logger
def add(x, y):
    print('{} + {} = {}'.format(x, y, x+y))

add(200, 50)

第二种：带参数的函数装饰器

写一个能传参的装饰器，可以根据不同的场景进行调整。

import time

def logger(tt):
    def wrapper1(func):   # 第一层接收被装饰函数
        def wrapper2(*args, **kw):   # 第二层实现装饰逻辑
            print('我在{}时刻开始执行:{}函数了'.format(tt, func.__name__))

            # 被装饰的函数真正执行
            func(*args, **kw)

            print('在{}时刻函数执行完啦'.format(time.time()))
        return wrapper2
    return wrapper1

@logger(time.time())
def add(x, y):
    print('{} + {} = {}'.format(x, y, x+y))

add(200, 50)

第三种：不带参数的类装饰器

基于类的装饰器，必须实现init和call两个内置函数，前者用于接收被装饰函数，后者用于实现装饰逻辑。

class logger(object):
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print('[INFO]: the function {func}() is running...'.format(func=self.func.__name__))
        return self.func(*args, **kwargs)

@logger
def add(x, y):
    print('{} + {} = {}'.format(x, y, x+y))

add(200, 50)

第四种：带参数的类装饰器

带参数的类装饰器和不带参数的类装饰器有很大不同，init不再接收被装饰函数而是接收传入参数，call既接收被装饰函数又要实现装饰逻辑。

class logger(object):
    def __init__(self, level='INFO'):
        self.level = level

    def __call__(self, func):  # 接收被装饰函数
        def wrapper(*args, **kwargs):
            print('[{level}]: the function {func}() is running...'.format(level=self.level, func=func.__name__))
            func(*args, **kwargs)
        return wrapper

@logger(level='WARNING')
def add(x, y):
    print('{} + {} = {}'.format(x, y, x+y))

add(200, 50)