from django.contrib import messages from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin from django.core.exceptions import PermissionDenied from django.shortcuts import get_object_or_404, redirect from django.utils.functional import cached_property from django.utils.translation import gettext as _ from moku.constants import EMOJI_CATEGORIES, Verbs from moku.forms.post import PostForm from moku.images import process_post_image from moku.models.post import Post from moku.models.recipe import Recipe from moku.views.base import FormView def _get_verbs(username): return ( (verb[0], verb[1] % {"user": f"@{username}", "food": "..."}) for verb in Verbs.CHOICES ) class EditPostview(LoginRequiredMixin, UserPassesTestMixin, FormView): """Allows users to edit their previous posts.""" template_name = "moku/post/edit.jinja" form_class = PostForm def form_valid(self, form): if ( form.instance.recipe and form.instance.recipe.created_by.id != self.request.user.id ): messages.error( self.request, _("you can't add someone else's recipe to your post!") ) return self.form_invalid(form) if "image" in form.changed_data and form.instance.image.name: form.instance.image = process_post_image(form.instance.image) form.save() messages.success(self.request, _("your post was updated!")) return redirect("feed") def get_context_data(self, **kwargs): return { **super().get_context_data(**kwargs), "post": self.post_object, "verbs": _get_verbs(self.request.user.username), "emoji": EMOJI_CATEGORIES, } def get_form(self): return self.form_class(instance=self.post_object, **self.get_form_kwargs()) @cached_property def post_object(self): return get_object_or_404(Post, uuid=self.kwargs.get("uuid")) def test_func(self): return self.post_object.created_by.id == self.request.user.id class FeedView(FormView): """Allows users to see recent posts and create a new post.""" template_name = "moku/post/index.jinja" form_class = PostForm def form_valid(self, form): if not self.request.user.is_authenticated: raise PermissionDenied if ( form.instance.recipe and form.instance.recipe.created_by.id != self.request.user.id ): messages.error( self.request, _("you can't add someone else's recipe to your post!") ) return redirect("feed") form.instance.created_by = self.request.user if "image" in form.changed_data and form.instance.image.name: form.instance.image = process_post_image(form.instance.image) form.save() messages.success(self.request, _("your post was made!")) return redirect("feed") def get_context_data(self, **kwargs): context = { **super().get_context_data(**kwargs), "posts": Post.objects.all()[:128], } if self.request.user.is_authenticated: return self.get_authenticated_context_data(context) return context def get_authenticated_context_data(self, context): return { **context, "emoji": EMOJI_CATEGORIES, "verbs": _get_verbs(self.request.user.username), } def get_form(self, form_class=None): form = super().get_form(form_class) if self.request.user.is_authenticated: form.fields["recipe"].queryset = Recipe.objects.filter( created_by=self.request.user ) return form