介绍

使用

一般使用

# 首先,初始化一个 TimeLimit 对象,并传入最高用时(毫秒)
limit = TimeLimit(1000) # 1 秒
# 然后,使用 with 语句来限制任务的执行时间
with limit:
 try:
 # 在任务中插入一些检查点
 limit.checkpoint() # 不触发
 except TimeoutError:
 print("任务超时!")
 # 在这里处理超时后的逻辑
 # 在这里编写需要限制时间的代码
 time.sleep(1.5) # 模拟耗时操作
 try: # 在任务中插入一些检查点
 limit.checkpoint() # 触发
 except TimeoutError:
 print("任务超时!")
 # 在这里处理超时后的逻辑

自定义处理程序

# 定义一个自定义的处理程序
def custom_handler():
 print("任务超时!")
 # 在这里处理超时后的逻辑
# 初始化一个 TimeLimit 对象,并传入最高用时(毫秒)和自定义处理程序
limit = TimeLimit(1000) # 1 秒
limit.set_custom_handler(custom_handler) # 在初始化后设置自定义处理程序
# 然后,使用 with 语句来限制任务的执行时间
with limit:
 # 在任务中插入一些检查点
 limit.checkpoint() # 不触发
 # 在这里编写需要限制时间的代码
 time.sleep(1.5) # 模拟耗时操作
 limit.checkpoint() # 触发