from django.contrib import admin

from models import Event


class EventAdmin(admin.ModelAdmin):
    list_display = ('name', 'start', 'end', 'source')
    list_filter = ('source',)
    search_fields = ['name', 'description']
    date_hierachy = 'start'

    def save_model(self, request, obj, form, change):
        """Sets the source to "admin".
        """
        if not obj.id:
            obj.source = Event.SOURCE_ADMIN
        obj.save()


admin.site.register(Event, EventAdmin)