python实现装饰器带参数_装饰器带参数

装饰器带参数

装饰器携带参数，可以实现更加强大的装饰器，本篇教程以http请求重试场景来讲解如何编写带参数的装饰器。

1. 重试场景

假设这样一个场景，你写了一个函数，用来从一个API接口获取数据，那么你必须考虑网络问题，也许网关解析会出错，也许API接口压力太大，短时间内响应超时，面对这种情况，你的函数应该具备重试功能。

请求失败以后，暂停一定时间，再次发起请求，重试的次数必须是可设置的。你可以把这个功能在函数中实现，但可以肯定，这不是一个好方法，原因在于每一个这样的函数你都需要实现一遍，而且不同的函数访问不同的API，暂停的时间以及重试的次数都有可能不同。

这种请求，就非常适合用装饰器来解决

2. 带参数装饰器

from functools import wraps

def retry(retry_count=5, sleep_time=1):

def wrapper(func):

@wraps(func)

def inner(*args, **kwargs):

for i in range(retry_count):

try:

res = func(*args, **kwargs)

return res

except:

time.sleep(sleep_time)

continue

return None

return inner

return wrapper

这段代码一共嵌套着定义了3个函数，在讲解作用域的时候强调过，嵌套作用域是一个相对概念，在函数inner里可以访问更上层的局部作用域里的数据，retry_count是函数retry的一个参数，在inner里根据retry_count来决定重试的次数，只要发生了异常，就会重试。

3. 如何使用

写爬虫时，你不得不考虑网络因素，一次请求可能无法得到所需要的响应数据，这时就需要重试

class HttpCodeException(Exception):

pass

@retry()

def get_html(url):

headers = {

'User-Agent': 'Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/73.0.3683.87 Mobile Safari/537.36',

'Host': 'movie.douban.com',

'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3',

'Referer': 'https://movie.douban.com/top250?start=25&filter='

}

res = requests.get(url, headers=headers)

if res.status_code != 200:

raise HttpCodeException

return res.text

这段代码是我编写过的一个爬虫的一小部分，在其他场景下，只要涉及到重试，都可以使用我写的这个装饰器。