源码时代python干货分享| python装饰的神奇魅力初体验
1.1.1.一个最简单的装饰行为
装饰器的一个特点是:在执行原函数的基础上,增加内容。
咱们举个简单的例子:
我们可以把它用作日志打印器,在执行函数前,会有礼貌的告知你一下
“亲爱的coder,我要执行了”
在执行结束后,也会告知各位coder。
是不是很有趣?
那么,咱们如何执行原函数那?
? 把原函数的名字交给装饰函数,装饰函数就具备了执行老函数的能力。
旧函数,待装饰的函数
def fun(): print(‘我是老函数‘) def factory(old_fun): print(‘=‘*20) old_fun() print(‘=‘*20) factory(fun)
1.1.2.完善装饰行为
? 上面的装饰行为还有哪些问题?
◇ 需要通过装饰工厂来调用原函数,原来调用这个功能的业务部门,将大面积修改代码;
? 那么如何改进这个问题那?
如何设计一个方式,在不改变原函数名称的情况下,调用时也执行装饰工厂那。
◇ 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()
1.1.3.python的装饰器语法糖
old_name = factory(old_name)这句话就是生成装饰器的核心语句。为了避免每次都写这样的无脑代码,python提供了一个便捷的写法,我们称之为语法糖。
语法糖的写法:
· 找到要装饰的函数
· 在上面写上@,然后跟装饰器函数的名称
1.2.python装饰器的几种形式
1.2.1.无参数装饰器
Demo1: 增加函数什么时候运行的功能:
import time def time_fun(func): def my_time(): print(f"{func.__name__} running at {time.ctime()}") func() return my_time @time_fun def fun1(): print("+++++++++") fun1() time.sleep(2) fun1()
1.2.2.被装饰函数有参数
?装饰器的核心是内部函数的行为被重命名;
?内部函数的形式要满足原函数形式;
import time def time_fun(func): def my_time(arg1): # 内部函数里预留接口 print(f"{func.__name__} running at {time.ctime()}") func(arg1) # 原函数传递参数 return my_time @time_fun def fun1(arg1): print("+++++{}++++".format(arg1)) fun1(10) time.sleep(2) fun1(10)
1.2.3.被装饰函数有不定参数和返回值
原函数有返回值,内部装饰函数也接收返回值进行返回。
import time
def time_fun(func): def my_time(*args, **kwargs): print(f"{func.__name__} running at {time.ctime()}") func(*args, **kwargs) return my_time @time_fun def fun1(arg1, arg2, name): print("+++++{},{},{}++++".format(arg1, arg2, name)) fun1(10, 20, name=‘rocky‘) time.sleep(2) fun1(10, 20, name=‘jim‘)
1.2.4.改变装饰器行为(给装饰器传参数)
在装饰器中,我们也有一种需求,根据传入的参数不同,装饰的行为方式也会不同,那么如何定义这种装饰器那。
首先再理解装饰器的语法:
@time_fun def fun1(arg1, arg2, name): print("+++++{},{},{}++++".format(arg1, arg2, name)) 原材料名称= 装饰器名称(原材料名称) @time_fun(“itsource”) def fun1(arg1, arg2, name): print("+++++{},{},{}++++".format(arg1, arg2, name)) fun1 = time_fun(“itsource”)(fun1)
这样来看,必须要保证time_fun(“itsource”)返回的内容恰好是标准的装饰器就可以满足了。
import time
def time_fun(flags): def time_arg(func): def my_time(*args, **kwargs): print(f"{func.__name__} running at {time.ctime()}") print("the flags is {}".format(flags)) return func(*args, **kwargs) return my_time return time_arg @time_fun("itsource") def fun1(arg1, arg2, name): print("+++++{},{},{}++++".format(arg1, arg2, name)) fun1(10, 20, name=‘rocky‘) time.sleep(2) fun1(10, 20, name=‘jim‘)
1.2.5.用类来定义装饰器(了解)
? 可执行对象究竟是什么?
? 具有call方法的对象空间,就称之为可执行对象,每定义一个函数,就相当于在空间中定义了call方法。
? 如何利用类来实现装饰器
@Test # fun = Test(fun)
def fun():
pass
?Test的init方法应该接收被装饰函数的名称
?调用fun()实际就是调用了Test类的call方法
class Test():
def init(self, fn):
self.old_fun = fn
def __call__(self,): ... ... # 装饰内容 ret = self.old_fun()
@Test
def fun(a, b, name):
pass
fun()
1.3.多层装饰
在实际开发时,一个原函数,可以使用多个装饰函数进行修饰,这种行为就称之为多层装饰。
对于多层装饰,要理解他的执行过程。
def bold(fn): def wrapper(): return f‘<b>{fn()}</b>‘ return wrapper def italic(fn): def wrapper(): return f‘<i>{fn()}</i>‘ return wrapper @italic @bold def hello(): return ‘Hello World‘ print(hello())
【结果】:
<i><b>Hello World</b></i>
1.4.总结
?装饰器实际也是一个函数(可调用对象);
?接收一个老函数的名称;
?返回新函数的名称;
?在新函数里调用老函数;