python装饰器的实现
2019/11/05 15:13 分类： 技术交流 浏览：1
饰器的一个特点是：在执行原函数的基础上，增加内容。
❓ 那么，如何执行原函数那？
✔ 把原函数的名字交给装饰函数，装饰函数就具备了执行老函数的能力。
# 旧函数，待装饰的函数
def fun():
print('我是老函数')
def factory(old_fun):
print('='*20)
old_fun()
factory(fun)
❓ 上面的装饰行为还有哪些问题？
◇ 需要通过装饰工厂来调用原函数，原来调用这个功能的业务部门，将大面积修改代码；
❓ 那么如何改进这个问题那？
如何设计一个方式，在不改变原函数名称的情况下，调用时也执行装饰工厂那。
◇ old_name = factory(old_name)
def fun(): print("我是fun") def factory(a): def tmp(): print("="*10) a() print("="*10) return tmp fun = factory(fun) fun()
old_name = factory(old_name)这句话就是生成装饰器的核心语句。为了避免每次都写这样的无脑代码，python提供了一个便捷的写法，我们称之为语法糖。
· 找到要装饰的函数
· 在上面写上@，然后跟装饰器函数的名称
@decortate
print('in fun() ...')
python的装饰器写法思路：
1.定义一个中介函数，该函数命名为装饰行为，接收一个可调用对象
def get_timer(fn):
2.在中介函数内部实现装饰行为函数
def wrapper():
... ... # 装饰代码实现
3.在中介函数最后返回装饰行为的函数名：
return wrapper
【注意】：返回的是函数名，而不是函数调用
4.在需要被装饰的函数前，定义@装饰器名称
@get_timer
Demo1: 增加函数什么时候运行的功能:
import time
def time_fun(func):
def my_time():
print(f"{func.__name__} running at {time.ctime()}")
return my_time
@time_fun
def fun1():
print("+++++++++")
time.sleep(2)
def my_time(arg1): # 内部函数里预留接口
func(arg1) # 原函数传递参数
def fun1(arg1):
print("+++++{}++++".format(arg1))
fun1(10)
原函数有返回值，内部装饰函数也接收返回值进行返回。
def my_time(*args, **kwargs):
func(*args, **kwargs)
def fun1(arg1, arg2, name):
print("+++++{},{},{}++++".format(arg1, arg2, name))
fun1(10, 20, name='rocky')
fun1(10, 20, name='jim')
在装饰器中，我们也有一种需求，根据传入的参数不同，装饰的行为方式也会不同，那么如何定义这种装饰器那。
首先再理解装饰器的语法：
原材料名称= 装饰器名称(原材料名称)
@time_fun(“itsource”)
fun1 = time_fun(“itsource”)(fun1)
这样来看，必须要保证time_fun(“itsource”)返回的内容恰好是标准的装饰器就可以满足了。
def time_fun(flags):
def time_arg(func):
print("the flags is {}".format(flags))
return func(*args, **kwargs)
return time_arg
@time_fun("itsource")
❓ 可执行对象究竟是什么？
✔ 具有__call__方法的对象空间，就称之为可执行对象，每定义一个函数，就相当于在空间中定义了__call__方法。
❓ 如何利用类来实现装饰器
@Test # fun = Test(fun)
class Test():
def __init__(self, fn):
self.old_fun = fn
def __call__(self,):
... ... # 装饰内容
ret = self.old_fun()
def fun(a, b, name):
在实际开发时，一个原函数，可以使用多个装饰函数进行修饰，这种行为就称之为多层装饰。
对于多层装饰，要理解他的执行过程。
def bold(fn):
return f'<b>{fn()}</b>'
def italic(fn):
return f'<i>{fn()}</i>'
def hello():
return 'Hello World'
print(hello())
<i><b>Hello World</b></i>