Python装饰器使用你可能不知道的几种姿势

前言

在Python中，装饰器是一种十分强大并且好用的语法，一些重复的代码使用装饰器语法的话能够使代码更容易理解及阅读。

因此在这里简单总结了一下Python中装饰器的几种用法以及需要注意的事情。

一、在装饰器中获取被装饰函数的参数

假设我们在开发web的时候，需要做反爬。要判断接口的访问来源我们就可以通过下面装饰器的方法来实现：

def mydecorator(func): def wrapped(*args, **kwargs):  print("进入装饰器")  if args[0]['header'] == 'spider':   print("code: 400")   return  result = func(*args, **kwargs)  return result return wrapped@mydecoratordef request_page(request): print("一个访问请求") print("返回了response")if __name__ == '__main__': request = {  'data': 100,  'header': 'spider' } request_page(request)

在这个装饰器中，我们在装饰器中获取了request中的header参数，如果判断访问来源于爬虫，那么便给它返回一个400。
使用装饰器的写法等同于下面不使用装饰器的写法

def mydecorator(*args, **kwargs): print("进入函数") if args[0]['header'] == 'spider':  print("code: 400")  return False return Truedef request_page(request): if not mydecorator(request):  return print("访问一个网页") print("得到了response")if __name__ == '__main__': request = {  'data': 100,  'header': 'spider' } request_page(request)

在只需要装饰一个函数的时候后面一种写法可能更优于装饰器的写法，但是在需要装饰很多个函数的时候，使用装饰器明显是更好的选择。

二、在装饰器获取函数的返回值

有的时候我们需要对函数的返回值做出判断，但又不想直接将判断写在函数里的时候，我们也可以使用装饰器来实现：

def mydecorator(func): def wrapped(*args, **kwargs):  print("进入装饰器")  result = func(*args, **kwargs)  if result == 400:   print("response is 400!")   return False  return True return wrapped@mydecoratordef request_page(): print("访问一个网页") print("得到了response") return 200if __name__ == '__main__': print(request_page())

三、给装饰器传入参数

在实际应用中，我们有时需要根据函数的执行状态来重复执行。例如在编写爬虫的时候，可能由于网络的原因会导致一些页面访问失败，这时我们就需要根据爬虫的返回结果进行重复请求。

def retry(MAXRETRY=3): def decorator(func):  def wrapped(*args, **kwargs):   print("进入装饰器")   result = 0   retry = 1   while result != 200 and retry <= MAXRETRY:    result = func(*args, **kwargs)    print("重试第%s次" % retry)    retry += 1   return result  return wrapped return decorator @retry(5)def request_page(): print("访问一个网页") print("得到了response") return 400

在这里我们假设访问一个网页得到400的时候便重新请求。我们在retry装饰器里传了一个5，这表示我们希望重试的最大次数为5次，如果不传入这个值，那么它的默认重试次数则为3次。

在熟悉了基本装饰器的写法后，传参装饰器的写法也十分的好理解了。就是在外面多加了一层函数，用于传入参数。