在Python中,装饰器是一种十分强大并且好用的语法,一些重复的代码使用装饰器语法的话能够使代码更容易理解及阅读。
因此在这里简单总结了一下Python中装饰器的几种用法以及需要注意的事情。
假设我们在开发网络的时候,需要做反爬。要判断接口的访问来源我们就可以通过下面装饰器的方法来实现:
def mydecorator(函数):
def包装(* args, * * kwargs):
打印(“进入装饰器”)
如果参数[0]['标题')==爸┲搿?
print(代码:400)
返回
结果=func (* args, * * kwargs)
返回结果
返回包裹
@mydecorator
def request_page(请求):
打印(“一个访问请求”)
打印(“返回了响应”)
if __name__==癬_main__”:
请求={
“数据”:100年,
“头”:“蜘蛛”
}
request_page(请求)
在这个装饰器中,我们在装饰器中获取中了请求的头参数,如果判断访问来源于爬虫,那么便给它返回一个400。
使用装饰器的写法等同于下面不使用装饰器的写法
def mydecorator (* args, * * kwargs):
打印(“进入函数”)
如果参数[0]['标题')==爸┲搿?
print(代码:400)
返回假
还真
def request_page(请求):
如果不是mydecorator(请求):
返回
打印(“访问一个网页”)
打印(“得到反应了”)
if __name__==癬_main__”:
请求={
“数据”:100年,
“头”:“蜘蛛”
}
request_page(请求)
在只需要装饰一个函数的时候后面一种写法可能更优于装饰器的写法,但是在需要装饰很多个函数的时候,使用装饰器明显是更好的选择。
有的时候我们需要对函数的返回值做出判断,但又不想直接将判断写在函数里的时候,我们也可以使用装饰器来实现:
def mydecorator(函数):
def包装(* args, * * kwargs):
打印(“进入装饰器”)
结果=func (* args, * * kwargs)
如果结果==400:
打印(“响应400 !”)
返回假
还真
返回包裹
@mydecorator
def request_page ():
打印(“访问一个网页”)
打印(“得到反应了”)
返回200
if __name__==癬_main__”:
print (request_page ())
在实际应用中,我们有时需要根据函数的执行状态来重复执行。例如在编写爬虫的时候,可能由于网络的原因会导致一些页面访问失败,这时我们就需要根据爬虫的返回结果进行重复请求。
def重试(MAXRETRY=3):
def装饰(函数):
def包装(* args, * * kwargs):
打印(“进入装饰器”)
结果=0
重试=1
虽然结果!=200和重试& lt;=MAXRETRY:
结果=func (* args, * * kwargs)
打印(“重试第% s次“%重试)
重试+=1
返回结果
返回包裹
返回装饰
@retry (5)
def request_page ():
打印(“访问一个网页”)
打印(“得到反应了”)
返回400
在这里我们假设访问一个网页得到400的时候便重新请求。我们在重试装饰器里传了一个5,这表示我们希望重试的最大次数为5次,如果不传入这个值,那么它的默认重试次数则为3次。
在熟悉了基本装饰器的写法后,传参装饰器的写法也十分的好理解了。就是在外面多加了一层函数,用于传入参数。
我们都知道通过魔术方法__doc__可以获取我们写在代码中的文档,那么你是否知道使用装饰器后,会造成被包装函数的文档被装饰器的文档覆盖的问题呢。
def request_page ():
“‘
request_page函数文档
返回:
“‘
打印(“访问一个网页”)
打印(“得到反应了”)
if __name__==癬_main__”:
打印(request_page.__doc__)
在上面对上面未使用装饰的代码使用__doc__方法的时候,我们得到的结果是:
在[3]:request_page.__doc__
出[3]:“\ n request_page函数文档\ n:返回:\ n”
