52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
# -*- 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))
|
|
|