Misago/misago/threads/posting/threadclose.py

from misago.threads.forms.posting import ThreadCloseForm
from misago.threads.posting import PostingMiddleware, START


class ThreadCloseFormMiddleware(PostingMiddleware):
    def use_this_middleware(self):
        if self.forum.acl['can_close_threads']:
            self.thread_is_closed = self.thread.is_closed
            return True
        else:
            return False

    def make_form(self):
        if self.request.method == 'POST':
            return ThreadCloseForm(self.request.POST, prefix=self.prefix)
        else:
            initial = {'is_closed': self.thread_is_closed}
            return ThreadCloseForm(prefix=self.prefix, initial=initial)

    def pre_save(self, form):
        if form.is_valid():
            if self.thread_is_closed != form.cleaned_data.get('is_closed'):
                self.thread.is_closed = form.cleaned_data.get('is_closed')
                self.thread.update_fields.append('is_closed')