diff options
Diffstat (limited to 'helsinki.program/views.py')
| -rw-r--r-- | helsinki.program/views.py | 150 |
1 files changed, 150 insertions, 0 deletions
diff --git a/helsinki.program/views.py b/helsinki.program/views.py new file mode 100644 index 0000000..23c1483 --- /dev/null +++ b/helsinki.program/views.py @@ -0,0 +1,150 @@ +from django.views.generic.list import ListView +from django.views.generic.base import TemplateView +from django.shortcuts import get_object_or_404 + +from models import BroadcastFormat, MusicFocus, Note, Show, ShowInformation, ShowTopic, TimeSlot + +from datetime import date, datetime, time, timedelta + +class ShowListView(ListView): + context_object_name = 'shows' + + def get_context_data(self, **kwargs): + context = super(ShowListView, self).get_context_data(**kwargs) + + context['broadcastformats'] = BroadcastFormat.objects.all() + context['musicfoci'] = MusicFocus.objects.all() + context['showinformations'] = ShowInformation.objects.all() + context['showtopics'] = ShowTopic.objects.all() + + return context + + def get_queryset(self): + if 'broadcastformat' in self.request.GET: + broadcastformat = get_object_or_404(BroadcastFormat, slug=self.request.GET['broadcastformat']) + + return Show.objects.filter(broadcastformat=broadcastformat) + elif 'musicfocus' in self.request.GET: + musicfocus = get_object_or_404(MusicFocus, slug=self.request.GET['musicfocus']) + + return Show.objects.filter(musicfocus=musicfocus) + elif 'showinformation' in self.request.GET: + showinformation = get_object_or_404(ShowInformation, slug=self.request.GET['showinformation']) + + return Show.objects.filter(showinformation=showinformation) + elif 'showtopic' in self.request.GET: + showtopic = get_object_or_404(ShowTopic, slug=self.request.GET['showtopic']) + + return Show.objects.filter(showtopic=showtopic) + else: + return Show.objects.all() + +class RecommendationsView(ListView): + context_object_name = 'recommendations' + template_name = 'program/recommendations.html' + + def get_queryset(self): + now = datetime.now() + in_one_week = now + timedelta(weeks=1) + + return Note.objects.filter(status=1, timeslot__start__range=(now, in_one_week))[:10] + +class TodayScheduleView(TemplateView): + template_name = 'program/day_schedule.html' + + def get_context_data(self, **kwargs): + context = super(TodayScheduleView, self).get_context_data(**kwargs) + + now = datetime.now() + today = datetime.combine(date.today(), time(6, 0)) + tomorrow = datetime.combine(date.today()+timedelta(days=1), time(6, 0)) + + context['day'] = today + context['broadcastformats'] = BroadcastFormat.objects.all() + context['recommendations'] = Note.objects.filter(status=1, timeslot__start__range=(now, tomorrow)) + + if 'broadcastformat' in self.request.GET: + broadcastformat = get_object_or_404(BroadcastFormat, slug=self.request.GET['broadcastformat']) + + context['timeslots'] = TimeSlot.objects.filter(start__range=(today, tomorrow), show__broadcastformat=broadcastformat) + else: + context['timeslots'] = TimeSlot.objects.filter(start__range=(today, tomorrow)) + + return context + +class DayScheduleView(TemplateView): + template_name = 'program/day_schedule.html' + + def get_context_data(self, **kwargs): + context = super(DayScheduleView, self).get_context_data(**kwargs) + + year = context['params']['year'] + month = context['params']['month'] + day = context['params']['day'] + + # start the day at 6 + this_day = datetime.strptime('%s__%s__%s__06__00' % (year, month, day), '%Y__%m__%d__%H__%M') + that_day = this_day+timedelta(days=1) + + context['day'] = this_day + context['broadcastformats'] = BroadcastFormat.objects.all() + context['recommendations'] = Note.objects.filter(status=1, timeslot__start__range=(this_day, that_day)) + + if 'broadcastformat' in self.request.GET: + broadcastformat = get_object_or_404(BroadcastFormat, slug=self.request.GET['broadcastformat']) + + context['timeslots'] = TimeSlot.objects.filter(start__range=(this_day, that_day), show__broadcastformat=broadcastformat) + else: + context['timeslots'] = TimeSlot.objects.filter(start__range=(this_day, that_day)) + + return context + +class CurrentShowView(TemplateView): + template_name = 'program/current_box.html' + + def get_context_data(self, **kwargs): + context = super(CurrentShowView, self).get_context_data(**kwargs) + + context['current'] = TimeSlot.objects.get_or_create_current() + context['next'] = TimeSlot.objects.get_or_create_current().get_next_by_start() + context['after_next'] = TimeSlot.objects.get_or_create_current().get_next_by_start().get_next_by_start() + + return context + +class WeekScheduleView(TemplateView): + template_name = 'program/week_schedule.html' + + def get_context_data(self, **kwargs): + context = super(WeekScheduleView, self).get_context_data(**kwargs) + + year = context['params']['year'] + week = context['params']['week'] + + # start the day at 6 + monday = datetime.strptime('%s__%s__1__06__00' % (year, week), '%Y__%W__%w__%H__%M') + + tuesday = monday+timedelta(days=1) + wednesday = monday+timedelta(days=2) + thursday = monday+timedelta(days=3) + friday = monday+timedelta(days=4) + saturday = monday+timedelta(days=5) + sunday = monday+timedelta(days=6) + next_monday = monday+timedelta(days=7) + + context['monday'] = monday + context['tuesday'] = tuesday + context['wednesday'] = wednesday + context['thursday'] = thursday + context['friday'] = friday + context['saturday'] = saturday + context['sunday'] = sunday + + context['monday_timeslots'] = TimeSlot.objects.filter(start__range=(monday, tuesday)) + context['tuesday_timeslots'] = TimeSlot.objects.filter(start__range=(tuesday, wednesday)) + context['wednesday_timeslots'] = TimeSlot.objects.filter(start__range=(wednesday, thursday)) + context['thursday_timeslots'] = TimeSlot.objects.filter(start__range=(thursday, friday)) + context['friday_timeslots'] = TimeSlot.objects.filter(start__range=(friday, saturday)) + context['saturday_timeslots'] = TimeSlot.objects.filter(start__range=(saturday, sunday)) + context['sunday_timeslots'] = TimeSlot.objects.filter(start__range=(sunday, next_monday)) + + return context |