from django.shortcuts import render from .forms import RegisterForm, LoginForm from django.contrib.auth import authenticate, login, logout # from django.views.generic.edit import FormView from django.views.generic import FormView, RedirectView from django.contrib.auth import get_user_model from django.http import HttpResponseRedirect from django.urls import reverse from django.contrib.auth.forms import AuthenticationForm, UserCreationForm from django.contrib.auth import REDIRECT_FIELD_NAME from django.views.decorators.csrf import csrf_protect from django.contrib import auth from django.views.decorators.cache import never_cache from django.shortcuts import redirect from django.utils.decorators import method_decorator from django.views.decorators.debug import sensitive_post_parameters from django.utils.http import is_safe_url # Create your views here. class RegisterView(FormView): form_class = RegisterForm template_name = 'account/registration_form.html' def form_valid(self, form): user = form.save(False) user.save(True) url = reverse('accounts:login') return HttpResponseRedirect(url) class LogoutView(RedirectView): url = '/login/' @method_decorator(never_cache) def dispatch(self, request, *args, **kwargs): return super(LogoutView, self).dispatch(request, *args, **kwargs) def get(self, request, *args, **kwargs): from website.utils import cache cache.clear() logout(request) return super(LogoutView, self).get(request, *args, **kwargs) class LoginView(FormView): form_class = LoginForm template_name = 'account/login.html' success_url = '/' redirect_field_name = REDIRECT_FIELD_NAME @method_decorator(sensitive_post_parameters('password')) @method_decorator(csrf_protect) @method_decorator(never_cache) def dispatch(self, request, *args, **kwargs): return super(LoginView, self).dispatch(request, *args, **kwargs) def get_context_data(self, **kwargs): redirect_to = self.request.GET.get(self.redirect_field_name) if redirect_to is None: redirect_to = '/' kwargs['redirect_to'] = redirect_to return super(LoginView, self).get_context_data(**kwargs) def form_valid(self, form): form = AuthenticationForm(data=self.request.POST, request=self.request) if form.is_valid(): from website.utils import cache if cache and cache is not None: cache.clear() print(self.redirect_field_name) redirect_to = self.request.GET.get(self.redirect_field_name) auth.login(self.request, form.get_user()) return super(LoginView, self).form_valid(form) # return HttpResponseRedirect('/') else: return self.render_to_response({ 'form': form }) @property def get_success_url(self): print(self.redirect_field_name) redirect_to = self.request.POST.get(self.redirect_field_name) if not is_safe_url(redirect_to, allowed_hosts=self.request.get_host()): redirect_to = self.success_url return redirect_to