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

from rest_framework import serializers

from django.utils.translation import ugettext_lazy

from misago.markup import common_flavour
from misago.threads.checksums import update_post_checksum
from misago.threads.validators import validate_post, validate_post_length, validate_title

from . import PostingEndpoint, PostingMiddleware


class ReplyMiddleware(PostingMiddleware):
    def get_serializer(self):
        if self.mode == PostingEndpoint.START:
            return ThreadSerializer(data=self.request.data, context=self.kwargs)
        else:
            return ReplySerializer(data=self.request.data, context=self.kwargs)

    def save(self, serializer):
        if self.mode == PostingEndpoint.START:
            self.new_thread(serializer.validated_data)

        parsing_result = serializer.validated_data['parsing_result']

        if self.mode == PostingEndpoint.EDIT:
            self.edit_post(serializer.validated_data, parsing_result)
        else:
            self.new_post(serializer.validated_data, parsing_result)

        if self.mode == PostingEndpoint.START:
            self.post.set_search_document(self.thread.title)
        else:
            self.post.set_search_document()

        self.post.updated_on = self.datetime
        self.post.save()

        self.post.update_search_vector()
        update_post_checksum(self.post)

        self.post.update_fields += ['checksum', 'search_vector']

        if self.mode == PostingEndpoint.START:
            self.thread.set_first_post(self.post)
        if self.mode != PostingEndpoint.EDIT:
            self.thread.set_last_post(self.post)

        self.thread.save()

        # annotate post for future middlewares
        self.post.parsing_result = parsing_result

    def new_thread(self, validated_data):
        self.thread.set_title(validated_data['title'])
        self.thread.starter_name = self.user.username
        self.thread.starter_slug = self.user.slug
        self.thread.last_poster_name = self.user.username
        self.thread.last_poster_slug = self.user.slug
        self.thread.started_on = self.datetime
        self.thread.last_post_on = self.datetime
        self.thread.save()

    def edit_post(self, validated_data, parsing_result):
        self.post.original = parsing_result['original_text']
        self.post.parsed = parsing_result['parsed_text']

    def new_post(self, validated_data, parsing_result):
        self.post.thread = self.thread
        self.post.poster = self.user
        self.post.poster_name = self.user.username
        self.post.poster_ip = self.request.user_ip
        self.post.posted_on = self.datetime

        self.post.original = parsing_result['original_text']
        self.post.parsed = parsing_result['parsed_text']


class ReplySerializer(serializers.Serializer):
    post = serializers.CharField(
        validators=[validate_post_length],
        error_messages={
            'required': ugettext_lazy("You have to enter a message."),
        }
    )

    def validate(self, data):
        if data.get('post'):
            data['parsing_result'] = self.parse_post(data['post'])
            data = validate_post(self.context, data)

        return data

    def parse_post(self, post):
        if self.context['mode'] == PostingEndpoint.START:
            return common_flavour(self.context['request'], self.context['user'], post)
        else:
            return common_flavour(self.context['request'], self.context['post'].poster, post)


class ThreadSerializer(ReplySerializer):
    title = serializers.CharField(
        validators=[validate_title],
        error_messages={
            'required': ugettext_lazy("You have to enter thread title."),
        }
    )