from django.contrib import admin from django.contrib.auth.admin import UserAdmin from django_otp.admin import OTPAdminSite from .forms import CustomUserCreationForm from .models import Plan, Session, User 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, ( "Stripe information", {"fields": ("stripe_id",)}, ), ( "Payment information", { "fields": ( "plans", "last_payment", ) }, ), ) admin.site.register(User, CustomUserAdmin) admin.site.register(Plan) admin.site.register(Session)