Python装饰器使用你可能不知道的几种姿势
前言
在Python中，装饰器是一种十分强大并且好用的语法，一些重复的代码使用装饰器语法的话能够使代码更容易理解及阅读。
因此在这里简单总结了一下Python中装饰器的几种用法以及需要注意的事情。
一、在装饰器中获取被装饰函数的参数
假设我们在开发web的时候，需要做反爬。要判断接口的访问来源我们就可以通过下面装饰器的方法来实现：
defmydecorator(func):
defwrapped(*args,**kwargs):
print("进入装饰器")
ifargs[0]['header']=='spider':
print("code:400")
return
result=func(*args,**kwargs)
returnresult
returnwrapped
@mydecorator
defrequest_page(request):
print("一个访问请求")
print("返回了response")
if__name__=='__main__':
request={
'data':100,
'header':'spider'
}
request_page(request)
在这个装饰器中，我们在装饰器中获取了request中的header参数，如果判断访问来源于爬虫，那么便给它返回一个400。
使用装饰器的写法等同于下面不使用装饰器的写法
defmydecorator(*args,**kwargs):
print("进入函数")
ifargs[0]['header']=='spider':
print("code:400")
returnFalse
returnTrue
defrequest_page(request):
ifnotmydecorator(request):
return
print("访问一个网页")
print("得到了response")
if__name__=='__main__':
request={
'data':100,
'header':'spider'
}
request_page(request)
在只需要装饰一个函数的时候后面一种写法可能更优于装饰器的写法，但是在需要装饰很多个函数的时候，使用装饰器明显是更好的选择。
二、在装饰器获取函数的返回值
有的时候我们需要对函数的返回值做出判断，但又不想直接将判断写在函数里的时候，我们也可以使用装饰器来实现：
defmydecorator(func):
defwrapped(*args,**kwargs):
print("进入装饰器")
result=func(*args,**kwargs)
ifresult==400:
print("responseis400!")
returnFalse
returnTrue
returnwrapped
@mydecorator
defrequest_page():
print("访问一个网页")
print("得到了response")
return200
if__name__=='__main__':
print(request_page())
三、给装饰器传入参数
在实际应用中，我们有时需要根据函数的执行状态来重复执行。例如在编写爬虫的时候，可能由于网络的原因会导致一些页面访问失败，这时我们就需要根据爬虫的返回结果进行重复请求。
defretry(MAXRETRY=3):
defdecorator(func):
defwrapped(*args,**kwargs):
print("进入装饰器")
result=0
retry=1
whileresult!=200andretry<=MAXRETRY:
result=func(*args,**kwargs)
print("重试第%s次"%retry)
retry+=1
returnresult
returnwrapped
returndecorator
@retry(5)
defrequest_page():
print("访问一个网页")
print("得到了response")
return400
在这里我们假设访问一个网页得到400的时候便重新请求。我们在retry装饰器里传了一个5，这表示我们希望重试的最大次数为5次，如果不传入这个值，那么它的默认重试次数则为3次。
在熟悉了基本装饰器的写法后，传参装饰器的写法也十分的好理解了。就是在外面多加了一层函数，用于传入参数。
四、装饰器文档的问题
我们都知道通过魔术方法__doc__可以获取我们写在代码中的文档，那么你是否知道使用装饰器后，会造成被包装函数的文档被装饰器的文档覆盖的问题呢。
defrequest_page():
'''
request_page函数文档
:return:
'''
print("访问一个网页")
print("得到了response")
if__name__=='__main__':
print(request_page.__doc__)
在上面对上面未使用装饰的代码使用__doc__方法的时候，我们得到的结果是：
In[3]:request_page.__doc__ Out[3]:'\nrequest_page函数文档\n:return:\n'
这是我们理想中的结果！
但是当我们将上述函数使用装饰器装饰后：
defdecorator(func):
defwrapped(*args,**kwargs):
'''
装饰器文档
:paramargs:
:paramkwargs:
:return:
'''
print("进入装饰器")
result=func(*args,**kwargs)
returnresult
returnwrapped
@decorator
defrequest_page():
'''
request_page函数文档
:return:
'''
print("访问一个网页")
print("得到了response")
我们再一次运行__doc__魔术方法的时候，得到的结果却是装饰器的内部文档:
In[4]:request_page.__doc__ Out[4]:'\n装饰器文档\n:paramargs:\n:paramkwargs:\n:return:\n' In[5]:request_page.__name__ Out[5]:'wrapped'
这个问题会使得我们的调试变得困难，也会使许多自动文档生成工具失去效果。
解决这个问题的最好办法就是使用functools包的wraps()模块来将装饰器进行一个包装。
fromfunctoolsimportwraps
defdecorator(func):
@wraps(func)
defwrapped(*args,**kwargs):
'''
装饰器
:paramargs:
:paramkwargs:
:return:
'''
print("进入装饰器")
result=func(*args,**kwargs)
returnresult
returnwrapped
@decorator
defrequest_page():
'''
request_page函数文档
:return:
'''
print("访问一个网页")
print("得到了response")
使用wraps将装饰器装饰后，这样我们的函数便能够保存它的一些重要数据了。
In[3]:request_page.__doc__ Out[3]:'\nrequest_page函数文档\n:return:\n' In[3]:request_page.__name__ Out[4]:'request_page'
五、使用class的写法来编写装饰器
虽然大多数的装饰器都是通过函数的写法来实现的，但同样的可以通过类的写法来实现装饰器。
使用类的写法，我们可以实现一些使用函数写法不太好实现的需求。例如记录一个函数执行的次数
classDecorator():
def__init__(self,func):
print('类初始化')
self.func=func
self.count=0
def__call__(self,*args,**kwargs):
print('进入装饰器')
result=self.func(*args,**kwargs)
self.count+=1
returnresult
@Decorator
defrequest_page():
'''
request_page
:return:
'''
print("访问一个网页")
print("得到了response")
六、总结
装饰器是Python里比较高级的一种语法，这里只是介绍了它的几种使用技巧，以及需要注意的问题。借用金庸先生的话，“武功无高低，修为有深浅”。想要更加灵活的使用装饰器，深入理解它的原理，我们在平时还是需要加强基本功的学习！
好了，以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，谢谢大家对毛票票的支持。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。