Misago/misago/threads/forms/moderation.py

from urlparse import urlparse

from django.core.urlresolvers import resolve
from django.http import Http404
from django.utils.translation import ugettext_lazy as _

from misago.acl import add_acl
from misago.core import forms
from misago.forums.forms import ForumChoiceField
from misago.forums.permissions import allow_see_forum, allow_browse_forum

from misago.threads.models import Thread
from misago.threads.permissions import allow_see_thread
from misago.threads.validators import validate_title


class MergeThreadsForm(forms.Form):
    merged_thread_title = forms.CharField(label=_("Merged thread title"),
                                          required=False)

    def clean(self):
        data = super(MergeThreadsForm, self).clean()

        merged_thread_title = data.get('merged_thread_title')
        if merged_thread_title:
            validate_title(merged_thread_title)
        else:
            message = _("You have to enter merged thread title.")
            raise forms.ValidationError(message)
        return data


class MoveThreadsForm(forms.Form):
    new_forum = ForumChoiceField(label=_("Move threads to forum"),
                                 empty_label=None)

    def __init__(self, *args, **kwargs):
        self.forum = kwargs.pop('forum')
        acl = kwargs.pop('acl')

        super(MoveThreadsForm, self).__init__(*args, **kwargs)

        self.fields['new_forum'].set_acl(acl)

    def clean(self):
        data = super(MoveThreadsForm, self).clean()

        new_forum = data.get('new_forum')
        if new_forum:
            if new_forum.is_category:
                message = _("You can't move threads to category.")
                raise forms.ValidationError(message)
            if new_forum.is_redirect:
                message = _("You can't move threads to redirect.")
                raise forms.ValidationError(message)
            if new_forum.pk == self.forum.pk:
                message = _("New forum is same as current one.")
                raise forms.ValidationError(message)
        else:
            raise forms.ValidationError(_("You have to select forum."))
        return data


class MoveThreadForm(MoveThreadsForm):
    new_forum = ForumChoiceField(label=_("Move thread to forum"),
                                 empty_label=None)


class MovePostsForm(forms.Form):
    new_thread_url = forms.CharField(
        label=_("New thread link"),
        help_text=_("Paste link to thread you want selected posts moved to."))

    def __init__(self, *args, **kwargs):
        self.user = kwargs.pop('user')
        self.thread = kwargs.pop('thread')
        self.new_thread = None

        super(MovePostsForm, self).__init__(*args, **kwargs)

    def clean(self):
        data = super(MovePostsForm, self).clean()

        new_thread_url = data.get('new_thread_url')
        try:
            if not new_thread_url:
                raise Http404()

            resolution = resolve(urlparse(new_thread_url).path)
            if not 'thread_id' in resolution.kwargs:
                raise Http404()

            queryset = Thread.objects.select_related('forum')
            self.new_thread = queryset.get(id=resolution.kwargs['thread_id'])

            add_acl(self.user, self.new_thread.forum)
            add_acl(self.user, self.new_thread)

            allow_see_forum(self.user, self.new_thread.forum)
            allow_browse_forum(self.user, self.new_thread.forum)
            allow_see_thread(self.user, self.new_thread)

        except (Http404, Thread.DoesNotExist):
            message = _("You have to enter valid link to thread.")
            raise forms.ValidationError(message)

        if self.thread == self.new_thread:
            message = _("New thread is same as current one.")
            raise forms.ValidationError(message)

        if self.new_thread.forum.special_role:
            message = _("You can't move posts to special threads.")
            raise forms.ValidationError(message)

        return data


class SplitThreadForm(forms.Form):
    forum = ForumChoiceField(label=_("New thread forum"),
                                 empty_label=None)

    thread_title = forms.CharField(label=_("New thread title"),
                                   required=False)

    def __init__(self, *args, **kwargs):
        acl = kwargs.pop('acl')

        super(SplitThreadForm, self).__init__(*args, **kwargs)

        self.fields['forum'].set_acl(acl)

    def clean(self):
        data = super(SplitThreadForm, self).clean()

        forum = data.get('forum')
        if forum:
            if forum.is_category:
                message = _("You can't start threads in category.")
                raise forms.ValidationError(message)
            if forum.is_redirect:
                message = _("You can't start threads in redirect.")
                raise forms.ValidationError(message)
        else:
            raise forms.ValidationError(_("You have to select forum."))

        thread_title = data.get('thread_title')
        if thread_title:
            validate_title(thread_title)
        else:
            message = _("You have to enter new thread title.")
            raise forms.ValidationError(message)

        return data