from django import forms from django.shortcuts import redirect from django.template.defaultfilters import linebreaks_filter from django.utils.decorators import method_decorator from django.views.generic import FormView, ListView from activities.models import Post, PostInteraction, TimelineEvent from users.decorators import identity_required @method_decorator(identity_required, name="dispatch") class Home(FormView): template_name = "activities/home.html" class form_class(forms.Form): text = forms.CharField( widget=forms.Textarea( attrs={ "placeholder": "What's on your mind?", }, ) ) content_warning = forms.CharField( required=False, widget=forms.TextInput( attrs={ "placeholder": "Content Warning", "class": "hidden", }, ), ) def get_context_data(self): context = super().get_context_data() context["events"] = list( TimelineEvent.objects.filter( identity=self.request.identity, type__in=[TimelineEvent.Types.post, TimelineEvent.Types.boost], ) .select_related("subject_post", "subject_post__author") .prefetch_related("subject_post__attachments") .order_by("-created")[:50] ) context["interactions"] = PostInteraction.get_event_interactions( context["events"], self.request.identity ) context["current_page"] = "home" return context def form_valid(self, form): Post.create_local( author=self.request.identity, content=linebreaks_filter(form.cleaned_data["text"]), summary=form.cleaned_data.get("content_warning"), ) return redirect(".") class Local(ListView): template_name = "activities/local.html" extra_context = {"current_page": "local"} paginate_by = 50 def get_queryset(self): return ( Post.objects.filter(visibility=Post.Visibilities.public, author__local=True) .select_related("author") .prefetch_related("attachments") .order_by("-created")[:50] ) @method_decorator(identity_required, name="dispatch") class Federated(ListView): template_name = "activities/federated.html" extra_context = {"current_page": "federated"} paginate_by = 50 def get_queryset(self): return ( Post.objects.filter(visibility=Post.Visibilities.public) .select_related("author") .prefetch_related("attachments") .order_by("-created")[:50] ) @method_decorator(identity_required, name="dispatch") class Notifications(ListView): template_name = "activities/notifications.html" extra_context = {"current_page": "notifications"} paginate_by = 50 def get_queryset(self): return ( TimelineEvent.objects.filter( identity=self.request.identity, type__in=[ TimelineEvent.Types.mentioned, TimelineEvent.Types.boosted, TimelineEvent.Types.liked, TimelineEvent.Types.followed, ], ) .order_by("-created")[:50] .select_related("subject_post", "subject_post__author", "subject_identity") )