mirror of
https://github.com/lemeow125/StudE-Backend.git
synced 2024-11-17 14:29:25 +08:00
34 lines
1,013 B
Python
34 lines
1,013 B
Python
from django import forms
|
|
from django.contrib import admin
|
|
from django.contrib.auth.admin import UserAdmin
|
|
from .models import CustomUser
|
|
from year_levels.models import Year_Level
|
|
from semesters.models import Semester
|
|
from courses.models import Course
|
|
|
|
|
|
class CustomUserForm(forms.ModelForm):
|
|
year_level = forms.ModelChoiceField(
|
|
queryset=Year_Level.objects.all(), required=False)
|
|
semester = forms.ModelChoiceField(
|
|
queryset=Semester.objects.all(), required=False)
|
|
course = forms.ModelChoiceField(
|
|
queryset=Course.objects.all(), required=False)
|
|
avatar = forms.ImageField(required=False)
|
|
|
|
class Meta:
|
|
model = CustomUser
|
|
fields = '__all__'
|
|
|
|
|
|
class CustomUserAdmin(UserAdmin):
|
|
model = CustomUser
|
|
form = CustomUserForm
|
|
|
|
fieldsets = UserAdmin.fieldsets + (
|
|
(None, {'fields': ('student_id_number',
|
|
'year_level', 'semester', 'course', 'avatar', 'is_student', 'is_banned')}),
|
|
)
|
|
|
|
|
|
admin.site.register(CustomUser, CustomUserAdmin)
|