Как реализовать таймер в Python?

Я хотел бы реализовать простой сторожевой таймер в Python с двумя вариантами использования:

  • Watchdog гарантирует, что функция не выполняется дольше, чем x секунд
  • Watchdog гарантирует, что определенная регулярно выполняемая функция действительно выполняет, по крайней мере, каждый y секунд

Как мне это сделать?

7
python watchdog

2 ответов

просто публикую свое собственное решение:

from threading import Timer

class Watchdog:
    def __init__(self, timeout, userHandler=None):  # timeout in seconds
        self.timeout = timeout
        self.handler = userHandler if userHandler is not None else self.defaultHandler
        self.timer = Timer(self.timeout, self.handler)
        self.timer.start()

    def reset(self):
        self.timer.cancel()
        self.timer = Timer(self.timeout, self.handler)
        self.timer.start()

    def stop(self):
        self.timer.cancel()

    def defaultHandler(self):
        raise self

использование, если вы хотите убедиться, что функция заканчивается менее чем x секунд:

watchdog = Watchdog(x)
try:
  # do something that might take too long
except Watchdog:
  # handle watchdog error
watchdog.stop()

использование, если вы регулярно выполняете что-то и хотите убедиться, что оно выполняется по крайней мере каждый y секунд:

import sys

def myHandler():
  print "Whoa! Watchdog expired. Holy heavens!"
  sys.exit()

watchdog = Watchdog(y, myHandler)

def doSomethingRegularly():
  # make sure you do not return in here or call watchdog.reset() before returning
  watchdog.reset()
8

signal.alarm() устанавливает тайм-аут для вашей программы, и вы можете вызвать его в своем основном цикле и установить его в больший из двух раз, когда вы готовы терпеть:

import signal
while True:
    signal.alarm(10)
    infloop()
1