93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
from django.contrib import messages
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.db import IntegrityError
|
|
from django.shortcuts import redirect
|
|
from django.utils.translation import gettext as _
|
|
|
|
from moku.forms.user import ProfileForm, UserForm, UserSettingsForm
|
|
from moku.images import process_avatar_image
|
|
from moku.models.user import User
|
|
from moku.views.base import FormView, View
|
|
|
|
|
|
class EditProfileView(LoginRequiredMixin, FormView):
|
|
"""Allows a user to edit information within their user profile."""
|
|
|
|
template_name = "moku/profile/edit.jinja"
|
|
form_class = ProfileForm
|
|
|
|
def form_valid(self, form):
|
|
if "avatar" in form.changed_data and form.instance.avatar is not None:
|
|
form.instance.avatar = process_avatar_image(form.instance.avatar)
|
|
form.save()
|
|
messages.success(self.request, _("profile updated successfully!"))
|
|
return redirect("profile", username=form.instance.username)
|
|
|
|
def get_form(self):
|
|
return self.form_class(instance=self.request.user, **self.get_form_kwargs())
|
|
|
|
|
|
class EditSettingsView(LoginRequiredMixin, FormView):
|
|
"""Allows a user to edit information within their user settings."""
|
|
|
|
template_name = "moku/settings.jinja"
|
|
form_class = UserSettingsForm
|
|
|
|
def form_valid(self, form):
|
|
form.instance.user = self.request.user
|
|
try:
|
|
form.save()
|
|
except IntegrityError:
|
|
messages.error(
|
|
self.request, _("uh oh. i think something went a little bit oopsie.")
|
|
)
|
|
return self.form_invalid(form)
|
|
messages.success(self.request, _("settings updated!"))
|
|
return redirect("settings")
|
|
|
|
def get_form(self):
|
|
if hasattr(self.request.user, "settings"):
|
|
return self.form_class(
|
|
instance=self.request.user.settings, **self.get_form_kwargs()
|
|
)
|
|
return super().get_form()
|
|
|
|
|
|
class ProfileView(View):
|
|
"""Shows a user's profile along with a list of their recent posts."""
|
|
|
|
template_name = "moku/profile/show.jinja"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
user = User.objects.get(username=self.kwargs.get("username"))
|
|
return {
|
|
**super().get_context_data(**kwargs),
|
|
"profile": user,
|
|
"posts": user.posts.order_by("-created_at").all(),
|
|
}
|
|
|
|
|
|
class SignupView(FormView):
|
|
"""Allows non-authenticated users to create an account on the site."""
|
|
|
|
template_name = "moku/signup.jinja"
|
|
form_class = UserForm
|
|
|
|
def form_valid(self, form):
|
|
form.instance.username = form.instance.username.lower()
|
|
try:
|
|
form.save()
|
|
except IntegrityError:
|
|
messages.error(
|
|
self.request, _("sorry! someone else got to that username first.")
|
|
)
|
|
return self.form_invalid(form)
|
|
messages.success(
|
|
self.request, _("that's it! just log in, and you're ready to go.")
|
|
)
|
|
return redirect("login")
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
if self.request.user.is_authenticated:
|
|
return redirect("feed")
|
|
return super().get(request, *args, **kwargs)
|