Renew with 2FA and Podman

core/views/base.py

@@ -1,6 +1,6 @@
 import logging
 
-import stripe
+# import stripe
 from django.conf import settings
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.http import JsonResponse
@@ -10,8 +10,7 @@ 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
+from core.lib.notify import raw_sendmsg
 
 logger = logging.getLogger(__name__)
 
@@ -25,81 +24,22 @@ class Home(View):
         return render(request, self.template_name)
 
 
-class Billing(LoginRequiredMixin, View):
-    template_name = "billing.html"
-
-    def get(self, request):
-        if not settings.STRIPE_ENABLED:
-            return redirect(reverse("home"))
-        plans = Plan.objects.all()
-        user_plans = request.user.plans.all()
-        context = {"plans": plans, "user_plans": user_plans}
-        return render(request, self.template_name, context)
-
-
-class Order(LoginRequiredMixin, View):
-    def get(self, request, plan_name):
-        if not settings.STRIPE_ENABLED:
-            return redirect(reverse("home"))
-        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": 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)
-            Session.objects.create(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):
-    def get(self, request, plan_name):
-        if not settings.STRIPE_ENABLED:
-            return redirect(reverse("home"))
-        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")
+    success_url = reverse_lazy("two_factor:login")
     template_name = "registration/signup.html"
 
+    def form_valid(self, form):
+        """If the form is valid, save the associated model."""
+        self.object = form.save()
+        raw_sendmsg(
+            f"New user signup: {self.object.username} - {self.object.email}",
+            title="New user",
+            topic=settings.NOTIFY_TOPIC,
+        )
+        return super().form_valid(form)
+
     def get(self, request, *args, **kwargs):
         if not settings.REGISTRATION_OPEN:
             return render(request, "registration/registration_closed.html")
-        super().get(request, *args, **kwargs)
-
-
-class Portal(LoginRequiredMixin, View):
-    def get(self, request):
-        if not settings.STRIPE_ENABLED:
-            return redirect(reverse("home"))
-        session = stripe.billing_portal.Session.create(
-            customer=request.user.stripe_id,
-            return_url=request.build_absolute_uri(reverse("billing")),
-        )
-        return redirect(session.url)
+        return super().get(request, *args, **kwargs)