python装饰器使用实例分析
这篇“python装饰器使用实例分析”文章的知识点大部分人都不太理解,所以小编给大家总结了以下内容,内容详细,步骤清晰,具有一定的借鉴价值,希望大家阅读完这篇文章能有所收获,下面我们一起来看看这篇“python装饰器使用实例分析”文章吧。
一、装饰器使用场景经常用于有切面需求的场景,比如:插入日志、性能测试、事务处理、缓存、权限校验等场景。装饰器是解决这类问题的绝佳设计,有了装饰器,我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。
概括的讲,装饰器的作用就是为已经存在的对象添加额外的功能。
二、为什么需要装饰器1、先来看一个简单例子:
deffoo():print('iamfoo')2、增加需求
现在有一个新的需求,希望可以记录下函数的执行日志,于是在代码中添加日志代码:
deffoo():print('iamfoo')print("fooisrunning")3、又有需求
假设现在有100个函数需要增加这个需求,并且后续可能还要对这一百个函数都增加执行前打印日志的需求,怎么办?还一个个改吗?
当然不了,这样会造成大量雷同的代码,为了减少重复写代码,我们可以这样做,重新定义一个函数:专门处理日志 ,日志处理完之后再执行真正的业务代码。
defuse_logging(func):print("%sisrunning"%func.__name__)func()defbar():print('iambar')use_logging(bar)运行结果:#barisrunning#iambar
函数use_logging就是装饰器,它把执行真正业务方法的func包裹在函数里面,看起来像bar被use_logging装饰了。在这个例子中,函数进入和退出时 ,被称为一个横切面(Aspect),这种编程方式被称为面向切面的编程(Aspect-Oriented Programming)。
通过以上use_logging函数我们增加了日志功能,不管以后有多少函数需要增加日志或者修改日志的格式我们只需要修改use_logging函数,并执行use_logging(被装饰的函数)就达到了我们想要的效果。
defuse_logging(func):print("%sisrunning"%func.__name__)returnfunc@use_loggingdefbar():print('iambar')bar()三、基础装饰器入门
1、装饰器语法糖
python提供了@符号作为装饰器的语法糖,使我们更方便的应用装饰函数;但使用语法糖要求装饰函数必须return一个函数对象。因此我们将上面的func函数使用内嵌函数包裹并return。
装饰器相当于执行了装饰函数use_loggin后又返回被装饰函数bar,因此bar()被调用的时候相当于执行了两个函数。等价于use_logging(bar)()
defuse_logging(func):def_deco():print("%sisrunning"%func.__name__)func()return_deco@use_loggingdefbar():print('iambar')bar()2、对带参数的函数进行装饰
现在我们的参数需要传入两个参数并计算值,因此我们需要对内层函数进行改动传入我们的两个参数a和b,等价于use_logging(bar)(1,2)
defuse_logging(func):def_deco(a,b):print("%sisrunning"%func.__name__)func(a,b)return_deco@use_loggingdefbar(a,b):print('iambar:%s'%(a+b))bar(1,2)
我们装饰的函数可能参数的个数和类型都不一样,每一次我们都需要对装饰器做修改吗?这样做当然是不科学的,因此我们使用python的变长参数*args和**kwargs来解决我们的参数问题。
3、函数参数数量不确定不带参数装饰器版本,这个格式适用于不带参数的装饰器。
经过以下修改,我们已经适应了各种长度和类型的参数。这个版本的装饰器可以装饰任意类型的无参数函数。
defuse_logging(func):def_deco(*args,**kwargs):print("%sisrunning"%func.__name__)func(*args,**kwargs)return_deco@use_loggingdefbar(a,b):print('iambar:%s'%(a+b))@use_loggingdeffoo(a,b,c):print('iambar:%s'%(a+b+c))bar(1,2)foo(1,2,3)4、装饰器带参数
带参数的装饰器,这个格式适用于带参数的装饰器。
某些情况我们需要让装饰器带上参数,那就需要编写一个返回一个装饰器的高阶函数,写出来会更复杂。比如:
#!/usr/bin/envpython#-*-coding:utf-8-*-#__author__="TKQ"defuse_logging(level):def_deco(func):def__deco(*args,**kwargs):iflevel=="warn":print"%sisrunning"%func.__name__returnfunc(*args,**kwargs)return__decoreturn_deco@use_logging(level="warn")defbar(a,b):print('iambar:%s'%(a+b))bar(1,3)#等价于use_logging(level="warn")(bar)(1,3)5、functools.wraps
使用装饰器极大地复用了代码,但是他有一个缺点就是原函数的元信息不见了,比如函数的docstring、__name__、参数列表,先看例子:
defuse_logging(func):def_deco(*args,**kwargs):print("%sisrunning"%func.__name__)func(*args,**kwargs)return_deco@use_loggingdefbar():print('iambar')print(bar.__name__)bar()#barisrunning#iambar#_deco#函数名变为_deco而不是bar,这个情况在使用反射的特性的时候就会造成问题。因此引入了functools.wraps解决这个问题。
使用functools.wraps:
importfunctoolsdefuse_logging(func):@functools.wraps(func)def_deco(*args,**kwargs):print("%sisrunning"%func.__name__)func(*args,**kwargs)return_deco@use_loggingdefbar():print('iambar')print(bar.__name__)bar()#result:#barisrunning#iambar#bar,这个结果是我们想要的。OK啦!6、实现带参数和不带参数的装饰器自适应
importfunctoolsdefuse_logging(arg):ifcallable(arg):#判断参入的参数是否是函数,不带参数的装饰器调用这个分支@functools.wraps(arg)def_deco(*args,**kwargs):print("%sisrunning"%arg.__name__)arg(*args,**kwargs)return_decoelse:#带参数的装饰器调用这个分支def_deco(func):@functools.wraps(func)def__deco(*args,**kwargs):ifarg=="warn":print"warn%sisrunning"%func.__name__returnfunc(*args,**kwargs)return__decoreturn_deco@use_logging("warn")#@use_loggingdefbar():print('iambar')print(bar.__name__)bar()三、类装饰器
使用类装饰器可以实现带参数装饰器的效果,但实现的更加优雅简洁,而且可以通过继承来灵活的扩展.
1、类装饰器classloging(object):def__init__(self,level="warn"):self.level=leveldef__call__(self,func):@functools.wraps(func)def_deco(*args,**kwargs):ifself.level=="warn":self.notify(func)returnfunc(*args,**kwargs)return_decodefnotify(self,func):#logit只打日志,不做别的print"%sisrunning"%func.__name__@loging(level="warn")#执行__call__方法defbar(a,b):print('iambar:%s'%(a+b))bar(1,3)2、继承扩展类装饰器
classemail_loging(Loging):'''一个loging的实现版本,可以在函数调用时发送email给管理员'''def__init__(self,email='admin@myproject.com',*args,**kwargs):self.email=emailsuper(email_loging,self).__init__(*args,**kwargs)defnotify(self,func):#发送一封email到self.emailprint"%sisrunning"%func.__name__print"sendingemailto%s"%self.email@email_loging(level="warn")defbar(a,b):print('iambar:%s'%(a+b))bar(1,3)
以上就是关于“python装饰器使用实例分析”这篇文章的内容,相信大家都有了一定的了解,希望小编分享的内容对大家有帮助,若想了解更多相关的知识内容,请关注亿速云行业资讯频道。
声明:本站所有文章资源内容,如无特殊说明或标注,均为采集网络资源。如若本站内容侵犯了原著者的合法权益,可联系本站删除。