18 lines
644 B
Python
18 lines
644 B
Python
from django.contrib import admin
|
|
from django.contrib.auth.admin import UserAdmin
|
|
from .models import CustomUser, AbsencePeriod
|
|
|
|
class CustomUserAdmin(UserAdmin):
|
|
model = CustomUser
|
|
list_display = ['email', 'username', 'team', 'display_parents', 'is_staff']
|
|
fieldsets = UserAdmin.fieldsets + (
|
|
(None, {'fields': ('team', 'parents')}),
|
|
)
|
|
filter_horizontal = ('parents',)
|
|
|
|
def display_parents(self, obj):
|
|
return ", ".join([parent.username for parent in obj.parents.all()])
|
|
display_parents.short_description = 'Parents'
|
|
|
|
admin.site.register(CustomUser, CustomUserAdmin)
|
|
admin.site.register(AbsencePeriod) |