python 进程超时自动退出 装饰器

思路
利用python的多线程一个线程控制主进程运行时间，时间到则利用线程os._exit()迫使主进程退出。另一个线程调用主进程函数执行脚本。

代码

from functools import wraps
import time
import os
from threading import Thread

def set_time_limit(t):
    def auto_quit(t1):
        '''此为控制进程超时退出的线程函数'''
        time.sleep(t1)
        print("time out {}".format(t1))
        os._exit(1) #此函数专门用于线程控制主进程退出，有兴趣的可以看一下和sys.exit()的区别
    def decorator(f):
        '''此函数用于传入被装饰函数f'''
        @wraps(f)
        def wrapper(*args,**kwargs):
            '''装饰器内部遵循的逻辑是：
            1.auto_quit先执行完，进程结束
            2.被修饰函数f先执行完，auto_quit函数停止执行
            3.被修饰函数执行完，下面的代码才能运行
            '''
            t1=Thread(target=auto_quit,args=(t,))  #此处的t是set_time_limit函数的形参，是auto_quit函数的实参
            t2=Thread(target=f,args=args,kwargs=kwargs)
            t1.setDaemon(True) #满足第2点
            t1.start()
            t2.start()
            t2.join() #满足第3点
        return wrapper
    return decorator

测试代码
设置超时时间为3s,主函数执行时间为2s

@set_time_limit(3)
def a(t):
    time.sleep(t)
    print("execuate finished")

if __name__=="__main__":
    a(2)
    print("执行完毕")

执行完结果
执行完成，auto_quit函数没有调用

测试代码
设置超时时间为2s，主函数执行时间为3s

@set_time_limit(2)
def a(t):
    time.sleep(t)
    print("execuate finished")

if __name__=="__main__":
    a(3)
    print("执行完毕")

执行结果
追进程超时退出

你们也可以把t1.setDaemon(True)注释掉（超时时间设长一些，执行时间短一些）执行一下看一下这样就会理解啥是守护线程
你会发现执行一直不结束，在等待auto_quit执行结束

你们也可以把t2.join()注释掉
会发现无论程序是否超时退出都先打印：执行完毕