!111: tests: aircox.management (#114)

!111

Co-authored-by: bkfox <thomas bkfox net>
Reviewed-on: rc/aircox#114

aircox/management/commands/sounds_monitor.py

@@ -1,4 +1,5 @@
 #! /usr/bin/env python3
+# TODO: SoundMonitor class
 
 """Monitor sound files; For each program, check for:
 
@@ -23,20 +24,12 @@ To check quality of files, call the command sound_quality_check using the
 parameters given by the setting SOUND_QUALITY. This script requires
 Sox (and soxi).
 """
-import atexit
-import concurrent.futures as futures
 import logging
-import os
-import time
 from argparse import RawTextHelpFormatter
 
 from django.core.management.base import BaseCommand
-from watchdog.observers import Observer
 
-from aircox.conf import settings
-from aircox.management.sound_file import SoundFile
-from aircox.management.sound_monitor import MonitorHandler
-from aircox.models import Program, Sound
+from aircox.controllers.sound_monitor import SoundMonitor
 
 logger = logging.getLogger("aircox.commands")
 
@@ -44,109 +37,6 @@ logger = logging.getLogger("aircox.commands")
 class Command(BaseCommand):
     help = __doc__
 
-    def report(self, program=None, component=None, *content):
-        if not component:
-            logger.info(
-                "%s: %s", str(program), " ".join([str(c) for c in content])
-            )
-        else:
-            logger.info(
-                "%s, %s: %s",
-                str(program),
-                str(component),
-                " ".join([str(c) for c in content]),
-            )
-
-    def scan(self):
-        """For all programs, scan dirs."""
-        logger.info("scan all programs...")
-        programs = Program.objects.filter()
-
-        dirs = []
-        for program in programs:
-            logger.info("#%d %s", program.id, program.title)
-            self.scan_for_program(
-                program,
-                settings.SOUND_ARCHIVES_SUBDIR,
-                type=Sound.TYPE_ARCHIVE,
-            )
-            self.scan_for_program(
-                program,
-                settings.SOUND_EXCERPTS_SUBDIR,
-                type=Sound.TYPE_EXCERPT,
-            )
-            dirs.append(os.path.join(program.abspath))
-        return dirs
-
-    def scan_for_program(self, program, subdir, **sound_kwargs):
-        """Scan a given directory that is associated to the given program, and
-        update sounds information."""
-        logger.info("- %s/", subdir)
-        if not program.ensure_dir(subdir):
-            return
-
-        subdir = os.path.join(program.abspath, subdir)
-        sounds = []
-
-        # sounds in directory
-        for path in os.listdir(subdir):
-            path = os.path.join(subdir, path)
-            if not path.endswith(settings.SOUND_FILE_EXT):
-                continue
-
-            sound_file = SoundFile(path)
-            sound_file.sync(program=program, **sound_kwargs)
-            sounds.append(sound_file.sound.pk)
-
-        # sounds in db & unchecked
-        sounds = Sound.objects.filter(file__startswith=subdir).exclude(
-            pk__in=sounds
-        )
-        self.check_sounds(sounds, program=program)
-
-    def check_sounds(self, qs, **sync_kwargs):
-        """Only check for the sound existence or update."""
-        # check files
-        for sound in qs:
-            if sound.check_on_file():
-                SoundFile(sound.file.path).sync(sound=sound, **sync_kwargs)
-
-    def monitor(self):
-        """Run in monitor mode."""
-        with futures.ThreadPoolExecutor() as pool:
-            archives_handler = MonitorHandler(
-                settings.SOUND_ARCHIVES_SUBDIR,
-                pool,
-                type=Sound.TYPE_ARCHIVE,
-            )
-            excerpts_handler = MonitorHandler(
-                settings.SOUND_EXCERPTS_SUBDIR,
-                pool,
-                type=Sound.TYPE_EXCERPT,
-            )
-
-            observer = Observer()
-            observer.schedule(
-                archives_handler,
-                settings.PROGRAMS_DIR_ABS,
-                recursive=True,
-            )
-            observer.schedule(
-                excerpts_handler,
-                settings.PROGRAMS_DIR_ABS,
-                recursive=True,
-            )
-            observer.start()
-
-            def leave():
-                observer.stop()
-                observer.join()
-
-            atexit.register(leave)
-
-            while True:
-                time.sleep(1)
-
     def add_arguments(self, parser):
         parser.formatter_class = RawTextHelpFormatter
         parser.add_argument(
@@ -172,6 +62,7 @@ class Command(BaseCommand):
         )
 
     def handle(self, *args, **options):
+        SoundMonitor()
         if options.get("scan"):
             self.scan()
         # if options.get('quality_check'):