-
Notifications
You must be signed in to change notification settings - Fork 2.3k
/
admin.py
40 lines (31 loc) · 1.57 KB
/
admin.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
"""This file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license.
"""
from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from django.conf import settings
from django.contrib.auth.models import Group
from users.models import User
from projects.models import Project
from ml.models import MLBackend, MLBackendTrainJob
from tasks.models import Task, Annotation, Prediction
from organizations.models import Organization, OrganizationMember
class UserAdminShort(UserAdmin):
def __init__(self, *args, **kwargs):
super(UserAdminShort, self).__init__(*args, **kwargs)
# we have empty username - remove it to escape confuse about empty fields in admin web
self.list_display = [l for l in self.list_display if l != 'username']
self.fieldsets = ((None, {'fields': ('password', )}),
('Personal info', {'fields': ('first_name', 'last_name', 'email')}),
('Permissions', {'fields': ('is_active', 'is_staff', 'is_superuser',)}),
('Important dates', {'fields': ('last_login', 'date_joined')}))
admin.site.register(User, UserAdminShort)
admin.site.register(Project)
admin.site.register(MLBackend)
admin.site.register(MLBackendTrainJob)
admin.site.register(Task)
admin.site.register(Annotation)
admin.site.register(Prediction)
admin.site.register(Organization)
admin.site.register(OrganizationMember)
# remove unused django groups
admin.site.unregister(Group)