python 装饰器

__author__ = 'Kaiming'

# 装饰器按照装饰函数是否带参数和被装饰函数是否带参数可以分为4类

import functools

# P1——两者都不带参数

# 定义装饰函数，被装饰函数作为参数

def decorator_one(f1):

print('start')

print('call ' + f1.__name__)

print('end')

# 被装饰函数不带参数，返回被装饰函数,最外层函数必须接受一个返回函数

return f1

@decorator_one

# 采取和上面定义不同的名字，否则会出错

def f1():

pass

f1()

# P2——装饰函数不带参数，被装饰函数带参数

# 定义装饰函数，被装饰函数作为参数

def decorator_two(f2):

# 定义包装函数，用于传递给被装饰函数当做参数,使用可变参数和关键字参数可以提升灵活性

def wrapper_two(*args, **kw):

print('start')

print('call' + f2.__name__)

f2(*args, **kw)

print('end')

# 这里的return返回的是decorator_two这个函数的返回值

return wrapper_two

@decorator_two

def f2(s):

print(s)

'''流程是:f2=decorator_two(f2),然后运行wrapper_two函数，在f2执行前后进行

处理,最后还要把函数最为一个返回值，这个是装饰函数要求的'''

f2('被装饰函数f2的参数')

# P3——装饰函数带参数，被装饰函数不带参数

# 装饰函数外层加一层函数，用来传递装饰函数的参数

def my_decorator_three(*args, **kw):

# 这里才是装饰函数开始的地方，传入装饰函数参数f3

def decorator_three(f3):

# @functools.wraps(f3) 这句在这里可加可不加，反正不会出现函数名字指定错误

print('start')

print('f3的装饰器函数自带参数为:', *args)

print('call ' + f3.__name__)

print('end')

# 这里需要返回被装饰函数本身f3,当做

return f3

return decorator_three

'''注意P3的调用逻辑是f3=my_decorator_three(*args,**kw)(f3)(),最后的(f3)在这里是

使用前面的返回函数decorator_three(f3),可以完成装饰函数自带参数实现的功能，最后的()实际上

再使用不带参数的f3()函数本身完成被装饰函数本身的功能'''

# 这里采用最外层的函数

@my_decorator_three('my_decorator_three')

def f3():

pass

f3()

# P4——装饰器函数和被装饰函数都带参数

# 这层传递的参数是装饰函数本身需要使用的参数

def my_decorator_four(*args_decorator, **kw_decorator):

# 这一层函数传递需要被装饰的函数f4

def decorator_four(f4):

@functools.wraps(f4)

# 这一层包装函数用于传递被装饰函数f4所需要的参数

def wrapper(*args, **kw):

print('start')

print(f4.__name__ + ' is working')

# 在这个函数内已经使用了带参数的被装饰函数f4，所以最后也不需要返回函数f4了

f4(*args, **kw)

print('end')

return wrapper

return decorator_four

'''P4的调用逻辑是f4=my_decorator_four(**args_decorator,**kw_decorator)(f4)(*args,**kw);这个逻辑比较长，首

先是他用最外层的函数，传递装饰函数需要的参数；然后用于最外层函数返回的函数是decorator_four()，所以

后面的(f4)实际上等价于decorator_four(f4),然后由于decorator_four函数的返回值是wrapper,所以(*args,**kw)

等价于wrapper(*args,**kw)'''

@my_decorator_four('my_decorator_four')

def f4(s):

print(s)

f4('被装饰函数f4的参数')

# P5 课后习题，通过分析课后习题可以知道，需要设计一个装饰函数带可变参数，被装饰函数不带参数的装饰器

# 为了保持前后的统一性，函数名我就不用log了，而是用my_decorator_five代替log,f5代替f

def my_decorator_five(*args, **kw):

def decorator_five(f5):

print(*args)

print('begin call')

# 在函数中使用了被装饰函数，就不用再返回这个函数了，因为已经在这被执行过了

f5()

print('end call')

return decorator_five

@my_decorator_five()

def f5():

pass

@my_decorator_five('execute')

def f5():

pass

'''P5的装饰器函数调用逻辑实际上和P3类似，f5=my_decorator_five(*args,**kw)(f5),好P3的差别就是最后不用加()，因

为最后没有返回被装饰函数'''