feat: reset password working

web/views/users.py

@@ -1,16 +1,19 @@
-from django.contrib.auth import login as login_user, authenticate
-from django.contrib.auth import logout as logout_user
+from django.conf import settings
 from django.contrib.auth import get_user_model
+from django.contrib.auth import login as login_user
+from django.contrib.auth import logout as logout_user
+from django.contrib.auth.tokens import default_token_generator
+from django.core.exceptions import ValidationError
 from django.db import IntegrityError
 from django.shortcuts import redirect, render, reverse
+from django.utils.http import url_has_allowed_host_and_scheme, urlsafe_base64_decode
 from django.utils.text import gettext_lazy as _
 from django.views.generic import TemplateView, View
 
 from shop.models import CustomerAddress
-from users.forms import LoginForm, RegisterForm
+from users.forms import LoginForm, PasswordResetForm, RegisterForm, SetPasswordForm
 from web.models import WebSettings
 
-
 User = get_user_model()
 
 
@@ -123,7 +126,108 @@ class LogoutView(View, RedirectionMixin):
 
 
 class ResetPasswordView(TemplateView):
-    template_name = "web/reset-password.html"
+    template_name = "users/reset_password.html"
+    form_class = PasswordResetForm
+    redirect_to = "web:reset_password_email_sent"
+    email_reset_password_template_name = "mails/reset_password.txt"
+    html_reset_password_template_name = "mails/reset_password.html"
+    subject_template_name = "mails/subject_reset_password.txt"
+
+    def get_email_options(self):
+        return {
+            "extra_email_context": {},
+            "email_template_name": self.email_reset_password_template_name,
+            "html_email_template_name": self.html_reset_password_template_name,
+            "subject_template_name": self.subject_template_name,
+        }
+
+    def get_base_context(self):
+        settings = WebSettings.load()
+
+        return {
+            "title": settings.web_title,
+            "description": _("Restablecer contraseña"),
+        }
+
+    def get_context_data(self, **kwargs):
+        context = self.get_base_context()
+        context.update({"form": self.form_class()})
+        return context
+
+    def post(self, request, *args, **kwargs):
+        form = self.form_class(request.POST)
+
+        if form.is_valid():
+            opts = {
+                "use_https": request.is_secure(),
+                "from_email": getattr(settings, "DEFAULT_FROM_EMAIL"),
+                "request": request,
+            }
+
+            opts.update(self.get_email_options())
+            form.save(**opts)
+            return redirect(self.redirect_to)
+
+        context = self.get_base_context()
+        context.update({"form": form})
+
+        return render(request, self.template_name, context)
+
+
+class ResetPasswordEmailSent(TemplateView):
+    template_name = "users/reset_password_email_sent.html"
+
+
+class ResetPasswordConfirmView(TemplateView):
+    template_name = "users/reset_password_confirm.html"
+    form_class = SetPasswordForm
+    redirect_to = "web:login"
+    token_generator = default_token_generator
+
+    def get_user(self):
+        try:
+            uidb64 = self.kwargs.get("uidb64")
+            uid = urlsafe_base64_decode(uidb64).decode()
+            user = User.objects.get(pk=uid)
+        except (
+            TypeError,
+            ValueError,
+            OverflowError,
+            User.DoesNotExist,
+            ValidationError,
+        ):
+            user = None
+        return user
+
+    def generate_new_token_for_user(self, user):
+        return self.token_generator.make_token(user)
+
+    def get_context_data(self, **kwargs):
+        user = self.get_user()
+        valid_token = self.check_token(user)
+        new_token = self.generate_new_token_for_user(user)
+
+        return {
+            "title": WebSettings.load().web_title,
+            "description": _("Establecer nueva contraseña"),
+            "form": self.form_class(user, {'token': new_token}),
+            "valid_token": valid_token,
+            "new_token": new_token,
+        }
+
+    def check_token(self, user):
+        token = self.kwargs.get("token")
+        valid_token = self.token_generator.check_token(user, token)
+        return valid_token
+
+    def post(self, request, *args, **kwargs):
+        user = self.get_user()
+        form = self.form_class(user, request.POST)
+
+        if form.is_valid():
+            form.save()
+            return redirect(self.redirect_to)
+        return render(request, self.template_name, {})
 
 
 class MyAccountView(TemplateView):
@@ -143,4 +247,7 @@ logout = LogoutView.as_view()
 register = RegisterView.as_view()
 reset_password = ResetPasswordView.as_view()
 
+reset_password_confirm = ResetPasswordConfirmView.as_view()
+reset_password_email_sent = ResetPasswordEmailSent.as_view()
+
 my_account = MyAccountView.as_view()