函数重做的装饰器

函数重做的装饰器

def retry_times(attempt):
    def decorator(func):
        def wrapper(*args, **kw):
            att = 0
            while att < attempt:
                try:
                    return func(*args, **kw)
                except Exception as e:
                    print(e)
                    time.sleep(2)
                    att += 1
        return wrapper
    return decorator

@retry_times(3)
def make_trouble():
    '''Retry until succeed'''
    print ('retrying...')
    raise

if __name__ == '__main__':
    make_trouble()