Timeout decorator

gistfile1.py

import signal
 
class TimeOut(Exception):
    def __init__(self, msg='Time out error'):
        super(Exception, self).__init__(msg)
 
def on_alarm(signum, frame): # при надходженні сигналу кидати виняток
    raise TimeOut()
 
def limit_time(t, default=None):
    signal.signal(signal.SIGALRM, on_alarm) # встановити обробник сигналу
 
    def decorator(f): 
        def decorated(*args, **kwargs):
            signal.alarm(t) # попросити ОС прислати нам сигнал через t секунд
            try:
                f(*args, **kwargs)
            except TimeOut:
                return default
        return decorated
    return decorator
 
 
import time
 
@limit_time(5) # працювати не довше 5 секунд
def test():
    for i in range(10):
        print i
        time.sleep(1)
 
test()

Декоратор який використовуючи сигнал Unix SIGALRM перериває роботу функції якщо та працює довше ніж задане число секунд