Protecting against SIGINT using ThreadPoolExecutor

protect.py

# Common imports
from time import sleep

# Synchronous imports
from concurrent.futures import ThreadPoolExecutor

# Asynchronous imports
# from gevent.threadpool import ThreadPoolExecutor


def protect(fn, *args, **kwargs):
    with ThreadPoolExecutor(max_workers=1) as pool:
        future = pool.submit(fn, *args, **kwargs)
        return future.result()


def ask(arg):
    for i in range(5):
        print('<debug>', i)
        sleep(1)
    return arg


def main():
    while 'n' not in input('Run ([y]/n)? ').lower():
        try:
            print(protect(ask, 'hello'))
        except KeyboardInterrupt:
            print('Interrupted')


if __name__ == '__main__':
    main()