From 2b78ba19de9922e3260fc3d57a0db4c2c8233b55 Mon Sep 17 00:00:00 2001
From: bkfox <thomas bkfox net>
Date: Tue, 30 Jun 2015 17:31:01 +0200
Subject: [PATCH] work on monitor command

---
 programs/management/commands/programs.py | 31 ++++++++++++++++++++
 programs/models.py                       | 37 ++++++++++++++++++++----
 programs/settings.py                     |  3 +-
 3 files changed, 65 insertions(+), 6 deletions(-)

diff --git a/programs/management/commands/programs.py b/programs/management/commands/programs.py
index 13c5dcd..4e0ec25 100644
--- a/programs/management/commands/programs.py
+++ b/programs/management/commands/programs.py
@@ -1,3 +1,5 @@
+import argparse
+
 from django.core.management.base    import BaseCommand, CommandError
 from django.utils                   import timezone
 import programs.models              as models
@@ -17,6 +19,14 @@ class Model:
         self.post = post
 
 
+    def to_string (self):
+        return '\n'.join(
+                    [ '    - required: {}'.format(', '.join(self.required))
+                    , '    - optional: {}'.format(', '.join(self.optional))
+                    , (self.post is AddTags and '    - tags available\n') or
+                        '\n'
+                    ])
+
     def check_or_raise (self, options):
         for req in self.required:
             if req not in options:
@@ -145,11 +155,24 @@ models = {
                       }
                     , AddTags
                     )
+  , 'episode': Model( models.Episode
+                    , { 'title': str }
+                    , { 'subtitle': str, 'can_comment': bool, 'date': DateTime
+                      , 'parent_id': int, 'public': bool
+                      }
+                    , AddTags
+                    )
   , 'schedule': Model( models.Schedule
                     , { 'parent_id': int, 'date': DateTime, 'duration': Time
                       , 'frequency': int }
                     , { 'rerun': bool }
                     )
+  , 'soundfile': Model( models.SoundFile
+                    , { 'parent_id': int, 'date': DateTime, 'file': str
+                      , 'duration': Time}
+                    , { 'fragment': bool, 'embed': str, 'removed': bool }
+                    , AddTags
+                    )
 }
 
 
@@ -158,6 +181,7 @@ class Command (BaseCommand):
     help='Create, update, delete or dump an element of the given model.' \
          ' If no action is given, dump it'
 
+
     def add_arguments (self, parser):
         parser.add_argument( 'model', type=str
                            , metavar="MODEL"
@@ -210,6 +234,13 @@ class Command (BaseCommand):
         group.add_argument('--frequency',  type=int)
         group.add_argument('--rerun',      action='store_true')
 
+        # fields
+        parser.formatter_class=argparse.RawDescriptionHelpFormatter
+        parser.epilog = 'available fields per model:'
+        for name, model in models.items():
+            parser.epilog += '\n  ' + model.model.type() + ': \n' \
+                           + model.to_string()
+
 
     def handle (self, *args, **options):
         model = options.get('model')
diff --git a/programs/models.py b/programs/models.py
index b542a1d..8b8b395 100755
--- a/programs/models.py
+++ b/programs/models.py
@@ -280,6 +280,21 @@ class SoundFile (Metadata):
                   )
 
 
+    def get_mtime (self):
+        """
+        Get the last modification date from file
+        """
+        mtime = os.stat(self.file.path).st_mtime
+        mtime = timezone.datetime.fromtimestamp(mtime)
+        return timezone.make_aware(mtime, timezone.get_current_timezone())
+
+
+    def save (self, *args, **kwargs):
+        if not self.pk:
+            self.date = self.get_mtime()
+        super(SoundFile, self).save(*args, **kwargs)
+
+
     def __upload_path (self, filename):
         if self.parent and self.parent.parent:
             path = self.parent.parent.path
@@ -297,6 +312,7 @@ class SoundFile (Metadata):
         verbose_name_plural = _('Sounds')
 
 
+
 class Schedule (Model):
     parent      = models.ForeignKey( 'Program', blank = True, null = True )
     date        = models.DateTimeField(_('start'))
@@ -312,12 +328,12 @@ class Schedule (Model):
     rerun       = models.BooleanField(_('rerun'), default = False)
 
 
-    def match_date (self, at = timezone.datetime.today()):
+    def match_date (self, at = timezone.datetime.today(), check_time = False):
         """
         Return True if the given datetime matches the schedule
         """
-        if self.date.weekday() == at.weekday() and self.match_week(date):
-            return self.date.time() == at.date.time()
+        if self.date.weekday() == at.weekday() and self.match_week(at):
+            return (check_time and self.date.time() == at.date.time()) or True
         return False
 
 
@@ -401,7 +417,6 @@ class Schedule (Model):
 
 
 
-
 class Article (Publication):
     parent      = models.ForeignKey(
                       'self'
@@ -451,12 +466,24 @@ class Program (Publication):
                   )
 
     @property
-    def path(self):
+    def path (self):
         return os.path.join( settings.AIRCOX_PROGRAMS_DIR
                            , slugify(self.title + '_' + str(self.id))
                            )
 
 
+    def find_schedules (self, date):
+        """
+        Return schedules that match a given date
+        """
+        schedules = Schedule.objects.filter(parent = self)
+        r = []
+        for schedule in schedules:
+            if schedule.match_date(date):
+                r.append(schedule)
+        return r
+
+
     class Meta:
         verbose_name = _('Program')
         verbose_name_plural = _('Programs')
diff --git a/programs/settings.py b/programs/settings.py
index e03d6ad..e5b2d6a 100755
--- a/programs/settings.py
+++ b/programs/settings.py
@@ -10,6 +10,7 @@ ensure('AIRCOX_PROGRAMS_DIR',
        os.path.join(settings.MEDIA_ROOT, 'programs'))
 ensure('AIRCOX_SOUNDFILE_DEFAULT_DIR',
        os.path.join(AIRCOX_PROGRAMS_DIR + 'default'))
-
+ensure('AIRCOX_SOUNDFILE_EXT',
+        ('ogg','flac','wav','mp3','opus'))