python装饰器的集中使用姿势

年少再无少年

在Python中，装饰器是一种十分强大并且好用的语法，一些重复的代码使用装饰器语法的话能够使代码更容易理解及阅读。
因此在这里简单总结了一下Python中装饰器的几种用法以及需要注意的事情。
一、在装饰器中获取被装饰函数的参数

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

1. def mydecorator(func):
2.     def wrapped(*args, **kwargs):
3.         print("进入装饰器")
4.         if args[0]['header'] == 'spider':
5.             print("code: 400")
6.             return
8.         result = func(*args, **kwargs)
9.         return result
11.     return wrapped
12. @mydecorator
13. def request_page(request):
14.     print("一个访问请求")
15.     print("返回了response")
16. if __name__ == '__main__':
17.     request = {
18.         'data': 100,
19.         'header': 'spider'
20.     }
21.     request_page(request)

复制代码

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

1. def mydecorator(*args, **kwargs):
2.     print("进入函数")
3.     if args[0]['header'] == 'spider':
4.         print("code: 400")
5.         return False
6.     return True
7. def request_page(request):
8.     if not mydecorator(request):
9.         return
10.     print("访问一个网页")
11.     print("得到了response")
12. if __name__ == '__main__':
13.     request = {
14.         'data': 100,
15.         'header': 'spider'
16.     }
17.     request_page(request)

复制代码

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

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

1. def mydecorator(func):
2.     def wrapped(*args, **kwargs):
3.         print("进入装饰器")
4.         result = func(*args, **kwargs)
5.         if result == 400:
6.             print("response is 400!")
7.             return False
8.         return True
9.     return wrapped
11. @mydecorator
12. def request_page():
13.     print("访问一个网页")
14.     print("得到了response")
15.     return 200
17. if __name__ == '__main__':
18.     print(request_page())

复制代码

三、给装饰器传入参数

在实际应用中，我们有时需要根据函数的执行状态来重复执行。例如在编写爬虫的时候，可能由于网络的原因会导致一些页面访问失败，这时我们就需要根据爬虫的返回结果进行重复请求。
[code]def retry(MAXRETRY=3):    def decorator(func):        def wrapped(*args, **kwargs):            print("进入装饰器")            result = 0            retry = 1            while result != 200 and retry