start to restructure the project tree

aircox/liquidsoap/management/commands/liquidsoap.py

@@ -0,0 +1,55 @@
+"""
+Monitor Liquidsoap's sources, logs, and even print what's on air.
+"""
+import time
+from argparse import RawTextHelpFormatter
+
+from django.core.management.base import BaseCommand, CommandError
+from django.utils import timezone as tz
+
+import aircox.liquidsoap.settings as settings
+import aircox.liquidsoap.utils as utils
+import aircox.programs.models as models
+
+
+class Command (BaseCommand):
+    help= __doc__
+    output_dir = settings.AIRCOX_LIQUIDSOAP_MEDIA
+
+    def add_arguments (self, parser):
+        parser.formatter_class=RawTextHelpFormatter
+        parser.add_argument(
+            '-o', '--on_air', action='store_true',
+            help='Print what is on air'
+        )
+        parser.add_argument(
+            '-m', '--monitor', action='store_true',
+            help='Runs in monitor mode'
+        )
+        parser.add_argument(
+            '-d', '--delay', type=int,
+            default=1000,
+            help='Time to sleep in milliseconds before update on monitor'
+        )
+
+
+    def handle (self, *args, **options):
+        connector = utils.Connector()
+        self.monitor = utils.Monitor(connector)
+        self.monitor.update()
+
+        if options.get('on_air'):
+            for id, controller in self.monitor.controller.items():
+                print(id, controller.on_air)
+
+        if options.get('monitor'):
+            delay = options.get('delay') / 1000
+            while True:
+                for controller in self.monitor.controllers.values():
+                    try:
+                        controller.monitor()
+                    except Exception, e:
+                        print(e)
+                time.sleep(delay)
+
+

aircox/liquidsoap/management/commands/liquidsoap_files.py

@@ -0,0 +1,119 @@
+"""
+Generate configuration files and playlists for liquidsoap using settings, streams and
+so on
+"""
+import os
+import re
+from argparse import RawTextHelpFormatter
+
+from django.core.management.base import BaseCommand, CommandError
+from django.views.generic.base import View
+from django.template.loader import render_to_string
+
+import aircox.liquidsoap.settings as settings
+import aircox.liquidsoap.utils as utils
+import aircox.programs.settings as programs_settings
+import aircox.programs.models as models
+
+
+class Command (BaseCommand):
+    help= __doc__
+    output_dir = settings.AIRCOX_LIQUIDSOAP_MEDIA
+
+    def add_arguments (self, parser):
+        parser.formatter_class=RawTextHelpFormatter
+        parser.add_argument(
+            'output', metavar='PATH', type=str, nargs='?',
+            help='force output to file (- to stdout) for single actions; to a '
+                 'given dir when using --all')
+        parser.add_argument(
+            '-c', '--config', action='store_true',
+            help='Generate liquidsoap config file'
+        )
+        parser.add_argument(
+            '-d', '--diffusion', action='store_true',
+            help='Generate the playlist for the current scheduled diffusion'
+        )
+        parser.add_argument(
+            '-s', '--stream', type=int,
+            help='Generate the playlist of a stream with the given id'
+        )
+        parser.add_argument(
+            '-S', '--streams', action='store_true',
+            help='Generate all stream playlists'
+        )
+        parser.add_argument(
+            '-a', '--all', action='store_true',
+            help='Generate all playlists (stream and scheduled diffusion) '
+                 'and config file'
+        )
+
+    def handle (self, *args, **options):
+        output = options.get('output') or None
+        if options.get('config'):
+            data = self.get_config(output = output)
+            return
+
+        if options.get('stream'):
+            stream = options['stream']
+            if type(stream) is int:
+                stream = models.Stream.objects.get(id = stream,
+                                                   program__active = True)
+
+            data = self.get_playlist(stream, output = output)
+            return
+
+        if options.get('all') or options.get('streams'):
+            if output:
+                if not os.path.isdir(output):
+                    raise CommandError('given output is not a directory')
+                self.output_dir = output
+
+            if options.get('all'):
+                self.handle(config = True)
+
+            for stream in models.Stream.objects.filter(program__active = True):
+                self.handle(stream = stream)
+            self.output_dir = settings.AIRCOX_LIQUIDSOAP_MEDIA
+            return
+
+        raise CommandError('nothing to do')
+
+    def print (self, data, path, default):
+        if path and path == '-':
+            print(data)
+            return
+
+        if not path:
+            path = os.path.join(self.output_dir, default)
+        with open(path, 'w+') as file:
+            file.write(data)
+
+    def get_config (self, output = None):
+        context = {
+            'monitor': utils.Monitor(),
+            'settings': settings,
+        }
+
+        data = render_to_string('aircox_liquidsoap/config.liq', context)
+        data = re.sub(r'\s*\\\n', r'#\\n#', data)
+        data = data.replace('\n', '')
+        data = re.sub(r'#\\n#', '\n', data)
+        self.print(data, output, 'aircox.liq')
+
+    def get_playlist (self, stream = None, output = None):
+        program = stream.program
+        source = utils.Source(program = 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
+            )
+        )
+        data = '\n'.join(sound.path for sound in sounds)
+        self.print(data, output, utils.Source(program = program).path)
+
+