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)