mirror of
https://github.com/th3r00t/pyShelf.git
synced 2026-04-28 01:59:35 -04:00
30 lines
1.1 KiB
Python
Executable File
30 lines
1.1 KiB
Python
Executable File
from django.contrib import admin
|
|
from django.contrib.auth.admin import UserAdmin
|
|
|
|
from .models import Books, Collections, Favorites, Navigation, User
|
|
from .forms import CustomUserCreationForm, CustomUserChangeForm
|
|
|
|
|
|
class CustomUserAdmin(UserAdmin):
|
|
model = User
|
|
add_form = CustomUserCreationForm
|
|
form = CustomUserChangeForm
|
|
list_display = ["email", "username", "facebook", "twitter", "sponsorid", "matrixid"]
|
|
fieldsets = UserAdmin.fieldsets + (
|
|
(None, {"fields": ()}),
|
|
("Personal info", {"fields": ("facebook", "twitter", "matrixid")}),
|
|
("Permissions", {"fields": ("sponsorid",)}),
|
|
)
|
|
# add_fieldsets is not a standard ModelAdmin attribute. UserAdmin
|
|
# overrides get_fieldsets to use this attribute when creating a user.
|
|
add_fieldsets = UserAdmin.add_fieldsets + (
|
|
(None, {"classes": ("wide",), "fields": ("facebook", "twitter", "sponsorid", "matrixid")},),
|
|
)
|
|
|
|
|
|
admin.site.register(Books)
|
|
admin.site.register(Collections)
|
|
admin.site.register(Favorites)
|
|
admin.site.register(Navigation)
|
|
admin.site.register(User, CustomUserAdmin)
|