59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
#!/usr/bin/env python
|
|
# vim:fileencoding=utf-8
|
|
# License: GPL v3 Copyright: 2016, Kovid Goyal <kovid at kovidgoyal.net>
|
|
|
|
from collections import namedtuple
|
|
from operator import itemgetter
|
|
from time import monotonic
|
|
|
|
from .utils import safe_print
|
|
|
|
Event = namedtuple('Event', 'at callback args')
|
|
get_at = itemgetter(0)
|
|
|
|
|
|
class Timers:
|
|
|
|
def __init__(self):
|
|
self.timers = []
|
|
|
|
def _add(self, delay, callback, args):
|
|
self.timers.append(Event(monotonic() + delay, callback, args))
|
|
self.timers.sort(key=get_at)
|
|
|
|
def add(self, delay, callback, *args):
|
|
self.remove(callback)
|
|
self._add(delay, callback, args)
|
|
|
|
def add_if_missing(self, delay, callback, *args):
|
|
for ev in self.timers:
|
|
if ev.callback == callback:
|
|
return
|
|
self._add(delay, callback, args)
|
|
|
|
def remove(self, callback):
|
|
for i, ev in enumerate(self.timers):
|
|
if ev.callback == callback:
|
|
break
|
|
else:
|
|
return
|
|
del self.timers[i]
|
|
|
|
def timeout(self):
|
|
if self.timers:
|
|
return max(0, self.timers[0][0] - monotonic())
|
|
|
|
def __call__(self):
|
|
if self.timers:
|
|
now = monotonic()
|
|
expired_timers, waiting_timers = [], []
|
|
for ev in self.timers:
|
|
(expired_timers if ev[0] <= now else waiting_timers).append(ev)
|
|
self.timers = waiting_timers
|
|
for ev in expired_timers:
|
|
try:
|
|
ev.callback(*ev.args)
|
|
except Exception:
|
|
import traceback
|
|
safe_print(traceback.format_exc())
|