python中装饰器是什么
这篇文章主要介绍了python中装饰器是什么,具有一定借鉴价值,需要的朋友可以参考下。希望大家阅读完这篇文章后大有收获。下面让小编带着大家一起了解一下。
合理使用装饰器可以简化开发,并且使得代码更加清晰。下面我们分别介绍两种装饰器,不带参数的装饰器和带参数的装饰器。
一、不带参数的装饰器
我们用一个实际的例子来引入装饰器的概念,比如我们现在有一个方法A(),然后我们需要在方法A()执行之前在终端打印"function is running",这时候我们可以在方法A()的开始部分加上下面的代码:
print("functionisrunning")
但是如果我们不想修改方法A()的代码,也可以重新写一个方法decorator(),然后把A()作为入参,代码如下:
defdecorator(fun):@functools.wraps(fun)defwrapper(*args,**kwargs):print("Methodisrunning")returnfun()returnwrapper
python中的方法可以作为参数传入另一个方法,所以当我们执行decorator(A)的时候,返回的是一个方法,这个方法的本质是先执行print语句,然后再执行A()。
但是这样我们的调用代码就需要进行修改,也就是说之前调用方法A(),现在的写法是调用decorator(A)(),为了不影响调用的代码,我们使用python的语法糖改造方法A()的代码。
@decoratordefA():#dosomething
实际上只是在A()的前面加上一行@decorator,这样在我们的调用代码就不需要改变了。
装饰器中的@functools.wraps(fun)是为了保证返回的方法__name__属性和入参方法fun的__name__属性相同。
二、带入参的装饰器
在上面不带参数的装饰器中,我们实现了在执行方法A()之前打印一行固定的字符串,如果我们想要打印不同的字符串就需要用带参数的装饰器。实际上就是在不带参数的装饰器上再封装一层即可,代码如下:
defdecorator_par(name):defdecorator(fun):@functools.wraps(fun)defwrapper(*args,**kwargs):print("Method{}isrunning".format(name))returnfun()returnwrapperreturndecorator
这样我们就可以在使用装饰器的时候设置参数name,实现打印不同的信息。完整的代码如下:
importfunctoolsdefdecorator_par(name):defdecorator(fun):@functools.wraps(fun)defwrapper(*args,**kwargs):print("Method{}isrunning".format(name))returnfun()returnwrapperreturndecorator@decorator_par("A")defA():print("IamA")@decorator_par("B")defB():print("IamB")A()B()
执行输出为:
MethodAisrunningIamAMethodBisrunningIamB
可以看到我们在不改变方法调用代码的情况下,实现了在该方法之前打印特定的信息。
感谢你能够认真阅读完这篇文章,希望小编分享python中装饰器是什么内容对大家有帮助,同时也希望大家多多支持亿速云,关注亿速云行业资讯频道,遇到问题就找亿速云,详细的解决方法等着你来学习!
声明:本站所有文章资源内容,如无特殊说明或标注,均为采集网络资源。如若本站内容侵犯了原著者的合法权益,可联系本站删除。