from django.contrib import admin
from django.utils.translation import ugettext_lazy as _

from models import BroadcastFormat, MusicFocus, ShowInformation, ShowTopic, Language, Host, Note, ProgramSlot, Show, TimeSlot
from forms import MusicFocusForm

from datetime import date, datetime, timedelta


class ActivityFilter(admin.SimpleListFilter):
    title = _("Activity")

    def lookups(self, request, model_admin):
        return (
            ('yes', _("active")),
            ('no', _("inactive"))
        )

    def queryset(self, request, queryset):
        if self.parameter_name == 'has_timeslots':  # active/inactive ProgramSlots
            if self.value() == 'yes':
                return queryset.filter(until__gt=datetime.now).distinct()
            if self.value() == 'no':
                return queryset.filter(until__lt=datetime.now).distinct()

        if self.parameter_name == 'has_programslots_timeslots':  # active/inactive Shows
            if self.value() == 'yes':
                return queryset.filter(programslots__until__gt=datetime.now).distinct()
            if self.value() == 'no':
                return queryset.filter(programslots__until__lt=datetime.now).distinct()

        if self.parameter_name == 'has_shows_programslots_timeslots':  # active/inactive Hosts
            if self.value() == 'yes':
                return queryset.filter(shows__programslots__until__gt=datetime.now).distinct()
            if self.value() == 'no':
                return queryset.filter(shows__programslots__until__lt=datetime.now).distinct()


class ActiveProgramSlotsFilter(ActivityFilter):
    parameter_name = 'has_timeslots'


class ActiveShowsFilter(ActivityFilter):
    parameter_name = 'has_programslots_timeslots'


class ActiveHostsFilter(ActivityFilter):
    parameter_name = 'has_shows_programslots_timeslots'


class BroadcastFormatAdmin(admin.ModelAdmin):
    list_display = ('format', 'admin_color', 'enabled')
    prepopulated_fields = {'slug': ('format',)}


class MusicFocusAdmin(admin.ModelAdmin):
    form = MusicFocusForm
    list_display = ('focus', 'abbrev', 'admin_buttons')
    prepopulated_fields = {'slug': ('focus',)}


class ShowInformationAdmin(admin.ModelAdmin):
    list_display = ('information', 'abbrev', 'admin_buttons')
    prepopulated_fields = {'slug': ('information',)}


class ShowTopicAdmin(admin.ModelAdmin):
    list_display = ('topic', 'abbrev', 'admin_buttons')
    prepopulated_fields = {'slug': ('topic',)}


class LanguageAdmin(admin.ModelAdmin):
    list_display = ('slug', 'name', 'native_name')


class HostAdmin(admin.ModelAdmin):
    list_display = ('name',)
    list_filter = (ActiveHostsFilter, 'is_always_visible',)


class NoteAdmin(admin.ModelAdmin):
    date_hierarchy = 'start'
    list_display = ('title', 'show', 'start', 'status')
    list_filter = ('status',)
    ordering = ('timeslot',)
    save_as = True

    def get_queryset(self, request):
        shows = request.user.shows.all()
        return super(NoteAdmin, self).get_queryset(request).filter(show__in=shows)

    def formfield_for_foreignkey(self, db_field, request=None, **kwargs):
        four_weeks_ago = datetime.now() - timedelta(weeks=4)
        in_twelf_weeks = datetime.now() + timedelta(weeks=12)
        if db_field.name == 'timeslot':
            try:
                timeslot_id = int(request.get_full_path().split('/')[-2])
            except ValueError:
                shows = request.user.shows.all()
                kwargs['queryset'] = TimeSlot.objects.filter(show__in=shows, note__isnull=True, start__gt=four_weeks_ago,
                                                             start__lt=in_twelf_weeks)
            else:
                kwargs['queryset'] = TimeSlot.objects.filter(note=timeslot_id)

        return super(NoteAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)

    def save_model(self, request, obj, form, change):
        obj.save()


class TimeSlotInline(admin.TabularInline):
    model = TimeSlot
    ordering = ('-end',)


class ProgramSlotAdmin(admin.ModelAdmin):
    actions = ('renew',)
    inlines = (TimeSlotInline,)
    fields = (('rrule', 'byweekday'), ('dstart', 'tstart', 'tend'), 'until', 'is_repetition', 'automation_id')
    list_display = ('get_show_name', 'byweekday', 'rrule', 'tstart', 'tend', 'until')
    list_filter = (ActiveProgramSlotsFilter, 'byweekday', 'rrule', 'is_repetition')
    ordering = ('byweekday', 'dstart')
    save_on_top = True
    search_fields = ('show__name',)

    def renew(self, request, queryset):
        next_year = date.today().year + 1
        until = date(next_year, 12, 31)
        renewed = queryset.update(until=until)
        if renewed == 1:
            message = _("1 program slot was renewed until %s") % until
        else:
            message = _("%s program slots were renewed until %s") % (renewed, until)
        self.message_user(request, message)
    renew.short_description = _("Renew selected program slots")

    def get_show_name(self, obj):
        return obj.show.name
    get_show_name.admin_order_field = 'show'
    get_show_name.short_description = "Show"


class ProgramSlotInline(admin.TabularInline):
    model = ProgramSlot
    ordering = ('-until',)


class ShowAdmin(admin.ModelAdmin):
    filter_horizontal = ('hosts', 'owners', 'musicfocus', 'showinformation', 'showtopic', 'language')
    inlines = (ProgramSlotInline,)
    list_display = ('name', 'short_description')
    list_filter = (ActiveShowsFilter, 'broadcastformat', 'showinformation', 'showtopic', 'musicfocus', 'language')
    ordering = ('slug',)
    prepopulated_fields = {'slug': ('name',)}
    search_fields = ('name', 'short_description', 'description')
    fields = (
        'predecessor', 'broadcastformat', 'name', 'slug', 'image', 'short_description', 'description',
        'email', 'website', 'hosts', 'owners', 'showinformation', 'showtopic', 'language',
        'musicfocus',
    )

    def formfield_for_foreignkey(self, db_field, request=None, **kwargs):
        try:
            show_id = int(request.get_full_path().split('/')[-2])
        except ValueError:
            show_id = None

        if db_field.name == 'predecessor' and show_id:
            kwargs['queryset'] = Show.objects.exclude(pk=show_id)

        return super(ShowAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)


admin.site.register(BroadcastFormat, BroadcastFormatAdmin)
admin.site.register(MusicFocus, MusicFocusAdmin)
admin.site.register(ShowInformation, ShowInformationAdmin)
admin.site.register(ShowTopic, ShowTopicAdmin)
admin.site.register(Language, LanguageAdmin)
admin.site.register(Host, HostAdmin)
admin.site.register(Note, NoteAdmin)
admin.site.register(ProgramSlot, ProgramSlotAdmin)
admin.site.register(Show, ShowAdmin)