77 lines
2.7 KiB
Python
77 lines
2.7 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 django.views.generic import FormView, TemplateView
|
|
|
|
from moku.images import process_avatar_image
|
|
from moku.forms.user import ProfileForm, UserForm, UserSettingsForm
|
|
from moku.models.user import User
|
|
|
|
|
|
class EditProfileView(LoginRequiredMixin, FormView):
|
|
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):
|
|
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(TemplateView):
|
|
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):
|
|
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)
|