python decorator example

decorator.py

def timeout_function(timeout, force_clean_func):
    def decorator_timeout_function(func):
        @functools.wraps(func)
        def run_func_in_thread(*args, **kwargs):
            thread = threading.Thread(target=func, args=args, kwargs=kwargs)
            thread.start()
            thread.join(timeout)
            if thread.is_alive():
                print("timeout, begin to call force_clean_func")
                force_clean_func()
                raise TimeoutException()
        return run_func_in_thread
    return decorator_timeout_function