Plankton_Detector/PlanktonDetector/UserManagement/views.py
2024-01-06 14:14:15 +01:00

57 lines
1.9 KiB
Python

from django.shortcuts import render, redirect
from django.contrib.auth.forms import AuthenticationForm, UserCreationForm
from django.contrib.auth import authenticate, login, logout
from django.urls import reverse_lazy
from django.views import View
class LoginView(View):
form_class = AuthenticationForm
template_name = "login_signup.html"
def get(self, request, *args, **kwargs):
form = self.form_class()
return render(request, self.template_name, {"form": form})
def post(self, request, *args, **kwargs):
redirection_path = request.GET.get("next")
form = self.form_class(data=request.POST)
if form.is_valid():
user = authenticate(
username=form.cleaned_data["username"],
password=form.cleaned_data["password"],
)
if user is not None:
login(request, user)
if redirection_path is not None:
return redirect(redirection_path)
else:
return redirect("detect/")
return render(request, self.template_name, {"form": form})
class SignupView(View):
form_class = UserCreationForm
template_name = "login_signup.html"
def get(self, request, *args, **kwargs):
form = self.form_class()
return render(request, self.template_name, {"form": form, "signup": True})
def post(self, request, *args, **kwargs):
form = self.form_class(data=request.POST)
if form.is_valid():
user = form.save()
login(request, user)
return redirect("detect/")
return render(request, self.template_name, {"form": form, "signup": True})
def logout_view(request):
redirection_path = request.GET.get("next")
logout(request)
if redirection_path is not None:
return redirect(redirection_path)
else:
return redirect("detect/")