getting real.

1 files changed, 92 insertions, 0 deletions

diff --git a/program/views.py b/program/views.py
new file mode 100644
index 0000000..59e8fdf
--- /dev/null
+++ b/program/views.py
@@ -0,0 +1,92 @@
+from django.views.generic import list_detail, simple
+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
+
+def show_list(request):
+    if 'broadcastformat' in request.GET:
+        broadcastformat = get_object_or_404(BroadcastFormat, slug=request.GET['broadcastformat'])
+
+        queryset = Show.objects.filter(broadcastformat=broadcastformat)
+    elif 'musicfocus' in request.GET:
+        musicfocus = get_object_or_404(MusicFocus, slug=request.GET['musicfocus'])
+
+        queryset = Show.objects.filter(musicfocus=musicfocus)
+    elif 'showinformation' in request.GET:
+        showinformation = get_object_or_404(ShowInformation, slug=request.GET['showinformation'])
+
+        queryset = Show.objects.filter(showinformation=showinformation)
+    elif 'showtopic' in request.GET:
+        showtopic = get_object_or_404(ShowTopic, slug=request.GET['showtopic'])
+
+        queryset = Show.objects.filter(showtopic=showtopic)
+    else:
+        queryset = Show.objects.all()
+
+    return list_detail.object_list(request, queryset=queryset, template_object_name='show')
+
+def recommendations(request, template_name='program/recommendations.html'):
+    now = datetime.now()
+    in_one_week = now + timedelta(weeks=1)
+
+    queryset = Note.objects.filter(status=1, timeslot__start__range=(now, in_one_week))[:10]
+
+    return list_detail.object_list(request, queryset=queryset, template_name=template_name, template_object_name='recommendation')
+
+def 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(6, 0))
+    else:
+        today = datetime.strptime('%s__%s__%s__06__00' % (year, month, day), '%Y__%m__%d__%H__%M')
+
+    tomorrow = today+timedelta(days=1)
+
+    recommendations = Note.objects.filter(status=1, timeslot__start__range=(today, tomorrow))
+
+    extra_context = dict(day=today, recommendations=recommendations)
+
+    if 'broadcastformat' in request.GET:
+        broadcastformat = get_object_or_404(BroadcastFormat, slug=request.GET['broadcastformat'])
+
+        extra_context['timeslots'] = TimeSlot.objects.filter(start__range=(today, tomorrow), show__broadcastformat=broadcastformat)
+    else:
+        extra_context['timeslots'] = TimeSlot.objects.filter(start__range=(today, tomorrow))
+
+    return simple.direct_to_template(request, extra_context=extra_context, template='program/day_schedule.html')
+
+def current_show(request):
+    current = TimeSlot.objects.get_or_create_current()
+    next = current.get_next_by_start()
+    after_next = next.get_next_by_start()
+
+    extra_context = dict(current=current, next=next, after_next=after_next)
+
+    return simple.direct_to_template(request, template='program/boxes/current.html', extra_context=extra_context)
+
+def week_schedule(request, year=None, week=None):
+    if year is None and week is None:
+        year, week = datetime.strftime(datetime.today(), '%Y__%W').split('__')
+
+    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)
+
+    extra_context = dict(monday=monday, tuesday=tuesday, wednesday=wednesday, thursday=thursday, friday=friday, saturday=saturday, sunday=sunday)
+
+    extra_context['monday_timeslots'] = TimeSlot.objects.filter(start__range=(monday, tuesday))
+    extra_context['tuesday_timeslots'] = TimeSlot.objects.filter(start__range=(tuesday, wednesday))
+    extra_context['wednesday_timeslots'] = TimeSlot.objects.filter(start__range=(wednesday, thursday))
+    extra_context['thursday_timeslots'] = TimeSlot.objects.filter(start__range=(thursday, friday))
+    extra_context['friday_timeslots'] = TimeSlot.objects.filter(start__range=(friday, saturday))
+    extra_context['saturday_timeslots'] = TimeSlot.objects.filter(start__range=(saturday, sunday))
+    extra_context['sunday_timeslots'] = TimeSlot.objects.filter(start__range=(sunday, next_monday))
+
+    return simple.direct_to_template(request, template='program/week_schedule.html', extra_context=extra_context)