boards/views/thread.py

from django.core.urlresolvers import reverse
from django.db import transaction
from django.http import Http404
from django.shortcuts import get_object_or_404, render, redirect
from django.views.generic.edit import FormMixin

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

TEMPLATE_GALLERY = 'boards/thread_gallery.html'
TEMPLATE_NORMAL = 'boards/thread.html'

CONTEXT_POSTS = 'posts'
CONTEXT_OP = 'opening_post'
CONTEXT_BUMPLIMIT_PRG = 'bumplimit_progress'
CONTEXT_POSTS_LEFT = 'posts_left'
CONTEXT_LASTUPDATE = "last_update"
CONTEXT_MAX_REPLIES = 'max_replies'
CONTEXT_THREAD = 'thread'
CONTEXT_BUMPABLE = 'bumpable'
CONTEXT_WS_TOKEN = 'ws_token'
CONTEXT_WS_PROJECT = 'ws_project'
CONTEXT_WS_HOST = 'ws_host'
CONTEXT_WS_PORT = 'ws_port'

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

MODE_GALLERY = 'gallery'
MODE_NORMAL = 'normal'


class ThreadView(BaseBoardView, PostMixin, FormMixin):

    def get(self, request, post_id, mode=MODE_NORMAL, form=None):
        try:
            opening_post = Post.objects.filter(id=post_id).only('thread_new')[0]
        except IndexError:
            raise Http404

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

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

        thread_to_show = opening_post.get_thread()

        context = self.get_context_data(request=request)

        context[CONTEXT_FORM] = form
        context[CONTEXT_LASTUPDATE] = str(utils.datetime_to_epoch(
            thread_to_show.last_edit_time))
        context[CONTEXT_THREAD] = thread_to_show
        context[CONTEXT_MAX_REPLIES] = settings.MAX_POSTS_PER_THREAD

        if settings.WEBSOCKETS_ENABLED:
            context[CONTEXT_WS_TOKEN] = utils.get_websocket_token(
                timestamp=context[CONTEXT_LASTUPDATE])
            context[CONTEXT_WS_PROJECT] = neboard.settings.CENTRIFUGE_PROJECT_ID
            context[CONTEXT_WS_HOST] = request.get_host().split(':')[0]
            context[CONTEXT_WS_PORT] = neboard.settings.CENTRIFUGE_PORT

        # TODO Move this to subclasses: NormalThreadView, GalleryThreadView etc
        if MODE_NORMAL == mode:
            bumpable = thread_to_show.can_bump()
            context[CONTEXT_BUMPABLE] = bumpable
            if bumpable:
                left_posts = settings.MAX_POSTS_PER_THREAD \
                             - thread_to_show.get_reply_count()
                context[CONTEXT_POSTS_LEFT] = left_posts
                context[CONTEXT_BUMPLIMIT_PRG] = str(
                    float(left_posts) / settings.MAX_POSTS_PER_THREAD * 100)

            context[CONTEXT_OP] = opening_post

            document = TEMPLATE_NORMAL
        elif MODE_GALLERY == mode:
            context[CONTEXT_POSTS] = thread_to_show.get_replies_with_images(
                view_fields_only=True)

            document = TEMPLATE_GALLERY
        else:
            raise Http404

        # TODO Use dict here
        return render(request, document, context_instance=context)

    def post(self, request, post_id, mode=MODE_NORMAL):
        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, mode, form)

    @transaction.atomic
    def new_post(self, request, form, opening_post=None, html_response=True):
        """Add a new post (in thread or as a reply)."""

        ip = utils.get_client_ip(request)
        is_banned = Ban.objects.filter(ip=ip).exists()

        if is_banned:
            if html_response:
                return redirect(BannedView().as_view())
            else:
                return None

        data = form.cleaned_data

        title = data[FORM_TITLE]
        text = data[FORM_TEXT]

        text = self._remove_invalid_links(text)

        if FORM_IMAGE in list(data.keys()):
            image = data[FORM_IMAGE]
        else:
            image = None

        tags = []

        post_thread = opening_post.get_thread()

        post = Post.objects.create_post(title=title, text=text, image=image,
                                        thread=post_thread, ip=ip, tags=tags)
        post.send_to_websocket(request, recursive=False)

        thread_to_show = (opening_post.id if opening_post else post.id)

        if html_response:
            if opening_post:
                return redirect(
                    reverse('thread', kwargs={'post_id': thread_to_show})
                    + '#' + str(post.id))
        else:
            return post