Generic timeout decorator

timeout_decorator.py

import sys
import functools
import signal
import logging

def timeout(seconds: int):

    def decorator(function):

        def handle_timeout(signum, frame):
            logging.error("%s received %s after %i s", frame.f_code.co_name,
                         signal.Signals(signum).name, seconds)
            sys.tracebacklimit = 0
            raise TimeoutError("Execution exceeded {} seconds".format(seconds))

        @functools.wraps(function)
        def wrapper(*args, **kwargs):
            if seconds < 1:
                raise TimeoutError("Timeout immediately")
            signal.signal(signal.SIGALRM, handle_timeout)
            signal.alarm(seconds)
            try:
                return function(*args, **kwargs)
            finally:
                signal.alarm(0)

        return wrapper

    return decorator