import json from datetime import date, datetime, time, timedelta from django.db.models import Q from django.http import HttpResponse from django.shortcuts import get_object_or_404 from django.views.generic.base import TemplateView from django.views.generic.list import ListView from models import BroadcastFormat, MusicFocus, Note, Show, ShowInformation, ShowTopic, TimeSlot from program.utils import tofirstdayinisoweek class ShowListView(ListView): def get_queryset(self): queryset = Show.objects.filter(programslots__until__gt=date.today()).exclude(id=1).distinct() if 'broadcastformat' in self.request.GET: broadcastformat = get_object_or_404(BroadcastFormat, slug=self.request.GET['broadcastformat']) queryset = queryset.filter(broadcastformat=broadcastformat) elif 'musicfocus' in self.request.GET: musicfocus = get_object_or_404(MusicFocus, slug=self.request.GET['musicfocus']) queryset = queryset.filter(musicfocus=musicfocus) elif 'showinformation' in self.request.GET: showinformation = get_object_or_404(ShowInformation, slug=self.request.GET['showinformation']) queryset = queryset.filter(showinformation=showinformation) elif 'showtopic' in self.request.GET: showtopic = get_object_or_404(ShowTopic, slug=self.request.GET['showtopic']) queryset = queryset.filter(showtopic=showtopic) return queryset class RecommendationsListView(ListView): context_object_name = 'recommendation_list' template_name = 'recommendation_list.html' now = datetime.now() end = now + timedelta(weeks=1) queryset = TimeSlot.objects.filter(Q(note__isnull=False, note__status=1, start__range=(now, end)) | Q(show__broadcastformat__slug='sondersendung', start__range=(now, end))).order_by('start')[:20] class RecommendationsBoxView(RecommendationsListView): template_name='boxes/recommendation.html' class DayScheduleView(TemplateView): template_name = 'day_schedule.html' def get_context_data(self, **kwargs): year = self.kwargs.get('year', None) month = self.kwargs.get('month', None) day = self.kwargs.get('day', None) if year is None and month is None and day is None: today = datetime.combine(date.today(), time(6, 0)) else: today = datetime.strptime('%s__%s__%s__06__00' % (year, month, day), '%Y__%m__%d__%H__%M') tomorrow = today + timedelta(days=1) context = super(DayScheduleView, self).get_context_data(**kwargs) context['day'] = today context['recommendations'] = Note.objects.filter(status=1, timeslot__start__range=(today, tomorrow)) context['default_show'] = Show.objects.get(pk=1) timeslots = TimeSlot.objects.get_day_timeslots(today) if 'broadcastformat' in self.request.GET: broadcastformat = get_object_or_404(BroadcastFormat, slug=self.request.GET['broadcastformat']) context['timeslots'] = timeslots.filter(show__broadcastformat=broadcastformat) elif 'musicfocus' in self.request.GET: musicfocus = get_object_or_404(MusicFocus, slug=self.request.GET['musicfocus']) context['timeslots'] = timeslots.filter(show__musicfocus=musicfocus) elif 'showinformation' in self.request.GET: showinformation = get_object_or_404(ShowInformation, slug=self.request.GET['showinformation']) context['timeslots'] = timeslots.filter(show__showinformation=showinformation) elif 'showtopic' in self.request.GET: showtopic = get_object_or_404(ShowTopic, slug=self.request.GET['showtopic']) context['showtopic'] = timeslots.filter(show__showtopic=showtopic) else: context['timeslots'] = timeslots return context class CurrentShowBoxView(TemplateView): context_object_name = 'recommendation_list' template_name = 'boxes/current.html' def get_context_data(self, **kwargs): current_timeslot = TimeSlot.objects.get_or_create_current() previuos_timeslot = current_timeslot.get_previous_by_start() next_timeslot = current_timeslot.get_next_by_start() after_next_timeslot = next_timeslot.get_next_by_start() context = super(CurrentShowBoxView, self).get_context_data(**kwargs) context['current_timeslot'] = current_timeslot context['previuos_timeslot'] = previuos_timeslot context['next_timeslot'] = next_timeslot context['after_next_timeslot'] = after_next_timeslot return context class WeekScheduleView(TemplateView): template_name = 'week_schedule.html' def get_context_data(self, **kwargs): year = self.kwargs.get('year', None) week = self.kwargs.get('week', None) if year is None and week is None: year, week = datetime.now().strftime('%G__%V').split('__') monday = tofirstdayinisoweek(int(year), int(week)) 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) context = super(WeekScheduleView, self).get_context_data() context['monday'] = monday context['tuesday'] = tuesday context['wednesday'] = wednesday context['thursday'] = thursday context['friday'] = friday context['saturday'] = saturday context['sunday'] = sunday context['default_show'] = Show.objects.get(pk=1) context['monday_timeslots'] = TimeSlot.objects.get_day_timeslots(monday) context['tuesday_timeslots'] = TimeSlot.objects.get_day_timeslots(tuesday) context['wednesday_timeslots'] = TimeSlot.objects.get_day_timeslots(wednesday) context['thursday_timeslots'] = TimeSlot.objects.get_day_timeslots(thursday) context['friday_timeslots'] = TimeSlot.objects.get_day_timeslots(friday) context['saturday_timeslots'] = TimeSlot.objects.get_day_timeslots(saturday) context['sunday_timeslots'] = TimeSlot.objects.get_day_timeslots(sunday) context['last_w'] = datetime.strftime(monday - timedelta(days=7), '%G/%V') context['cur_w'] = datetime.strftime(monday, '%G/%V') context['next_w1'] = datetime.strftime(monday + timedelta(days=7), '%G/%V') context['next_w2'] = datetime.strftime(monday + timedelta(days=14), '%G/%V') context['next_w3'] = datetime.strftime(monday + timedelta(days=21), '%G/%V') context['next_w4'] = datetime.strftime(monday + timedelta(days=28), '%G/%V') return context def week_schedule(request, year=None, week=None): if year is None and week is None: year, week = datetime.now().strftime('%G__%V').split('__') monday = tofirstdayinisoweek(int(year), int(week)) 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) default_show = Show.objects.get(pk=1) extra_context = dict(monday=monday, tuesday=tuesday, wednesday=wednesday, thursday=thursday, friday=friday, saturday=saturday, sunday=sunday, default_show=default_show) extra_context['monday_timeslots'] = TimeSlot.objects.get_day_timeslots(monday) extra_context['tuesday_timeslots'] = TimeSlot.objects.get_day_timeslots(tuesday) extra_context['wednesday_timeslots'] = TimeSlot.objects.get_day_timeslots(wednesday) extra_context['thursday_timeslots'] = TimeSlot.objects.get_day_timeslots(thursday) extra_context['friday_timeslots'] = TimeSlot.objects.get_day_timeslots(friday) extra_context['saturday_timeslots'] = TimeSlot.objects.get_day_timeslots(saturday) extra_context['sunday_timeslots'] = TimeSlot.objects.get_day_timeslots(sunday) extra_context['last_w'] = datetime.strftime(monday - timedelta(days=7), '%G/%V') extra_context['cur_w'] = datetime.strftime(monday, '%G/%V') extra_context['next_w1'] = datetime.strftime(monday + timedelta(days=7), '%G/%V') extra_context['next_w2'] = datetime.strftime(monday + timedelta(days=14), '%G/%V') extra_context['next_w3'] = datetime.strftime(monday + timedelta(days=21), '%G/%V') extra_context['next_w4'] = datetime.strftime(monday + timedelta(days=28), '%G/%V') return TemplateView.as_view(template='week_schedule.html', extra_context=extra_context)(request) class StylesView(TemplateView): template_name = 'styles.css' content_type = 'text/css' def get_context_data(self, **kwargs): context = super(StylesView, self).get_context_data(**kwargs) context['broadcastformats'] = BroadcastFormat.objects.filter(enabled=True) context['musicfocus'] = MusicFocus.objects.all() context['showinformation'] = ShowInformation.objects.all() context['showtopic'] = ShowTopic.objects.all() return context def json_day_schedule(request, year=None, month=None, day=None): if year is None and month is None and day is None: today = datetime.combine(date.today(), time(0, 0)) else: today = datetime.strptime('%s__%s__%s__00__00' % (year, month, day), '%Y__%m__%d__%H__%M') timeslots = TimeSlot.objects.get_24h_timeslots(today) schedule = [] for ts in timeslots: if ts.programslot.automation_id: schedule.append((ts.start.strftime('%H:%M:%S'), ts.programslot.show.name, ts.programslot.automation_id)) elif ts.programslot.show.automation_id: schedule.append((ts.start.strftime('%H:%M:%S'), ts.programslot.show.name, ts.programslot.show.automation_id)) else: schedule.append((ts.start.strftime('%H:%M:%S'), ts.programslot.show.name, -1)) return HttpResponse(json.dumps(schedule, ensure_ascii=False, encoding='utf8').encode('utf8'), content_type="application/json; charset=utf-8")