from django.contrib import admin
from .models import Employee, Shift, Location, Event, Helper, Reinigung


class HelperAdmin(admin.ModelAdmin):
    list_display = ('date', 'ben', 'best', 'info')
    search_fields = ('date', 'info')

class ReinigungAdmin(admin.ModelAdmin):
    list_display = ('date', 'auftrag', 'info')
    search_fields = ('date', 'info')


class EmployeeAdmin(admin.ModelAdmin):
    list_display = ('name', 'daily_workhours', 'info')
    search_fields = ('name',)


class ShiftAdmin(admin.ModelAdmin):
    list_display = ('date', 'start', 'end', 'employee')
    list_filter = ('date', 'shifttype', 'employee')
    search_fields = ('date','employee')


class EventAdmin(admin.ModelAdmin):
    list_display = ('date', 'name', 'event_type')
    list_filter = ('date', 'name', 'event_type')
    search_fields = ('date', 'name', 'location', 'event_type')

admin.site.register(Employee, EmployeeAdmin)
admin.site.register(Shift, ShiftAdmin)
admin.site.register(Event, EventAdmin)
admin.site.register(Location)
admin.site.register(Helper, HelperAdmin)
admin.site.register(Reinigung, ReinigungAdmin)