from django.contrib import admin
from .models import Employee, Shift


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')


admin.site.register(Employee, EmployeeAdmin)
admin.site.register(Shift, ShiftAdmin)