Misago/misago/threads/views/posting.py

from django.core.urlresolvers import reverse
from django.shortcuts import redirect
from django.template import RequestContext
from django.utils import timezone
from django.utils.translation import ugettext as _
from misago.acl.utils import ACLError403, ACLError404
from misago.forms import FormLayout
from misago.forums.models import Forum
from misago.markdown import post_markdown
from misago.messages import Message
from misago.threads.forms import PostForm
from misago.threads.models import Thread, Post
from misago.threads.views.base import BaseView
from misago.views import error403, error404
from misago.utils import slugify

class PostingView(BaseView):
    def fetch_target(self, kwargs):
        if self.mode == 'new_thread':
            self.fetch_forum(kwargs)
        if self.mode in ('edit_thread', 'new_post', 'new_post_quick'):
            self.fetch_thread(kwargs)
    
    def fetch_forum(self, kwargs):
        self.forum = Forum.objects.get(pk=kwargs['forum'], type='forum')
        self.request.acl.forums.allow_forum_view(self.forum)
        self.request.acl.threads.allow_new_threads(self.forum)
        self.parents = self.forum.get_ancestors(include_self=True).filter(level__gt=1)
    
    def fetch_thread(self, kwargs):
        self.thread = Thread.objects.get(pk=kwargs['thread'])
        self.forum = self.thread.forum
        self.request.acl.forums.allow_forum_view(self.forum)
        self.request.acl.threads.allow_reply(self.thread)
        self.parents = self.forum.get_ancestors(include_self=True).filter(level__gt=1)
        
    def get_form(self, bound=False):            
        if bound:            
            return PostForm(self.request.POST,request=self.request,mode=self.mode)
        return PostForm(request=self.request,mode=self.mode)
            
    def __call__(self, request, **kwargs):
        self.request = request
        self.forum = None
        self.thread = None
        self.post = None
        self.parents = None
        self.mode = kwargs.get('mode')
        if self.request.POST.get('quick_reply') and self.mode == 'new_post':
            self.mode = 'new_post_quick'
        try:
            self.fetch_target(kwargs)
            if not request.user.is_authenticated():
                raise ACLError403(_("Guest, you have to sign-in to post."))
        except (Forum.DoesNotExist, Thread.DoesNotExist, Post.DoesNotExist):
            return error404(self.request)
        except ACLError403 as e:
            return error403(request, e.message)
        except ACLError404 as e:
            return error404(request, e.message)
        
        message = request.messages.get_message('threads')
        if request.method == 'POST':
            form = self.get_form(True)
            if form.is_valid():
                # Get or create new thread
                if self.mode in ['new_thread', 'edit_thread']:
                    thread = Thread.objects.create(
                                                   forum=self.forum,
                                                   name=form.cleaned_data['thread_name'],
                                                   slug=slugify(form.cleaned_data['thread_name']),
                                                   start=timezone.now(),
                                                   last=timezone.now(),
                                                   )
                else:
                    thread = self.thread
                
                # Create new message
                post = Post.objects.create(
                                           forum=self.forum,
                                           thread=thread,
                                           user=request.user,
                                           user_name=request.user.username,
                                           ip=request.session.get_ip(request),
                                           agent=request.META.get('HTTP_USER_AGENT'),
                                           post=form.cleaned_data['post'],
                                           post_preparsed=post_markdown(request, form.cleaned_data['post']),
                                           date=timezone.now()
                                           )
                
                if self.mode == 'new_thread':
                    thread.start_post = post
                    thread.start_poster = request.user
                    thread.start_poster_name = request.user.username
                    thread.start_poster_slug = request.user.username_slug
                    if request.user.rank and request.user.rank.style:
                        thread.start_poster_style = request.user.rank.style
                    
                thread.last = timezone.now()
                thread.last_post = post
                thread.last_poster = request.user
                thread.last_poster_name = request.user.username
                thread.last_poster_slug = request.user.username_slug
                if request.user.rank and request.user.rank.style:
                    thread.last_poster_style = request.user.rank.style
                if self.mode in ['new_post', 'new_post_quick']:
                    thread.replies += 1
                thread.save(force_update=True)
                
                # Update forum
                if self.mode == 'new_thread':
                    self.forum.threads += 1
                    self.forum.threads_delta += 1
                    
                if self.mode in ['new_post', 'new_post_quick']:
                    self.forum.posts += 1
                    self.forum.posts_delta += 1
                    
                self.forum.last_thread = thread
                self.forum.last_thread_name = thread.name
                self.forum.last_thread_slug = thread.slug
                self.forum.last_thread_date = thread.last
                self.forum.last_poster = thread.last_poster
                self.forum.last_poster_name = thread.last_poster_name
                self.forum.last_poster_slug = thread.last_poster_slug
                self.forum.last_poster_style = thread.last_poster_style
                self.forum.save(force_update=True)
                
                # Update user
                if self.mode == 'new_thread':
                    request.user.threads += 1
                request.user.posts += 1
                request.user.last_post = thread.last
                request.user.save(force_update=True)
                
                if self.mode == 'new_thread':
                    request.messages.set_flash(Message(_("New thread has been posted.")), 'success', 'threads')
                    return redirect(reverse('thread', kwargs={'thread': thread.pk, 'slug': thread.slug}) + ('#post-%s' % post.pk))
                
                if self.mode in ['new_post', 'new_post_quick']:
                    request.messages.set_flash(Message(_("Your reply has been posted.")), 'success', 'threads')
                    return redirect(reverse('thread', kwargs={'thread': self.thread.pk, 'slug': self.thread.slug}) + ('#post-%s' % post.pk))
            message = Message(form.non_field_errors()[0], 'error')
        else:
            form = self.get_form()
        
        return request.theme.render_to_response('threads/posting.html',
                                                {
                                                 'mode': self.mode,
                                                 'forum': self.forum,
                                                 'thread': self.thread,
                                                 'post': self.post,
                                                 'parents': self.parents,
                                                 'message': message,
                                                 'form': FormLayout(form),
                                                 },
                                                context_instance=RequestContext(request));