fix ups

2018-06-27 18:59:11 +02:00
parent f70b692aff
commit 08f519451a
4 changed files with 11 additions and 6 deletions
--- a/aircox_cms/models/init.py
+++ b/aircox_cms/models/init.py
@ -811,8 +811,11 @@ class TimetablePage(DatedListPage):
    def get_queryset(self, request, context):
        diffs = []
        for date in context['nav_dates']['dates']:
-            items = aircox.models.Diffusion.objects.at(self.station, date)
-            items = [ DiffusionPage.as_item(item) for item in items ]
+            items = [
+                DiffusionPage.as_item(item)
+                for item in aircox.models.Diffusion.objects \
+                                  .station(self.station).at(date)
+            ]
            diffs.append((date, items))
        return diffs