74 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			74 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| """passbook core user views"""
 | |
| from django.contrib import messages
 | |
| from django.contrib.auth import logout, update_session_auth_hash
 | |
| from django.contrib.auth.mixins import LoginRequiredMixin
 | |
| from django.contrib.messages.views import SuccessMessageMixin
 | |
| from django.forms.utils import ErrorList
 | |
| from django.shortcuts import redirect, reverse
 | |
| from django.urls import reverse_lazy
 | |
| from django.utils.translation import gettext as _
 | |
| from django.views.generic import DeleteView, FormView, UpdateView
 | |
| 
 | |
| from passbook.core.forms.users import PasswordChangeForm, UserDetailForm
 | |
| from passbook.factors.password.exceptions import PasswordPolicyInvalid
 | |
| from passbook.lib.config import CONFIG
 | |
| 
 | |
| 
 | |
| class UserSettingsView(SuccessMessageMixin, LoginRequiredMixin, UpdateView):
 | |
|     """Update User settings"""
 | |
| 
 | |
|     template_name = "user/settings.html"
 | |
|     form_class = UserDetailForm
 | |
| 
 | |
|     success_message = _("Successfully updated user.")
 | |
|     success_url = reverse_lazy("passbook_core:user-settings")
 | |
| 
 | |
|     def get_object(self):
 | |
|         return self.request.user
 | |
| 
 | |
| 
 | |
| class UserDeleteView(LoginRequiredMixin, DeleteView):
 | |
|     """Delete user account"""
 | |
| 
 | |
|     template_name = "generic/delete.html"
 | |
| 
 | |
|     def get_object(self):
 | |
|         return self.request.user
 | |
| 
 | |
|     def get_success_url(self):
 | |
|         messages.success(self.request, _("Successfully deleted user."))
 | |
|         logout(self.request)
 | |
|         return reverse("passbook_core:auth-login")
 | |
| 
 | |
| 
 | |
| class UserChangePasswordView(LoginRequiredMixin, FormView):
 | |
|     """View for users to update their password"""
 | |
| 
 | |
|     form_class = PasswordChangeForm
 | |
|     template_name = "login/form_with_user.html"
 | |
| 
 | |
|     def form_valid(self, form: PasswordChangeForm):
 | |
|         try:
 | |
|             # user.set_password checks against Policies so we don't need to manually do it here
 | |
|             self.request.user.set_password(form.cleaned_data.get("password"))
 | |
|             self.request.user.save()
 | |
|             update_session_auth_hash(self.request, self.request.user)
 | |
|             messages.success(self.request, _("Successfully changed password"))
 | |
|         except PasswordPolicyInvalid as exc:
 | |
|             # Manually inject error into form
 | |
|             # pylint: disable=protected-access
 | |
|             errors = form._errors.setdefault("password_repeat", ErrorList(""))
 | |
|             # pylint: disable=protected-access
 | |
|             errors = form._errors.setdefault("password", ErrorList())
 | |
|             for error in exc.messages:
 | |
|                 errors.append(error)
 | |
|             return self.form_invalid(form)
 | |
|         return redirect("passbook_core:overview")
 | |
| 
 | |
|     def get_context_data(self, **kwargs):
 | |
|         kwargs["config"] = CONFIG.y("passbook")
 | |
|         kwargs["is_login"] = True
 | |
|         kwargs["title"] = _("Change Password")
 | |
|         kwargs["primary_action"] = _("Change")
 | |
|         return super().get_context_data(**kwargs)
 | 
