try: import queue except ImportError: import Queue as queue import time import threading import logging class Task(object): """Task object for the Scheduler class""" def __init__(self, name, seconds, callback, args=None, kwargs=None, repeat=False, qpointer=None): self.name = name self.seconds = seconds self.callback = callback self.args = args or tuple() self.kwargs = kwargs or {} self.repeat = repeat self.next = time.time() + self.seconds self.qpointer = qpointer def run(self): if self.qpointer is not None: self.qpointer.put(('schedule', self.callback, self.args)) else: self.callback(*self.args, **self.kwargs) self.reset() return self.repeat def reset(self): self.next = time.time() + self.seconds class Scheduler(object): """Threaded scheduler that allows for updates mid-execution unlike http://docs.python.org/library/sched.html#module-sched""" def __init__(self, parentqueue=None): self.addq = queue.Queue() self.schedule = [] self.thread = None self.run = False self.parentqueue = parentqueue def process(self, threaded=True): if threaded: self.thread = threading.Thread(name='shedulerprocess', target=self._process) self.thread.start() else: self._process() def _process(self): self.run = True while self.run: try: wait = 1 updated = False if self.schedule: wait = self.schedule[0].next - time.time() try: if wait <= 0.0: newtask = self.addq.get(False) else: newtask = self.addq.get(True, wait) except queue.Empty: cleanup = [] for task in self.schedule: if time.time() >= task.next: updated = True if not task.run(): cleanup.append(task) else: break for task in cleanup: x = self.schedule.pop(self.schedule.index(task)) else: updated = True self.schedule.append(newtask) finally: if updated: self.schedule = sorted(self.schedule, key=lambda task: task.next) except KeyboardInterrupt: self.run = False logging.debug("Quitting Scheduler thread") if self.parentqueue is not None: self.parentqueue.put(('quit', None, None)) def add(self, name, seconds, callback, args=None, kwargs=None, repeat=False, qpointer=None): self.addq.put(Task(name, seconds, callback, args, kwargs, repeat, qpointer)) def quit(self): self.run = False