41 lines
1014 B
Python
41 lines
1014 B
Python
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)
|