almost alpha version

python_apps/media-monitor2/media/monitor/bootstrap.py

@@ -0,0 +1,59 @@
+import os
+from pydispatch import dispatcher
+from media.monitor.events import OrganizeFile, NewFile, DeleteFile
+import media.monitor.pure as mmp
+from media.monitor.log import Loggable
+
+class Bootstrapper(Loggable):
+    """
+    Bootstrapper reads all the info in the filesystem flushes organize
+    events and watch events
+    """
+    def __init__(self,db,last_ran,org_channels,watch_channels):
+        self.db = db
+        self.org_channels = org_channels
+        self.watch_channels = watch_channels
+        self.last_ran = last_ran
+
+    def flush_organize(self):
+        """
+        walks the organize directories and sends an organize event for every file manually
+        """
+        flushed = 0
+        for pc in self.org_channels:
+            for f in mmp.walk_supported(pc.path, clean_empties=True):
+                self.logger.info("Bootstrapping: File in 'organize' directory: '%s'" % f)
+                dispatcher.send(signal=pc.signal, sender=self, event=OrganizeFile(f))
+                flushed += 1
+        self.logger.info("Flushed organized directory with %d files" % flushed)
+
+    def flush_watch(self):
+        """
+        Syncs the file system into the database. Walks over deleted/new/modified files since
+        the last run in mediamonitor and sends requests to make the database consistent with
+        file system
+        """
+        songs = set()
+        modded = 0
+        deleted = 0
+        for pc in self.watch_channels:
+            for f in mmp.walk_supported(pc.path, clean_empties=False):
+                songs.add(f)
+                if os.path.getmtime(f) > self.last_ran:
+                    modded += 1
+                    dispatcher.send(signal=pc.signal, sender=self, event=DeleteFile(f))
+                    dispatcher.send(signal=pc.signal, sender=self, event=NewFile(f))
+        # Want all files in the database that are not in the filesystem
+        for to_delete in (self.db - songs):
+            for pc in self.watch_channels:
+                if os.path.commonprefix([pc.path, to_delete]) == pc.path:
+                    dispatcher.send(signal=pc.signal, sender=self, event=DeleteFile(f))
+                    os.remove(to_delete)
+                    deleted += 1
+                    break
+            else:
+                self.logger.info("Error, could not find watch directory of would be deleted \
+                                  file '%s'" % to_delete)
+        self.logger.info("Flushed watch directories. (modified, deleted) = (%d, %d)"
+                         % (modded, deleted) )
+