fjsj / retry_on_exceptions

Decorator for retrying a function N times by catching one of the specified exceptions and then retrying

retry_on_exceptions decorator

Decorator for retrying a function N times by catching one of the specified exceptions and then retrying. Specially useful for functions that throws errors sporadically, like ones that depends on external resources as web APIs, databases, etc.

Installation:

pip install retry_on_exceptions

or through source code:

git clone git://github.com/fjsj/retry_on_exceptions.git
cd retry_on_exceptions
python setup.py install

Usage:

from retry import retry_on_exceptions

current_try = 0
@retry_on_exceptions(types=[ZeroDivisionError, KeyError], tries=3)
def test():
    global current_try
    current_try += 1
    if current_try == 1:
        return 1 / 0
    elif current_try == 2:
        return dict()['key']
    else:
        return "Got it on last try!"    

if __name__ == "__main__":
    print test()

The code above prints (if root logger is active and on debug level):

• Retrying function test
• Retrying function test
• Last try... and I will raise up whatever exception is raised
• Got it on last try!

Optionally, you can also specify a delay (a float in seconds), making the current thread sleep between tries:

@retry_on_exceptions(types=[urllib2.URLError], tries=3, delay=3.5)