openmedialibrary/oml/tasks.py

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

import os
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 item.models import sync_metadata
        from user.models import export_list, update_user_peering
        while self.connected:
            m = self.q.get()
            if m:
                try:
                    action, data = m
                    logger.debug('run task %s', action)
                    if action == 'ping':
                        trigger_event('pong', data)
                    elif action == 'import':
                        item.scan.run_import(data)
                    elif action == 'export':
                        export_list(data)
                    elif action == 'scan':
                        item.scan.run_scan()
                    elif action == 'scanimport':
                        item.scan.import_folder()
                    elif action == 'peering':
                        update_user_peering(*data)
                    elif action == 'syncmetadata':
                        if not data:
                            data = []
                        sync_metadata(*data)
                    else:
                        trigger_event('error', {'error': 'unknown action'})
                    logger.debug('finished task %s', 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))