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)