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