SickGear/sickbeard/event_queue.py

50 lines
1.1 KiB
Python

from lib.six import moves
import threading
class Event:
def __init__(self, etype):
self._type = etype
@property
def type(self):
return self._type
class Events(threading.Thread):
def __init__(self, callback):
super(Events, self).__init__()
self.queue = moves.queue.Queue()
self.daemon = True
self.callback = callback
self.name = 'EVENT-QUEUE'
self._stop = threading.Event()
def put(self, etype):
self.queue.put(etype)
def stop(self):
self._stop.set()
def run(self):
while not self._stop.is_set():
try:
# get event type
etype = self.queue.get(True, 1)
# perform callback if we got a event type
self.callback(etype)
# event completed
self.queue.task_done()
except moves.queue.Empty:
pass
# exiting thread
self._stop.clear()
# System Events
class SystemEvent(Event):
RESTART = 'RESTART'
SHUTDOWN = 'SHUTDOWN'