diff --git a/client/djrandom_client/client.py b/client/djrandom_client/client.py index f10a8100df9e7220d9ce78c1cdf114a4487cd66c..9a625923cfe5deee5fe0f22e158c860bc057d44a 100644 --- a/client/djrandom_client/client.py +++ b/client/djrandom_client/client.py @@ -60,6 +60,15 @@ def run_client(server_url, music_dir, api_key, run_once, bwlimit): else: log.warn('inotify/fsevents support not enabled on this platform') + def _cleanup(signum, frame): + if enable_watcher: + log.info('stopping watcher...') + wtch.stop() + log.info('got signal %d, exiting...' % signum) + upl.stop() + signal.signal(signal.SIGINT, _cleanup) + signal.signal(signal.SIGTERM, _cleanup) + upl.run() diff --git a/client/djrandom_client/linux_watcher.py b/client/djrandom_client/linux_watcher.py index b3f59d451138562a78fc70f847a7c3066f17e5a1..797e1cf55cc1fb371d687a257916624162158b03 100644 --- a/client/djrandom_client/linux_watcher.py +++ b/client/djrandom_client/linux_watcher.py @@ -4,16 +4,21 @@ import pyinotify log = logging.getLogger(__name__) -class Watcher(object): +class Watcher(pyinotify.ProcessEvent): def __init__(self, base, queue): + self.queue = queue self.wm = pyinotify.WatchManager() - self.wm.add_watch('/tmp', pyinotify.IN_CLOSE_WRITE, rec=True) self.notifier = pyinotify.ThreadedNotifier(self.wm, self) self.notifier.setDaemon(True) self.notifier.start() + self.wm.add_watch(base, pyinotify.IN_CLOSE_WRITE, rec=True) + + def stop(self): + self.notifier.stop() def process_IN_CLOSE(self, event): if event.pathname.lower().endswith('.mp3'): log.debug('event in %s: %x' % (event.pathname, event.mask)) self.queue.put(event.pathname) + diff --git a/client/djrandom_client/upload.py b/client/djrandom_client/upload.py index 412643d1d3b3d9fdbeec7ac7dfc56626e4d3e714..2066716985f2fc91c68410671288a450f5c63c8e 100644 --- a/client/djrandom_client/upload.py +++ b/client/djrandom_client/upload.py @@ -99,3 +99,7 @@ class Uploader(object): finally: log.debug('uploader thread exiting') self.db.close() + + def stop(self): + self.queue.put(None) +