mybuddy/babybuddy/views.py

154 lines
5.4 KiB
Python
Raw Normal View History

2017-11-09 12:24:32 +00:00
# -*- coding: utf-8 -*-
from django.contrib import messages
2017-12-02 21:20:15 +00:00
from django.contrib.auth import update_session_auth_hash
from django.contrib.auth.forms import PasswordChangeForm
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.auth.models import User
from django.contrib.messages.views import SuccessMessageMixin
2017-11-11 22:27:42 +00:00
from django.shortcuts import redirect, render
from django.urls import reverse, reverse_lazy
from django.utils.text import format_lazy
2020-02-14 17:22:25 +00:00
from django.utils.translation import gettext as _, gettext_lazy
2017-11-11 22:27:42 +00:00
from django.views.generic import View
2017-11-09 12:24:32 +00:00
from django.views.generic.base import TemplateView, RedirectView
from django.views.generic.edit import CreateView, UpdateView, DeleteView
from django.views.i18n import set_language
from django_filters.views import FilterView
2017-11-09 12:24:32 +00:00
from babybuddy import forms
from babybuddy.mixins import PermissionRequired403Mixin, StaffOnlyMixin
from babybuddy.models import user_logged_in_callback
2017-11-09 12:24:32 +00:00
class RootRouter(LoginRequiredMixin, RedirectView):
2017-11-11 22:27:42 +00:00
"""
Redirects to the site dashboard.
2017-11-11 22:27:42 +00:00
"""
2017-11-09 12:24:32 +00:00
def get_redirect_url(self, *args, **kwargs):
self.url = reverse('dashboard:dashboard')
2017-11-09 12:24:32 +00:00
return super(RootRouter, self).get_redirect_url(self, *args, **kwargs)
class BabyBuddyFilterView(FilterView):
"""
Disables "strictness" for django-filter. It is unclear from the
documentation exactly what this does...
"""
# TODO Figure out the correct way to use this.
strict = False
class UserList(StaffOnlyMixin, BabyBuddyFilterView):
model = User
template_name = 'babybuddy/user_list.html'
ordering = 'username'
paginate_by = 10
filterset_fields = ('username', 'first_name', 'last_name', 'email')
class UserAdd(StaffOnlyMixin, PermissionRequired403Mixin, SuccessMessageMixin,
CreateView):
model = User
template_name = 'babybuddy/user_form.html'
permission_required = ('admin.add_user',)
form_class = forms.UserAddForm
success_url = reverse_lazy('babybuddy:user-list')
success_message = gettext_lazy('User %(username)s added!')
class UserUpdate(StaffOnlyMixin, PermissionRequired403Mixin,
SuccessMessageMixin, UpdateView):
model = User
template_name = 'babybuddy/user_form.html'
permission_required = ('admin.change_user',)
form_class = forms.UserUpdateForm
success_url = reverse_lazy('babybuddy:user-list')
success_message = gettext_lazy('User %(username)s updated.')
class UserDelete(StaffOnlyMixin, PermissionRequired403Mixin,
DeleteView):
model = User
template_name = 'babybuddy/user_confirm_delete.html'
permission_required = ('admin.delete_user',)
success_url = reverse_lazy('babybuddy:user-list')
def delete(self, request, *args, **kwargs):
success_message = format_lazy(gettext_lazy(
'User {user} deleted.'), user=self.get_object()
)
messages.success(request, success_message)
return super(UserDelete, self).delete(request, *args, **kwargs)
2017-12-02 21:20:15 +00:00
class UserPassword(LoginRequiredMixin, View):
"""
Handles user password changes.
"""
form_class = forms.UserPasswordForm
2017-12-02 21:20:15 +00:00
template_name = 'babybuddy/user_password_form.html'
def get(self, request):
return render(request, self.template_name, {
'form': self.form_class(request.user)
})
def post(self, request):
form = PasswordChangeForm(request.user, request.POST)
if form.is_valid():
user = form.save()
update_session_auth_hash(request, user)
messages.success(request, _('Password updated.'))
2017-12-02 21:20:15 +00:00
return render(request, self.template_name, {'form': form})
2017-11-11 22:27:42 +00:00
class UserSettings(LoginRequiredMixin, View):
"""
Handles both the User and Settings models.
Based on this SO answer: https://stackoverflow.com/a/45056835.
"""
form_user_class = forms.UserForm
form_settings_class = forms.UserSettingsForm
2017-11-11 22:27:42 +00:00
template_name = 'babybuddy/user_settings_form.html'
def get(self, request):
return render(request, self.template_name, {
'form_user': self.form_user_class(instance=request.user),
'form_settings': self.form_settings_class(
instance=request.user.settings)
})
def post(self, request):
if request.POST.get('api_key_regenerate'):
request.user.settings.api_key(reset=True)
messages.success(request, _('User API key regenerated.'))
return redirect('babybuddy:user-settings')
2017-11-11 22:27:42 +00:00
form_user = self.form_user_class(
instance=request.user,
data=request.POST)
form_settings = self.form_settings_class(
instance=request.user.settings,
data=request.POST)
if form_user.is_valid() and form_settings.is_valid():
user = form_user.save(commit=False)
user_settings = form_settings.save(commit=False)
user.settings = user_settings
user.save()
user_logged_in_callback(UserSettings, request, user)
messages.success(request, _('Settings saved!'))
return set_language(request)
2017-11-11 22:27:42 +00:00
return render(request, self.template_name, {
'user_form': form_user,
'settings_form': form_settings
})
2017-11-09 12:24:32 +00:00
class Welcome(LoginRequiredMixin, TemplateView):
2017-11-11 22:27:42 +00:00
"""
Basic introduction to Baby Buddy (meant to be shown when no data is in the
database).
"""
2017-11-09 12:24:32 +00:00
template_name = 'babybuddy/welcome.html'