refactor scanner, add progress bar

groove/shell/base.py

@@ -95,7 +95,7 @@ class BasePrompt(Completer):
     def autocomplete_values(self):
         return self._autocomplete_values
 
-    def get_completions(self, document, complete_event):
+    def get_completions(self, document, complete_event):  # pragma: no cover
         word = document.get_word_before_cursor()
         found = False
         for value in self.autocomplete_values:

groove/shell/interactive_shell.py

@@ -1,11 +1,13 @@
 from slugify import slugify
 
 from groove.db.manager import database_manager
-from groove.shell.base import BasePrompt, command, register_command
+from groove.db.scanner import MediaScanner
+from groove.shell.base import BasePrompt, command
+from groove.exceptions import InvalidPathError
 from groove import db
 from groove.playlist import Playlist
 
-from rich.table import Table, Column
+from rich.table import Column
 from rich import box
 
 from sqlalchemy import func
@@ -60,6 +62,34 @@ class InteractiveShell(BasePrompt):
         name = cmd + ' ' + ' '.join(parts)
         self.load([name.strip()])
 
+    @command("""
+    [title]SCANNING YOUR MEDIA[/title]
+
+    Use the [b]scan[/b] function to scan your media root for new, changed, and
+    deleted audio files. This process may take some time if you have a large
+    library!
+
+    Instead of scanning the entire MEDIA_ROOT, you can specify a PATH, which
+    must be a subdirectory of your MEDIA_ROOT. This is useful to import that
+    new new.
+
+    [title]USAGE[/title]
+
+        [link]> scan [PATH][/link]
+
+    """)
+    def scan(self, parts):
+        """
+        Scan your MEDIA_ROOT for changes.
+        """
+        path = ' '.join(parts) if parts else None
+        try:
+            scanner = MediaScanner(path=path, db=self.manager.session, console=self.console)
+        except InvalidPathError as e:
+            self.console.error(str(e))
+            return True
+        scanner.scan()
+
     @command("""
     [title]LISTS FOR THE LIST LOVER[/title]
 
@@ -75,7 +105,6 @@ class InteractiveShell(BasePrompt):
         """
         List all playlists.
         """
-        count = self.manager.session.query(func.count(db.playlist.c.id)).scalar()
         table = self.console.table(
             Column('#', justify='right', width=4),
             Column('Name'),
@@ -182,6 +211,7 @@ class InteractiveShell(BasePrompt):
         super().help(parts)
         return True
 
+
 def start():  # pragma: no cover
     with database_manager() as manager:
         InteractiveShell(manager).start()