threadweight.py 872 B

1234567891011121314151617181920212223
  1. from misago.threads.forms.posting import ThreadWeightForm
  2. from misago.threads.posting import PostingMiddleware
  3. class ThreadWeightFormMiddleware(PostingMiddleware):
  4. def use_this_middleware(self):
  5. if self.forum.acl['can_change_threads_weight']:
  6. self.thread_weight = self.thread.weight
  7. return True
  8. else:
  9. return False
  10. def make_form(self):
  11. if self.request.method == 'POST':
  12. return ThreadWeightForm(self.request.POST, prefix=self.prefix)
  13. else:
  14. initial = {'weight': self.thread_weight}
  15. return ThreadWeightForm(prefix=self.prefix, initial=initial)
  16. def pre_save(self, form):
  17. if self.thread_weight != form.cleaned_data.get('weight'):
  18. self.thread.weight = form.cleaned_data.get('weight')
  19. self.thread.update_fields.append('weight')