retry
偶然看到一篇文章,想到了前几天的一个需求,git pull性能不稳,需要加入重试机制,正好这个装饰器的实例符合这样的场景。
# coding:utf-8
import time
import logging
import socket
from functools import wraps
logging.basicConfig(level=logging.DEBUG)
def retry(retries=3, delay=1):
def wrapper(func):
@wraps(func)
def proxy(*args, **kwargs):
count = retries
error = None
while count > 0:
try:
return func(*args, **kwargs)
except Exception as e:
print("relay times: {}".format(count))
count -= 1
time.sleep(delay)
error = e
print(u"have retry {} times, but happen error: {}".format(count, error))
return error
return proxy
return wrapper
@retry(2, 3)
def check():
sk = socket.socket()
sk.settimeout(5)
sk.connect(('x.x.x.x', 80))
'''
# 上面写法属于简写方式,等价于下面的处理
maketry=retry(2,3)
@maketry
def check():
pass
'''
if __name__ == "__main__":
check()
print(check.__name__)
参考:
CSDN-郭 璞偶遇不错的装饰器:本文主要参考,但是该文示例有点小 bug,进一步优化了一下
Python之旅-装饰器:复习了一下带参数装饰器的写法,本文就属于带参数的装饰器的用法