neptune/core/admin.py

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin

from .forms import CustomUserCreationForm
from .models import Plan, Session, User


# 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)