openmedialibrary/oml/tasks.py

# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4


from queue import Queue
from threading import Thread

from websocket import trigger_event

import logging
logger = logging.getLogger(__name__)

class Tasks(Thread):

    def __init__(self):
        self.q = Queue()
        self.connected = True
        Thread.__init__(self)
        self.daemon = True
        self.start()
        self.queue('scan')

    def run(self):
        import item.scan
        from user.models import List
        while self.connected:
            m = self.q.get()
            if m:
                try:
                    action, data = m
                    if action == 'ping':
                        trigger_event('pong', data)
                    elif action == 'import':
                        item.scan.run_import(data)
                    elif action == 'export':
                        List.export(data)
                    elif action == 'scan':
                        item.scan.run_scan()
                    else:
                        trigger_event('error', {'error': 'unknown action'})
                except:
                    logger.debug('task failed', exc_info=1)
            self.q.task_done()

    def join(self):
        self.connected = False
        self.q.put(None)
        return Thread.join(self)

    def queue(self, action, data=None):
        self.q.put((action, data))