热搜：编程 Python Microsoft 编程语言 C++

python重试装饰器示例

发布时间：2024-04-26 09:31:52 来源：网络

.

生活没有真正的完美，只有不完美才是最真实的美；生活没有一帆风顺的，只有披荆斩棘才能路路顺；生活没有永远的成功，只有在挫折中站起才是真正的成功，只有闪光的人生才算是生命的永恒。

利用python 写一些网络服务的时候，当网络状况不好，或者资源占用过多，任务拥塞的情况下，总会抛出一些异常，当前任务就被终止了，可以很好的利用@装饰器，写一个重试的装饰器，这样比较python！
执行结果：


WARNING:root:timed out, Retrying in 3 seconds...
WARNING:root:timed out, Retrying in 6 seconds...
WARNING:root:timed out, Retrying in 12 seconds...


#!/usr/bin/env python
# -*- coding: utf-8 -*- 
# tanyewei@gmail.com
# 2014/01/27 10:36
import time
import logging
import socket
from functools import wraps
logging.basicConfig(level=logging.DEBUG)
def retry(MyException, tries=4, delay=3, backoff=2, logger=None):
 def deco_retry(f):
 @wraps(f)
 def f_retry(*args, **kwargs):
 mtries, mdelay = tries, delay
 while mtries > 1:
 try:
 return f(*args, **kwargs)
 except MyException as ex:
 msg = "%s, Retrying in %d seconds..." % (str(ex), mdelay)
 if logger:
 logger.warning(msg)
 else:
 print msg
 time.sleep(mdelay)
 mtries -= 1
 mdelay *= backoff
 return str(ex)
 return f_retry
 return deco_retry
@retry(Exception, logger=logging)
def check():
 sk = socket.socket()
 sk.settimeout(5)
 sk.connect(('6.6.6.6', 80))
if __name__ == "__main__":
 check()