Misago/misago/threads/api/postingendpoint/subscribe.py

from django.contrib.auth import get_user_model

from misago.threads.models import Subscription

from . import PostingEndpoint, PostingMiddleware


UserModel = get_user_model()


class SubscribeMiddleware(PostingMiddleware):
    def use_this_middleware(self):
        return self.mode != PostingEndpoint.EDIT

    def post_save(self, serializer):
        self.subscribe_new_thread()
        self.subscribe_replied_thread()

    def subscribe_new_thread(self):
        if self.mode != PostingEndpoint.START:
            return

        if self.user.subscribe_to_started_threads == UserModel.SUBSCRIPTION_NONE:
            return

        self.user.subscription_set.create(
            category=self.thread.category,
            thread=self.thread,
            send_email=self.user.subscribe_to_started_threads
            == UserModel.SUBSCRIPTION_ALL,
        )

    def subscribe_replied_thread(self):
        if self.mode != PostingEndpoint.REPLY:
            return

        if self.user.subscribe_to_replied_threads == UserModel.SUBSCRIPTION_NONE:
            return

        try:
            return self.user.subscription_set.get(thread=self.thread)
        except Subscription.DoesNotExist:
            pass

        # posts user's posts in this thread, minus events and current post
        posts_queryset = self.user.post_set.filter(
            thread=self.thread, is_event=False
        ).exclude(pk=self.post.pk)

        if posts_queryset.exists():
            return

        self.user.subscription_set.create(
            category=self.thread.category,
            thread=self.thread,
            send_email=self.user.subscribe_to_replied_threads
            == UserModel.SUBSCRIPTION_ALL,
        )