From d0926a8f86bd79613e3ed9cda98cb9d0c3eae29e Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@usody.com>
Date: Mon, 9 Oct 2023 10:49:56 +0200
Subject: [PATCH] add login

---
 idhub/templates/auth/login.html |  2 ++
 idhub/views.py                  | 35 +++++++++------------------------
 2 files changed, 11 insertions(+), 26 deletions(-)

diff --git a/idhub/templates/auth/login.html b/idhub/templates/auth/login.html
index 287ca73..6f43260 100644
--- a/idhub/templates/auth/login.html
+++ b/idhub/templates/auth/login.html
@@ -115,6 +115,8 @@
                         {% endfor %}
                         {% endif %}
 
+                        <input name="next" type="hidden" value="{{ success_url }}">
+
                         <div class="form-actions-no-box">
                             <input type="submit" name="submit" value="{% trans 'Log in' %}"
                                 class="btn btn-primary form-control" id="submit-id-submit">
diff --git a/idhub/views.py b/idhub/views.py
index 2ba5524..8ba1fd3 100644
--- a/idhub/views.py
+++ b/idhub/views.py
@@ -15,6 +15,7 @@ from django.views.generic.base import RedirectView, TemplateView
 from django.views.generic.detail import DetailView
 from django.views.generic.edit import DeleteView, FormView
 from django.views.generic.list import ListView
+from django.contrib.auth import views as auth_views
 from django.contrib.auth.models import User
 
 from .forms import LoginForm
@@ -24,11 +25,10 @@ logger = logging.getLogger(__name__)
 
 
 
-# class UserDashboardView(LoginRequiredMixin, TemplateView):
-class UserDashboardView(TemplateView):
+class UserDashboardView(LoginRequiredMixin, TemplateView):
     template_name = "idhub/user_dashboard.html"
     title = _('Dashboard')
-    #login_url = "/login/"
+    login_url = "/login/"
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
@@ -42,31 +42,14 @@ class AdminDashboardView(UserDashboardView):
     template_name = "idhub/admin_dashboard.html"
 
 
-class LoginView(FormView):
+class LoginView(auth_views.LoginView):
     template_name = 'auth/login.html'
-    form_class = LoginForm
-    success_url = reverse_lazy('idhub:user_dashboard')
     extra_context = {
         'title': _('Login'),
+        'success_url': reverse_lazy('idhub:user_dashboard'),
     }
 
-    def form_valid(self, form):
-        return super().form_valid(form)
-
-
-class LogoutView(RedirectView):
-    """
-    Log out the user.
-    """
-    permanent = False
-    pattern_name = 'login'
-
-    def get_redirect_url(self, *args, **kwargs):
-        """
-        Logs out the user.
-        """
-        return super().get_redirect_url(*args, **kwargs)
-
-    def post(self, request, *args, **kwargs):
-        """Logout may be done via POST."""
-        return self.get(request, *args, **kwargs)
+    def get(self, request):
+        if request.GET.get('next'):
+            self.extra_context['success_url'] = request.GET.get('next')
+        return super().get(request)