1
0
mirror of https://github.com/spl0k/supysonic.git synced 2024-12-23 01:16:18 +00:00

File removing handled the same way as scanning

This commit is contained in:
spl0k 2014-08-31 12:12:36 +02:00
parent deba8aeee4
commit 4ec9aa11f2

View File

@ -29,6 +29,10 @@ from watchdog.events import PatternMatchingEventHandler
from supysonic import config, db from supysonic import config, db
from supysonic.scanner import Scanner from supysonic.scanner import Scanner
OP_SCAN = 1
OP_REMOVE = 2
OP_MOVE = 3
class SupysonicWatcherEventHandler(PatternMatchingEventHandler): class SupysonicWatcherEventHandler(PatternMatchingEventHandler):
def __init__(self, queue, logger): def __init__(self, queue, logger):
extensions = config.get('base', 'scanner_extensions') extensions = config.get('base', 'scanner_extensions')
@ -40,27 +44,51 @@ class SupysonicWatcherEventHandler(PatternMatchingEventHandler):
def on_created(self, event): def on_created(self, event):
self.__logger.debug("File created: '%s'", event.src_path) self.__logger.debug("File created: '%s'", event.src_path)
self.__queue.put(event.src_path) self.__queue.put(event.src_path, OP_SCAN)
def on_deleted(self, event): def on_deleted(self, event):
self.__logger.debug("File deleted: '%s'", event.src_path) self.__logger.debug("File deleted: '%s'", event.src_path)
store = db.get_store(config.get('base', 'database_uri')) self.__queue.put(event.src_path, OP_REMOVE)
track = store.find(db.Track, db.Track.path == event.src_path).one()
if track:
scanner = Scanner(store)
scanner.remove_file(track.path)
scanner.finish()
store.commit()
else:
self.__logger.debug("Deleted file %s not in the database", event.src_path)
store.close()
def on_modified(self, event): def on_modified(self, event):
self.__logger.debug("File modified: '%s'", event.src_path) self.__logger.debug("File modified: '%s'", event.src_path)
self.__queue.put(event.src_path) self.__queue.put(event.src_path, OP_SCAN)
def on_moved(self, event): def on_moved(self, event):
pass self.__logger.debug("File moved: '%s' -> '%s'", event.src_path, event.dst_path)
self.__queue.put(event.src_path, OP_MOVE)
class Event(object):
def __init__(self, path, operation):
if operation & (OP_SCAN | OP_REMOVE) == (OP_SCAN | OP_REMOVE):
raise Exception("Flags SCAN and REMOVE both set")
self.__path = path
self.__time = time.time()
self.__op = operation
def set(self, operation):
if operation & (OP_SCAN | OP_REMOVE) == (OP_SCAN | OP_REMOVE):
raise Exception("Flags SCAN and REMOVE both set")
self.__time = time.time()
if operation & OP_SCAN:
self.__op &= ~OP_REMOVE
if operation & OP_REMOVE:
self.__op &= ~OP_SCAN
self.__op |= operation
@property
def path(self):
return self.__path
@property
def time(self):
return self.__time
@property
def operation(self):
return self.__op
class ScannerProcessingQueue(Thread): class ScannerProcessingQueue(Thread):
def __init__(self, logger): def __init__(self, logger):
@ -86,12 +114,17 @@ class ScannerProcessingQueue(Thread):
store = db.get_store(config.get('base', 'database_uri')) store = db.get_store(config.get('base', 'database_uri'))
scanner = Scanner(store) scanner = Scanner(store)
path = self.__next_item() item = self.__next_item()
while path: while item:
self.__logger.info("Scanning: '%s'", path) if item.operation & OP_SCAN:
scanner.scan_file(path) self.__logger.info("Scanning: '%s'", item.path)
path = self.__next_item() scanner.scan_file(item.path)
if item.operation & OP_REMOVE:
self.__logger.info("Removing: '%s'", item.path)
scanner.remove_file(item.path)
item = self.__next_item()
scanner.finish()
store.commit() store.commit()
store.close() store.close()
self.__logger.debug("Freeing scanner") self.__logger.debug("Freeing scanner")
@ -102,12 +135,16 @@ class ScannerProcessingQueue(Thread):
with self.__cond: with self.__cond:
self.__cond.notify() self.__cond.notify()
def put(self, path): def put(self, path, operation):
if not self.__running: if not self.__running:
raise RuntimeError("Trying to put an item in a stopped queue") raise RuntimeError("Trying to put an item in a stopped queue")
with self.__cond: with self.__cond:
self.__queue[path] = time.time() if path in self.__queue:
self.__queue[path].set(operation)
else:
self.__queue[path] = Event(path, operation)
if self.__timer: if self.__timer:
self.__timer.cancel() self.__timer.cancel()
self.__timer = Timer(5, self.__wakeup) self.__timer = Timer(5, self.__wakeup)
@ -123,10 +160,10 @@ class ScannerProcessingQueue(Thread):
if not self.__queue: if not self.__queue:
return None return None
next = sorted(self.__queue.iteritems(), key = lambda i: i[1])[0] next = min(self.__queue.iteritems(), key = lambda i: i[1].time)
if not self.__running or next[1] + 5 < time.time(): if not self.__running or next[1].time + 5 < time.time():
del self.__queue[next[0]] del self.__queue[next[0]]
return next[0] return next[1]
return None return None