Upload template project

core/views/base.py

@@ -0,0 +1,95 @@
+import logging
+
+import stripe
+from asgiref.sync import sync_to_async
+from django.conf import settings
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.http import JsonResponse
+from django.shortcuts import redirect, render
+from django.urls import reverse, reverse_lazy
+from django.views import View
+from django.views.generic.edit import CreateView
+
+from core.forms import NewUserForm
+from core.lib.products import assemble_plan_map
+from core.models import Plan, Session
+
+logger = logging.getLogger(__name__)
+
+# Create your views here
+
+
+class Home(View):
+    template_name = "index.html"
+
+    async def get(self, request):
+        return render(request, self.template_name)
+
+
+class Billing(LoginRequiredMixin, View):
+    template_name = "billing.html"
+
+    async def get(self, request):
+        plans = await sync_to_async(list)(Plan.objects.all())
+        user_plans = await sync_to_async(list)(request.user.plans.all())
+        context = {"plans": plans, "user_plans": user_plans}
+        return render(request, self.template_name, context)
+
+
+class Order(LoginRequiredMixin, View):
+    async def get(self, request, plan_name):
+        plan = Plan.objects.get(name=plan_name)
+        try:
+            cast = {
+                "payment_method_types": settings.ALLOWED_PAYMENT_METHODS,
+                "mode": "subscription",
+                "customer": request.user.stripe_id,
+                "line_items": await assemble_plan_map(
+                    product_id_filter=plan.product_id
+                ),
+                "success_url": request.build_absolute_uri(reverse("success")),
+                "cancel_url": request.build_absolute_uri(reverse("cancel")),
+            }
+            if request.user.is_superuser:
+                cast["discounts"] = [{"coupon": settings.STRIPE_ADMIN_COUPON}]
+            session = stripe.checkout.Session.create(**cast)
+            await Session.objects.acreate(user=request.user, session=session.id)
+            return redirect(session.url)
+            # return JsonResponse({'id': session.id})
+        except Exception as e:
+            # Raise a server error
+            return JsonResponse({"error": str(e)}, status=500)
+
+
+class Cancel(LoginRequiredMixin, View):
+    async def get(self, request, plan_name):
+        plan = Plan.objects.get(name=plan_name)
+        try:
+            subscriptions = stripe.Subscription.list(
+                customer=request.user.stripe_id, price=plan.product_id
+            )
+            for subscription in subscriptions["data"]:
+                items = subscription["items"]["data"]
+                for item in items:
+                    stripe.Subscription.delete(item["subscription"])
+                    return render(request, "subscriptioncancel.html", {"plan": plan})
+                    # return JsonResponse({'id': session.id})
+        except Exception as e:
+            # Raise a server error
+            logging.error(f"Error cancelling subscription for user: {e}")
+            return JsonResponse({"error": "True"}, status=500)
+
+
+class Signup(CreateView):
+    form_class = NewUserForm
+    success_url = reverse_lazy("login")
+    template_name = "registration/signup.html"
+
+
+class Portal(LoginRequiredMixin, View):
+    async def get(self, request):
+        session = stripe.billing_portal.Session.create(
+            customer=request.user.stripe_id,
+            return_url=request.build_absolute_uri(reverse("billing")),
+        )
+        return redirect(session.url)