wrapt_timeout_decorator_basic_function_timeout_example.py

1import time
2from wrapt_timeout_decorator import timeout
3
4@timeout(5)
5def mytest(message):
6    print(message)
7    for i in range(1, 10):
8        time.sleep(1)
9        print('{} {}'.format(message, i))
10
11if __name__ == '__main__':
12    try:
13        mytest('test')
14    except TimeoutError:
15        print('function did not finish within 5 seconds')