Add models, views and forms for authentication

2022-07-21 13:45:44 +01:00
parent adf7c0604a
commit 4efc10a4f9
9 changed files with 297 additions and 0 deletions
--- a/core/admin.py
+++ b/core/admin.py
@@ -0,0 +1,39 @@
+from django.contrib import admin
+from django.contrib.auth.admin import UserAdmin
+
+from .forms import CustomUserCreationForm
+from .models import Product, User
+
+
+# Register your models here.
+class CustomUserAdmin(UserAdmin):
+    list_filter = ["plans"]
+    model = User
+    add_form = CustomUserCreationForm
+    fieldsets = (
+        *UserAdmin.fieldsets,
+        (
+            "Stripe information",
+            {
+                "fields": (
+                    "stripe_id",
+                    "subscription_id",
+                )
+            },
+        ),
+        (
+            "Payment information",
+            {
+                "fields": (
+                    "subscription_active",
+                    "paid",
+                    "plans",
+                    "last_payment",
+                )
+            },
+        ),
+    )
+
+
+admin.site.register(User, CustomUserAdmin)
+admin.site.register(Product)