34 lines
979 B
Python
34 lines
979 B
Python
|
from django.contrib import admin
|
||
|
|
from django.contrib.auth.admin import UserAdmin
|
||
|
|
|
||
|
|
from .forms import CustomUserCreationForm
|
||
|
|
from .models import NotificationSettings, User # AssetRestriction,; Plan,; Session,
|
||
|
|
|
||
|
|
# admin.site.__class__ = OTPAdminSite
|
||
|
|
|
||
|
|
# otp_admin_site = OTPAdminSite(OTPAdminSite.name)
|
||
|
|
# for model_cls, model_admin in admin.site._registry.items():
|
||
|
|
# otp_admin_site.register(model_cls, model_admin.__class__)
|
||
|
|
|
||
|
|
|
||
|
|
# Register your models here.
|
||
|
|
class CustomUserAdmin(UserAdmin):
|
||
|
|
# list_filter = ["plans"]
|
||
|
|
model = User
|
||
|
|
add_form = CustomUserCreationForm
|
||
|
|
fieldsets = (
|
||
|
|
*UserAdmin.fieldsets,
|
||
|
|
(
|
||
|
|
"Billing information",
|
||
|
|
{"fields": ("billing_provider_id", "payment_provider_id")},
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class NotificationSettingsAdmin(admin.ModelAdmin):
|
||
|
|
list_display = ("user", "ntfy_topic", "ntfy_url")
|
||
|
|
|
||
|
|
|
||
|
|
admin.site.register(User, CustomUserAdmin)
|
||
|
|
admin.site.register(NotificationSettings, NotificationSettingsAdmin)
|