setup process timeout

setup_proc_timeout.py

def setup_proc_timeout(timeout: float, basedir: str, prefix: str=''):
    '''
    setup process timeout;
    dump traceback from all threads to `basedir` if timeout.
    '''

    import os
    import datetime
    import atexit
    import faulthandler

    def _get_timestr(utc=False):
        if utc:
            dt = datetime.datetime.utcnow()
        else:
            dt = datetime.datetime.now()
        dtstr = dt.isoformat()[:19]
        return dtstr.replace(':', '-')

    name = f'{prefix}{_get_timestr()}.log'
    path = os.path.join(basedir, name)

    os.makedirs(basedir, exist_ok=True)
    fp = open(path, 'w')

    faulthandler.dump_traceback_later(timeout, file=fp, exit=True)

    @atexit.register
    def _():
        fp.close()
        os.unlink(path=fp.name)