29 lines
659 B
Python
29 lines
659 B
Python
from django.contrib import admin
|
|
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
|
|
|
|
from notes.models import Note, User
|
|
|
|
# Register your models here.
|
|
#
|
|
|
|
|
|
@admin.register(User)
|
|
class UserAdmin(BaseUserAdmin):
|
|
fieldsets = [
|
|
*BaseUserAdmin.fieldsets,
|
|
("Notes", {"fields": ("allow_notes_from", "expiry_seconds")}),
|
|
]
|
|
add_fieldsets = [
|
|
*BaseUserAdmin.add_fieldsets,
|
|
("Notes", {"fields": ("allow_notes_from", "expiry_seconds")}),
|
|
]
|
|
filter_horizontal = (
|
|
*BaseUserAdmin.filter_horizontal,
|
|
"allow_notes_from",
|
|
)
|
|
|
|
|
|
@admin.register(Note)
|
|
class NoteAdmin(admin.ModelAdmin):
|
|
pass
|