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

from misago.acl import add_acl
from misago.categories import PRIVATE_THREADS_ROOT_NAME
from misago.categories.models import Category

from . import PostingEndpoint, PostingMiddleware


class PrivateThreadMiddleware(PostingMiddleware):
    """
    Middleware that sets private threads category for thread and post
    """

    def use_this_middleware(self):
        if self.mode == PostingEndpoint.START:
            return self.tree_name == PRIVATE_THREADS_ROOT_NAME
        return False

    def pre_save(self, serializer):
        category = Category.objects.private_threads()

        add_acl(self.user, category)

        # set flags for savechanges middleware
        category.update_all = False
        category.update_fields = []

        # assign category to thread and post
        self.thread.category = category
        self.post.category = category