boards/views/thread/thread.py

from django.core.exceptions import ObjectDoesNotExist
from django.http import Http404
from django.shortcuts import get_object_or_404, render, redirect
from django.views.generic.edit import FormMixin
from django.utils import timezone
from django.utils.dateformat import format

from boards import utils, settings
from boards.forms import PostForm, PlainErrorList
from boards.models import Post
from boards.views.base import BaseBoardView, CONTEXT_FORM
from boards.views.posting_mixin import PostMixin

import neboard


CONTEXT_LASTUPDATE = "last_update"
CONTEXT_THREAD = 'thread'
CONTEXT_WS_TOKEN = 'ws_token'
CONTEXT_WS_PROJECT = 'ws_project'
CONTEXT_WS_HOST = 'ws_host'
CONTEXT_WS_PORT = 'ws_port'
CONTEXT_WS_TIME = 'ws_token_time'
CONTEXT_MODE = 'mode'
CONTEXT_OP = 'opening_post'

FORM_TITLE = 'title'
FORM_TEXT = 'text'
FORM_IMAGE = 'image'
FORM_THREADS = 'threads'


class ThreadView(BaseBoardView, PostMixin, FormMixin):

    def get(self, request, post_id, form: PostForm=None):
        try:
            opening_post = Post.objects.get(id=post_id)
        except ObjectDoesNotExist:
            raise Http404

        # If this is not OP, don't show it as it is
        if not opening_post.is_opening():
            return redirect(opening_post.get_thread().get_opening_post()
                .get_absolute_url())

        if not form:
            form = PostForm(error_class=PlainErrorList)

        thread_to_show = opening_post.get_thread()

        params = dict()

        params[CONTEXT_FORM] = form
        params[CONTEXT_LASTUPDATE] = str(thread_to_show.last_edit_time)
        params[CONTEXT_THREAD] = thread_to_show
        params[CONTEXT_MODE] = self.get_mode()
        params[CONTEXT_OP] = opening_post

        if settings.get_bool('External', 'WebsocketsEnabled'):
            token_time = format(timezone.now(), u'U')

            params[CONTEXT_WS_TIME] = token_time
            params[CONTEXT_WS_TOKEN] = utils.get_websocket_token(
                timestamp=token_time)
            params[CONTEXT_WS_PROJECT] = neboard.settings.CENTRIFUGE_PROJECT_ID
            params[CONTEXT_WS_HOST] = request.get_host().split(':')[0]
            params[CONTEXT_WS_PORT] = neboard.settings.CENTRIFUGE_PORT

        params.update(self.get_data(thread_to_show))

        return render(request, self.get_template(), params)

    def post(self, request, post_id):
        opening_post = get_object_or_404(Post, id=post_id)

        # If this is not OP, don't show it as it is
        if not opening_post.is_opening():
            raise Http404

        if not opening_post.get_thread().archived:
            form = PostForm(request.POST, request.FILES,
                            error_class=PlainErrorList)
            form.session = request.session

            if form.is_valid():
                return self.new_post(request, form, opening_post)
            if form.need_to_ban:
                # Ban user because he is suspected to be a bot
                self._ban_current_user(request)

            return self.get(request, post_id, form)

    def new_post(self, request, form: PostForm, opening_post: Post=None,
                 html_response=True):
        """
        Adds a new post (in thread or as a reply).
        """

        ip = utils.get_client_ip(request)

        data = form.cleaned_data

        title = data[FORM_TITLE]
        text = data[FORM_TEXT]
        file = form.get_file()
        threads = data[FORM_THREADS]

        text = self._remove_invalid_links(text)

        post_thread = opening_post.get_thread()

        post = Post.objects.create_post(title=title, text=text, file=file,
                                        thread=post_thread, ip=ip,
                                        opening_posts=threads)
        post.notify_clients()

        if html_response:
            if opening_post:
                return redirect(post.get_absolute_url())
        else:
            return post

    def get_data(self, thread) -> dict:
        """
        Returns context params for the view.
        """

        return dict()

    def get_template(self) -> str:
        """
        Gets template to show the thread mode on.
        """

        pass

    def get_mode(self) -> str:
        pass