Initial commit

core/admin.py

@@ -0,0 +1,33 @@
+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)