aboutsummaryrefslogtreecommitdiff
path: root/src/Timers.py
blob: d3a297df26f798e6f1c5f8f6b6ce08297b7efecb (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import time, uuid

class Timer(object):
    def __init__(self, id, name, delay, next_due, kwargs):
        self.id = id
        self.name = name
        self.delay = delay
        if next_due:
            self.next_due = next_due
        else:
            self.set_next_due()
        self.kwargs = kwargs
        self._done = False

    def set_next_due(self):
        self.next_due = time.time()+self.delay
    def due(self):
        return self.time_left() <= 0
    def time_left(self):
        return self.next_due-time.time()

    def redo(self):
        self._done = False
        self.set_next_due()
    def finish():
        self._done = True
    def done(self):
        return self._done

class Timers(object):
    def __init__(self, events, log):
        self.events = events
        self.log = log
        self.timers = []

    def setup(self, timers):
        for name, timer in timers:
            id = name.split("timer-", 1)[1]
            self._add(timer["name"], timer["delay"], timer[
                "next-due"], id, False, timer["kwargs"])

    def _persist(self, timer):
        self.set_setting("timer-%s" % timer.id, {
            "name": timer.name, "delay": timer.delay,
            "next-due": timer.next_due, "kwargs": timer.kwargs})
    def _remove(self, timer):
        self.timers.remove(timer)
        self.del_setting("timer-%s" % timer.id)

    def add(self, name, delay, next_due=None, **kwargs):
        self._add(name, delay, next_due, None, False, kwargs)
    def add_persistent(self, name, delay, next_due=None, **kwargs):
        self._add(name, delay, next_due, None, True, kwargs)
    def _add(self, name, delay, next_due, id, persist, kwargs):
        id = id or uuid.uuid4().hex
        timer = Timer(id, name, delay, next_due, kwargs)
        if persist:
            self._persist(timer)
        self.timers.append(timer)

    def next(self):
        times = filter(None, [timer.time_left() for timer in self.timers])
        if not times:
            return None
        return max(min(times), 0)

    def call(self):
        for timer in self.timers[:]:
            if timer.due():
                timer.finish()
                self.events.on("timer.%s" % timer.name, timer=timer)
                if timer.done():
                    self._remove(timer)