add middleware for aircox used for timezone and current station selection in admin; add page summary for today's diffusions menu in link title

aircox/middleware.py

@@ -0,0 +1,80 @@
+import pytz
+import django.utils.timezone as tz
+
+import aircox.models as models
+
+
+class AircoxInfo:
+    """
+    Used to store informations about Aircox on a request. Some of theses
+    information are None when user is anonymous.
+    """
+    station = None
+    """
+    Current station
+    """
+    default_station = False
+    """
+    Default station is used as the current station
+    """
+
+    def __init__(self, **kwargs):
+        self.__dict__.update(kwargs)
+
+
+class AircoxMiddleware(object):
+    """
+    Middleware used to get default info for the given website. Theses
+    This middleware must be set after the middleware
+        'django.contrib.auth.middleware.AuthenticationMiddleware',
+    """
+    default_qs = models.Station.objects.filter(default = True)
+
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def init_station(self, aircox):
+        station = None
+        try:
+            pk = request.session.get('aircox.station')
+            if pk:
+                station = int(pk)
+                station = models.Station.objects.filter(pk = station).first()
+
+            if not station:
+                pk = None
+                station = self.default_qs.first() or \
+                            models.Station.objects.first()
+
+            aircox.station = station
+            aircox.default_station = (pk is None)
+        except:
+            pass
+
+    def init_timezone(self, aircox):
+        # note: later we can use http://freegeoip.net/ on user side if
+        # required
+        timezone = None
+        try:
+            timezone = request.session.get('aircox.timezone')
+            if timezone:
+                timezone = pytz.timezone(timezone)
+        except:
+            pass
+
+        if not timezone:
+            timezone = tz.get_current_timezone()
+            tz.activate(timezone)
+
+    def __call__(self, request):
+        tz.activate(pytz.timezone('Europe/Brussels'))
+        aircox = AircoxInfo()
+
+        if request.user.is_authenticated:
+            self.init_station(aircox)
+        self.init_timezone(aircox)
+
+        request.aircox = aircox
+        return self.get_response(request)
+
+

aircox/models.py

@@ -167,6 +167,11 @@ class Station(Nameable):
         max_length = 256,
         blank = True,
     )
+    default = models.BooleanField(
+        _('default station'),
+        default = True,
+        help_text = _('if checked, this station is used as the main one')
+    )
 
     #
     # Controllers
@@ -333,6 +338,12 @@ class Station(Nameable):
                 self.slug
             )
 
+        if self.default:
+            qs = Station.objects.filter(default = True)
+            if self.pk:
+                qs = qs.exclude(pk = self.pk)
+            qs.update(default = False)
+
         super().save(*args, **kwargs)
 
 

aircox_cms/wagtail_hooks.py

@@ -1,11 +1,15 @@
+import json
+
 from django.utils import timezone as tz
 from django.utils.translation import ugettext_lazy as _
 from django.core.urlresolvers import reverse
 from django.contrib.staticfiles.templatetags.staticfiles import static
 from django.utils.html import format_html
+from django.utils.safestring import mark_safe
 
 from wagtail.wagtailcore import hooks
 from wagtail.wagtailadmin.menu import MenuItem, Menu, SubmenuMenuItem
+from wagtail.wagtailcore.models import PageRevision
 
 from wagtail.contrib.modeladmin.options import \
     ModelAdmin, ModelAdminGroup, modeladmin_register
@@ -14,6 +18,10 @@ from wagtail.contrib.modeladmin.options import \
 import aircox.models
 import aircox_cms.models as models
 
+
+#
+# ModelAdmin items
+#
 class ProgramAdmin(ModelAdmin):
     model = aircox.models.Program
     menu_label = _('Programs')
@@ -79,16 +87,9 @@ class SoundAdmin(ModelAdmin):
 modeladmin_register(SoundAdmin)
 
 
-## Hooks
+#
-
+# Menus with sub-menus
-@hooks.register('insert_editor_css')
+#
-def editor_css():
-    return format_html(
-        '<link rel="stylesheet" href="{}">',
-        static('aircox_cms/css/cms.css')
-    )
-
-
 class GenericMenu(Menu):
     page_model = models.Publication
     explore = False
@@ -105,19 +106,19 @@ class GenericMenu(Menu):
         """
         pass
 
-    def get_title(self, item):
+    def make_item(self, item):
         """
-        Return the title of a menu-item for the given item
+        Return the instance of MenuItem for the given item in the queryset
         """
         pass
 
     def get_parent(self, item):
         """
-        Return id of the parent page for the given item
+        Return id of the parent page for the given item of the queryset
         """
         pass
 
-    def get_page_url(self, page_model, item):
+    def page_url(self, item):
         if item.page.count():
             name =  'wagtailadmin_explore' \
                     if self.explore else 'wagtailadmin_pages:edit'
@@ -129,7 +130,8 @@ class GenericMenu(Menu):
 
         return reverse(
             'wagtailadmin_pages:add', args= [
-                page_model._meta.app_label, page_model._meta.model_name,
+                self.page_model._meta.app_label,
+                self.page_model._meta.model_name,
                 parent_page.id
             ]
         )
@@ -141,14 +143,16 @@ class GenericMenu(Menu):
 
         qs = self.get_queryset()
         return [
-            MenuItem(self.get_title(x), self.get_page_url(self.page_model, x))
+            self.make_item(item) for item in qs
-            for x in qs
         ]
 
 
-class DiffusionsMenu(GenericMenu):
+#
+# Today's diffusions menu
+#
+class TodayMenu(GenericMenu):
     """
-    Menu to display diffusions of today
+    Menu to display today's diffusions
     """
     page_model = models.DiffusionPage
 
@@ -159,14 +163,25 @@ class DiffusionsMenu(GenericMenu):
             initial__isnull = True,
         ).order_by('start')
 
-    def get_title(self, item):
+    def make_item(self, item):
-        from django.utils.safestring import mark_safe
+        label = mark_safe(
-        from django.utils.timezone import localtime
+            '<i class="info">{}</i> {}'.format(
-        title = '<i class="info">{}</i> {}'.format(
+                tz.localtime(item.start).strftime('%H:%M'),
-            localtime(item.start).strftime('%H:%M'),
+                item.program.name
-            item.program.name
+            )
         )
-        return mark_safe(title)
+
+        attrs = {}
+
+        qs = PageRevision.objects.filter(page = item.page.first())
+        if qs.count():
+            summary = qs.latest('created_at').content_json
+            summary = json.loads(summary).get('summary')
+            attrs['title'] = summary
+        else:
+            summary = ''
+
+        return MenuItem(label, self.page_url(item), attrs = attrs)
 
     def get_parent(self, item):
         return item.program.page.first()
@@ -175,14 +190,17 @@ class DiffusionsMenu(GenericMenu):
 @hooks.register('register_admin_menu_item')
 def register_programs_menu_item():
     return SubmenuMenuItem(
-        _('Today\'s Diffusions'), DiffusionsMenu(),
+        _('Today\'s Diffusions'), TodayMenu(),
         classnames='icon icon-folder-open-inverse', order=101
     )
 
 
+#
+# Programs menu
+#
 class ProgramsMenu(GenericMenu):
     """
-    Menu to display all active programs
+    Display all active programs
     """
     page_model = models.DiffusionPage
     explore = True
@@ -193,8 +211,8 @@ class ProgramsMenu(GenericMenu):
                     .filter(stream__isnull = True) \
                     .order_by('name')
 
-    def get_title(self, item):
+    def make_item(self, item):
-        return item.name
+        return MenuItem(item.name, self.page_url(item))
 
     def get_parent(self, item):
         # TODO: #Station / get current site

instance/sample_settings.py

@@ -82,7 +82,7 @@ else:
     }
 
 # allowed hosts
-ALLOWED_HOSTS = ['127.0.0.1:8000']
+ALLOWED_HOSTS = ('127.0.0.1',)
 
 # secret key: you MUST put a consistent secret key
 SECRET_KEY = ''
@@ -145,7 +145,7 @@ INSTALLED_APPS = (
     'django.contrib.admin',
 )
 
-MIDDLEWARE_CLASSES = (
+MIDDLEWARE = (
     'django.middleware.gzip.GZipMiddleware',
     'htmlmin.middleware.HtmlMinifyMiddleware',
     'django.contrib.sessions.middleware.SessionMiddleware',

notes.md

@@ -12,6 +12,8 @@ This file is used as a reminder, can be used as crappy documentation too.
 - streamer:
     - restart running streamer on demand
     - add restart button for the streamer
+    \--> rewrite streamer for client-server controller
+    \--> move liquidsoap control on commands/streamer.py
 - cms:
     - button to select the current station