make login and register editable. (#5)

Author: Svtter

django_login_email/__init__.py

@@ -1 +1 @@
-__version__ = "0.5.0"
+__version__ = "0.5.1"

django_login_email/views/__init__.py

@@ -20,6 +20,7 @@ class EmailVerifyView(TemplateView, email.EmailVerifyMixin, MailRecordModelMixin
   """verify token in url"""
 
   tl = limit.LoginTimeLimit()
+  error_template: str = "login_email/error.html"
 
   def get(self, request: HttpRequest, *args: Any, **kwargs: Any) -> HttpResponse:
     token = request.GET.get("token", None)
@@ -28,7 +29,7 @@ def get(self, request: HttpRequest, *args: Any, **kwargs: Any) -> HttpResponse:
     try:
       self.verify_login_mail(request=request, token_v=token)
     except errors.ValidatedError as e:
-      return render(self.request, "login_email/error.html", {"error": e})
+      return render(self.request, self.error_template, {"error": e})
     except Exception as e:
       logger.error(e)
       raise Http404("Invalid Request")
@@ -39,9 +40,11 @@ def get_success_url(self):
 
 
 class EmailLogoutView(TemplateView, email.EmailLogoutMixin):
+  login_url: str = "login_email:login"
+
   def get(self, request: HttpRequest, *args: Any, **kwargs: Any) -> HttpResponse:
     self.logout(request=request)
-    return redirect("login_email:login")
+    return redirect(self.login_url)
 
 
 class HomeView(TemplateView):

django_login_email/views/login.py

@@ -18,6 +18,9 @@ class EmailLoginView(FormView, MailRecordModelMixin):
   """process login by email"""
 
   template_name = "login_email/login.html"
+  error_template: str = "login_email/error.html"
+  success_template: str = "login_email/success.html"
+
   form_class = forms.LoginForm
 
   login_info_class = email.EmailLoginInfo
@@ -42,5 +45,5 @@ def form_valid(self, form):
       self.send_login_mail(form.cleaned_data["email"])
     except Exception as e:
       logger.error(e)
-      return render(self.request, "login_email/error.html", {"error": e})
-    return render(self.request, "login_email/success.html", {"form": form})
+      return render(self.request, self.error_template, {"error": e})
+    return render(self.request, self.success_template, {"form": form})