forked from rc/aircox
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
from collections import OrderedDict
|
|
import datetime
|
|
|
|
from django.views.generic import ListView
|
|
|
|
from ..converters import WeekConverter
|
|
from ..models import Diffusion, Episode, Program, Sound
|
|
from .base import BaseView
|
|
from .program import ProgramPageDetailView
|
|
from .page import ParentMixin, PageListView
|
|
|
|
|
|
__all__ = ['EpisodeDetailView', 'EpisodeListView', 'TimetableView']
|
|
|
|
|
|
class EpisodeDetailView(ProgramPageDetailView):
|
|
model = Episode
|
|
|
|
def get_podcasts(self, diffusion):
|
|
return Sound.objects.diffusion(diffusion).podcasts()
|
|
|
|
def get_context_data(self, **kwargs):
|
|
self.program = kwargs.setdefault('program', self.object.program)
|
|
|
|
kwargs.setdefault('parent', self.program)
|
|
if not 'tracks' in kwargs:
|
|
kwargs['tracks'] = self.object.track_set.order_by('position')
|
|
if not 'podcasts' in kwargs:
|
|
kwargs['podcasts'] = self.object.sound_set.podcasts()
|
|
return super().get_context_data(**kwargs)
|
|
|
|
|
|
class EpisodeListView(ParentMixin, PageListView):
|
|
model = Episode
|
|
template_name = 'aircox/diffusion_list.html'
|
|
item_template_name = 'aircox/episode_item.html'
|
|
show_headline = True
|
|
|
|
parent_model = Program
|
|
fk_parent = 'program'
|
|
|
|
|
|
class TimetableView(BaseView, ListView):
|
|
""" View for timetables """
|
|
template_name_suffix = '_timetable'
|
|
model = Diffusion
|
|
# ordering = ('start',)
|
|
|
|
date = None
|
|
start = None
|
|
end = None
|
|
|
|
def get_queryset(self):
|
|
self.date = self.kwargs.get('date') or datetime.date.today()
|
|
self.start = self.date - datetime.timedelta(days=self.date.weekday())
|
|
self.end = self.start + datetime.timedelta(days=7)
|
|
return super().get_queryset().range(self.start, self.end) \
|
|
.order_by('start')
|
|
|
|
def get_context_data(self, **kwargs):
|
|
# regoup by dates
|
|
by_date = OrderedDict()
|
|
date = self.start
|
|
while date < self.end:
|
|
by_date[date] = []
|
|
date += datetime.timedelta(days=1)
|
|
|
|
for diffusion in self.object_list:
|
|
if diffusion.date not in by_date:
|
|
continue
|
|
by_date[diffusion.date].append(diffusion)
|
|
|
|
return super().get_context_data(
|
|
by_date=by_date,
|
|
date=self.date,
|
|
start=self.start,
|
|
end=self.end - datetime.timedelta(days=1),
|
|
prev_date=self.start - datetime.timedelta(days=1),
|
|
next_date=self.end + datetime.timedelta(days=1),
|
|
**kwargs
|
|
)
|
|
|
|
|