Python自動重試HTTP連接裝飾器

字號:


    這篇文章主要介紹了Python自動重試HTTP連接裝飾器,有時候我們要去別的接口取數(shù)據(jù),可能因為網(wǎng)絡(luò)原因偶爾失敗,為了能自動重試,寫了這么一個裝飾器,可以實現(xiàn)自動重連2次,需要的朋友可以參考下
    有時候我們要去別的接口取數(shù)據(jù),可能因為網(wǎng)絡(luò)原因偶爾失敗,為了能自動重試,寫了這么一個裝飾器。
    這個是python2.7x 的版本,python3.x可以用 nonlocal 來重寫。
    #-*- coding: utf-8 -*-
    #all decorators in this tool file
    #author: orangleliu
    ############################################################
    #http連接有問題時候,自動重連
    def conn_try_again(function):
    RETRIES = 0
    #重試的次數(shù)
    count = {"num": RETRIES}
    def wrapped(*args, **kwargs):
    try:
    return function(*args, **kwargs)
    except Exception, err:
    if count['num'] < 2:
    count['num'] += 1
    return wrapped(*args, **kwargs)
    else:
    raise Exception(err)
    return wrapped
    用法很的簡單,下面是一個程序片段。
    @conn_try_again
    def post_query_bandwidth_for_bandwidth(self, contract_no, data_month, product_code):
    #根據(jù)webluker接口情況獲取計費數(shù)據(jù)
    try:
    post_data = {'contract':contract_no, 'month': data_month, 'code':product_code}
    params = urllib.urlencode(post_data)
    response = urllib2.urlopen(WEBLUKER_BANDWITH_API + "?" +params)
    billdata = {}
    billdata = response.read()
    if not billdata:
    billdata = {}
    return billdata
    except Exception, err:
    err = u'與webluker接口間通信異常'
    raise Exception(err)
    如果try塊中有異常,就會自動重試2次。