threads.py 7.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235
  1. from django.contrib import messages
  2. from django.core.urlresolvers import reverse
  3. from django.shortcuts import redirect
  4. from django.utils.translation import ugettext_lazy, ugettext as _
  5. from misago.core.shortcuts import paginate
  6. from misago.readtracker import threadstracker
  7. from misago.threads.models import Label
  8. from misago.threads.moderation import ModerationError
  9. from misago.threads.views.generic.base import ViewBase
  10. __all__ = ['Actions', 'Sorting', 'Threads', 'ThreadsView']
  11. class Actions(object):
  12. select_threads_message = ugettext_lazy(
  13. "You have to select at least one thread.")
  14. def __init__(self, **kwargs):
  15. if kwargs.get('user').is_authenticated():
  16. self.available_actions = self.get_available_actions(kwargs)
  17. else:
  18. self.available_actions = []
  19. self.selected_ids = []
  20. def get_available_actions(self, kwargs):
  21. raise NotImplementedError("get_available_actions has to return list "
  22. "of dicts with allowed actions")
  23. def resolve_action(self, request):
  24. action_name = request.POST.get('action')
  25. for action in self.available_actions:
  26. if action['action'] == action_name:
  27. if ':' in action_name:
  28. action_bits = action_name.split(':')
  29. action_name = action_bits[0]
  30. action_arg = action_bits[1]
  31. else:
  32. action_arg = None
  33. action_callable = 'action_%s' % action_name
  34. return getattr(self, action_callable), action_arg
  35. else:
  36. raise ModerationError(_("Requested action is invalid."))
  37. def clean_selection(self, data):
  38. filtered_data = []
  39. for pk in data[:50]: # a tiny fail-safe to avoid too big workloads
  40. try:
  41. filtered_data.append(int(pk))
  42. except ValueError:
  43. pass
  44. if not filtered_data:
  45. raise ModerationError(self.select_threads_message)
  46. return filtered_data
  47. def handle_post(self, request, queryset):
  48. try:
  49. action, action_arg = self.resolve_action(request)
  50. self.selected_ids = self.clean_selection(
  51. request.POST.getlist('thread', []))
  52. filtered_queryset = queryset.filter(pk__in=self.selected_ids)
  53. if filtered_queryset.exists():
  54. if action_arg:
  55. response = action(request, filtered_queryset, action_arg)
  56. else:
  57. response = action(request, filtered_queryset)
  58. if response:
  59. return response
  60. else:
  61. # prepare default response: page reload
  62. return redirect(request.path)
  63. else:
  64. raise ModerationError(self.select_threads_message)
  65. except ModerationError as e:
  66. messages.error(request, e.message)
  67. return False
  68. def get_list(self):
  69. return self.available_actions
  70. def get_selected_ids(self):
  71. return self.selected_ids
  72. def action_approve(self, request, threads):
  73. changed_threads = 0
  74. for thread in threads:
  75. if moderation.approve_thread(request.user, thread):
  76. changed_threads += 1
  77. if changed_threads:
  78. message = ungettext(
  79. '%(changed)d thread was approved.',
  80. '%(changed)d threads were approved.',
  81. changed_threads)
  82. messages.success(request, message % {'changed': changed_threads})
  83. else:
  84. message = ("No threads were approved.")
  85. messages.info(request, message)
  86. class Sorting(object):
  87. sortings = (
  88. {
  89. 'method': 'recently-replied',
  90. 'name': ugettext_lazy("Recently replied"),
  91. 'order_by': '-last_post_on',
  92. },
  93. {
  94. 'method': 'last-replied',
  95. 'name': ugettext_lazy("Last replied"),
  96. 'order_by': 'last_post_on',
  97. },
  98. {
  99. 'method': 'most-replied',
  100. 'name': ugettext_lazy("Most replied"),
  101. 'order_by': '-replies',
  102. },
  103. {
  104. 'method': 'least-replied',
  105. 'name': ugettext_lazy("Least replied"),
  106. 'order_by': 'replies',
  107. },
  108. {
  109. 'method': 'newest',
  110. 'name': ugettext_lazy("Newest"),
  111. 'order_by': '-id',
  112. },
  113. {
  114. 'method': 'oldest',
  115. 'name': ugettext_lazy("Oldest"),
  116. 'order_by': 'id',
  117. },
  118. )
  119. def __init__(self, link_name, link_params):
  120. self.link_name = link_name
  121. self.link_params = link_params.copy()
  122. self.default_method = self.sortings[0]['method']
  123. def clean_kwargs(self, kwargs):
  124. sorting = kwargs.get('sort')
  125. if sorting:
  126. if sorting == self.default_method:
  127. kwargs.pop('sort')
  128. return kwargs
  129. available_sortings = [method['method'] for method in self.sortings]
  130. if sorting not in available_sortings:
  131. kwargs.pop('sort')
  132. return kwargs
  133. else:
  134. sorting = self.default_method
  135. self.set_sorting(sorting)
  136. return kwargs
  137. def set_sorting(self, method):
  138. for sorting in self.sortings:
  139. if sorting['method'] == method:
  140. self.sorting = sorting
  141. break
  142. @property
  143. def name(self):
  144. return self.sorting['name']
  145. def choices(self):
  146. choices = []
  147. for sorting in self.sortings:
  148. if sorting['method'] != self.sorting['method']:
  149. if sorting['method'] == self.default_method:
  150. self.link_params.pop('sort', None)
  151. else:
  152. self.link_params['sort'] = sorting['method']
  153. url = reverse(self.link_name, kwargs=self.link_params)
  154. choices.append({
  155. 'name': sorting['name'],
  156. 'url': url,
  157. })
  158. return choices
  159. def sort(self, threads):
  160. threads.sort(self.sorting['order_by'])
  161. class Threads(object):
  162. def __init__(self, user):
  163. self.user = user
  164. def sort(self, sort_by):
  165. self.queryset = self.queryset.order_by(sort_by)
  166. def label_threads(self, threads, labels=None):
  167. if labels:
  168. labels_dict = dict([(label.pk, label) for label in labels])
  169. else:
  170. labels_dict = Label.objects.get_cached_labels_dict()
  171. for thread in threads:
  172. thread.label = labels_dict.get(thread.label_id)
  173. def make_threads_read_aware(self, threads):
  174. threadstracker.make_read_aware(self.user, threads)
  175. class ThreadsView(ViewBase):
  176. def get_threads(self, request, kwargs):
  177. queryset = self.get_threads_queryset(request, forum)
  178. queryset = threads_qs.order_by('-last_post_id')
  179. page = paginate(threads_qs, kwargs.get('page', 0), 30, 10)
  180. threads = [thread for thread in page.object_list]
  181. return page, threads
  182. def get_threads_queryset(self, request):
  183. return forum.thread_set.all().order_by('-last_post_id')
  184. def clean_kwargs(self, request, kwargs):
  185. cleaned_kwargs = kwargs.copy()
  186. if request.user.is_anonymous():
  187. """we don't allow sort/filter for guests"""
  188. cleaned_kwargs.pop('sort', None)
  189. cleaned_kwargs.pop('show', None)
  190. return cleaned_kwargs