from django.contrib import admin

from users.models import Department
from users.models import DepartmentUser
from users.models import Lageruser


class LageruserAdmin(admin.ModelAdmin):
    search_fields = ('first_name', 'last_name', )
    list_display = ('username', '__str__', 'is_superuser', 'main_department')


class DepartmentUserAdmin(admin.ModelAdmin):
    search_fields = ('user__first_name', 'user__last_name', )


admin.site.register(Lageruser, LageruserAdmin)
admin.site.register(DepartmentUser, DepartmentUserAdmin)
admin.site.register(Department)
