move files

liquidsoap/management/commands/liquidsoap.py

@@ -0,0 +1,150 @@
+"""
+Main tool to work with liquidsoap. We can:
+- monitor Liquidsoap's sources and do logs, print what's on air.
+- generate configuration files and playlists for a given station
+"""
+import os
+import time
+import re
+from argparse import RawTextHelpFormatter
+
+from django.conf import settings as main_settings
+from django.core.management.base import BaseCommand, CommandError
+from django.template.loader import render_to_string
+from django.utils import timezone as tz
+
+import aircox.programs.models as models
+import aircox.programs.settings as programs_settings
+
+import aircox.liquidsoap.settings as settings
+import aircox.liquidsoap.utils as utils
+
+
+class StationConfig:
+    """
+    Configuration and playlist generator for a station.
+    """
+    controller = None
+
+    def __init__ (self, station):
+        self.controller = utils.Controller(station, False)
+
+    def handle (self, options):
+        os.makedirs(self.controller.path, exist_ok = True)
+        if options.get('config') or options.get('all'):
+            self.make_config()
+        if options.get('streams') or options.get('all'):
+            self.make_playlists()
+
+    def make_config (self):
+        log_script = main_settings.BASE_DIR \
+                     if hasattr(main_settings, 'BASE_DIR') else \
+                     main_settings.PROJECT_ROOT
+        log_script = os.path.join(log_script, 'manage.py') + \
+                        ' liquidsoap_log'
+
+
+        context = {
+            'controller': self.controller,
+            'settings': settings,
+            'log_script': log_script,
+        }
+
+        data = render_to_string('aircox/liquidsoap/station.liq', context)
+        data = re.sub(r'\s*\\\n', r'#\\n#', data)
+        data = data.replace('\n', '')
+        data = re.sub(r'#\\n#', '\n', data)
+        with open(self.controller.config_path, 'w+') as file:
+            file.write(data)
+
+    def make_playlists (self):
+        for stream in self.controller.streams.values():
+            program = stream.program
+
+            sounds = models.Sound.objects.filter(
+                # good_quality = True,
+                type = models.Sound.Type['archive'],
+                path__startswith = os.path.join(
+                    programs_settings.AIRCOX_SOUND_ARCHIVES_SUBDIR,
+                    program.path
+                )
+            )
+            with open(stream.path, 'w+') as file:
+                file.write('\n'.join(sound.path for sound in sounds))
+
+
+class Command (BaseCommand):
+    help= __doc__
+    output_dir = settings.AIRCOX_LIQUIDSOAP_MEDIA
+
+    def add_arguments (self, parser):
+        parser.formatter_class=RawTextHelpFormatter
+
+        group = parser.add_argument_group('monitor')
+        group.add_argument(
+            '-o', '--on_air', action='store_true',
+            help='print what is on air'
+        )
+        group.add_argument(
+            '-m', '--monitor', action='store_true',
+            help='run in monitor mode'
+        )
+        group.add_argument(
+            '-d', '--delay', type=int,
+            default=1000,
+            help='time to sleep in milliseconds before update on monitor'
+        )
+
+        group = parser.add_argument_group('configuration')
+        parser.add_argument(
+            '-s', '--station', type=int,
+            help='generate files for the given station (if not set, do it for'
+                 ' all available stations)'
+        )
+        parser.add_argument(
+            '-c', '--config', action='store_true',
+            help='generate liquidsoap config file'
+        )
+        parser.add_argument(
+            '-S', '--streams', action='store_true',
+            help='generate all stream playlists'
+        )
+        parser.add_argument(
+            '-a', '--all', action='store_true',
+            help='shortcut for -cS'
+        )
+
+
+    def handle (self, *args, **options):
+        if options.get('station'):
+            station = models.Station.objects.get(id = options.get('station'))
+            StationConfig(station).handle(options)
+        elif options.get('all') or options.get('config') or \
+                options.get('streams'):
+            for station in models.Station.objects.filter(active = True):
+                StationConfig(station).handle(options)
+
+        if options.get('on_air') or options.get('monitor'):
+            self.handle_monitor(options)
+
+    def handle_monitor (self, options):
+        self.monitor = utils.Monitor()
+        self.monitor.update()
+
+        if options.get('on_air'):
+            for id, controller in self.monitor.controller.items():
+                print(id, controller.on_air)
+            return
+
+        if options.get('monitor'):
+            delay = options.get('delay') / 1000
+            while True:
+                for controller in self.monitor.controllers.values():
+                    try:
+                        controller.monitor()
+                    except Exception as err:
+                        print(err)
+                time.sleep(delay)
+            return
+
+

liquidsoap/management/commands/liquidsoap_log.py

@@ -0,0 +1,62 @@
+"""
+This script is used by liquidsoap in order to log a file change. It should not
+be used for other purposes.
+"""
+import os
+from argparse import RawTextHelpFormatter
+
+from django.utils import timezone as tz
+from django.core.management.base import BaseCommand, CommandError
+
+import aircox.programs.models as programs
+
+
+class Command (BaseCommand):
+    help= __doc__
+
+    @staticmethod
+    def date(s):
+        try:
+            return tz.make_aware(tz.datetime.strptime(s, '%Y/%m/%d %H:%M:%S'))
+        except ValueError:
+            raise argparse.ArgumentTypeError('Invalid date format')
+
+
+    def add_arguments (self, parser):
+        parser.formatter_class=RawTextHelpFormatter
+        parser.add_argument(
+            '-c', '--comment', type=str,
+            help='log comment'
+        )
+        parser.add_argument(
+            '-s', '--source', type=str,
+            required=True,
+            help='source path'
+        )
+        parser.add_argument(
+            '-p', '--path', type=str,
+            required=True,
+            help='sound path to log'
+        )
+        parser.add_argument(
+            '-d', '--date', type=Command.date,
+            help='set date instead of now (using format "%Y/%m/%d %H:%M:%S")'
+        )
+
+
+    def handle (self, *args, **options):
+        comment = options.get('comment') or ''
+        path = os.path.realpath(options.get('path'))
+
+        sound = programs.Sound.objects.filter(path = path)
+        if sound:
+            sound = sound[0]
+        else:
+            sound = None
+            comment += '\nunregistered sound: {}'.format(path)
+
+        programs.Log(source = options.get('source'),
+                     comment = comment,
+                     related_object = sound).save()
+
+