views.py 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. from django.core.urlresolvers import reverse as django_reverse
  2. from django.db.models import Q
  3. from django.template import RequestContext
  4. from django.utils.translation import ugettext as _
  5. from misago.admin import site
  6. from misago.apps.admin.widgets import *
  7. from misago.messages import Message
  8. from misago.models import Ban
  9. from misago.apps.admin.bans.forms import BanForm, SearchBansForm
  10. def reverse(route, target=None):
  11. if target:
  12. return django_reverse(route, kwargs={'target': target.pk})
  13. return django_reverse(route)
  14. """
  15. Views
  16. """
  17. class List(ListWidget):
  18. """
  19. List Bans
  20. """
  21. admin = site.get_action('bans')
  22. id = 'list'
  23. columns = (
  24. ('ban', _("Ban"), 50),
  25. ('expires', _("Expires")),
  26. )
  27. default_sorting = 'expires'
  28. sortables = {
  29. 'ban': 1,
  30. 'expires': 0,
  31. }
  32. pagination = 20
  33. search_form = SearchBansForm
  34. empty_message = _('No bans are currently set.')
  35. empty_search_message = _('No bans have been found.')
  36. nothing_checked_message = _('You have to check at least one ban.')
  37. actions = (
  38. ('delete', _("Lift selected bans"), _("Are you sure you want to lift selected bans?")),
  39. )
  40. def set_filters(self, model, filters):
  41. if 'ban' in filters:
  42. model = model.filter(ban__contains=filters['ban'])
  43. if 'reason' in filters:
  44. model = model.filter(Q(reason_user__contains=filters['reason']) | Q(reason_admin__contains=filters['reason']))
  45. if 'type' in filters:
  46. model = model.filter(type__in=filters['type'])
  47. return model
  48. def get_item_actions(self, item):
  49. return (
  50. self.action('pencil', _("Edit Ban"), reverse('admin_bans_edit', item)),
  51. self.action('remove', _("Lift Ban"), reverse('admin_bans_delete', item), post=True, prompt=_("Are you sure you want to lift this ban?")),
  52. )
  53. def action_delete(self, items, checked):
  54. Ban.objects.filter(id__in=checked).delete()
  55. self.request.monitor['bans_version'] = int(self.request.monitor['bans_version']) + 1
  56. return Message(_('Selected bans have been lifted successfully.'), 'success'), reverse('admin_bans')
  57. class New(FormWidget):
  58. """
  59. Create Ban
  60. """
  61. admin = site.get_action('bans')
  62. id = 'new'
  63. fallback = 'admin_bans'
  64. form = BanForm
  65. submit_button = _("Set Ban")
  66. def get_new_url(self, model):
  67. return reverse('admin_bans_new')
  68. def get_edit_url(self, model):
  69. return reverse('admin_bans_edit', model)
  70. def submit_form(self, form, target):
  71. new_ban = Ban(
  72. type=form.cleaned_data['type'],
  73. ban=form.cleaned_data['ban'],
  74. reason_user=form.cleaned_data['reason_user'],
  75. reason_admin=form.cleaned_data['reason_admin'],
  76. expires=form.cleaned_data['expires']
  77. )
  78. new_ban.save(force_insert=True)
  79. self.request.monitor['bans_version'] = int(self.request.monitor['bans_version']) + 1
  80. return new_ban, Message(_('New Ban has been set.'), 'success')
  81. class Edit(FormWidget):
  82. """
  83. Edit Ban
  84. """
  85. admin = site.get_action('bans')
  86. id = 'edit'
  87. name = _("Edit Ban")
  88. fallback = 'admin_bans'
  89. form = BanForm
  90. target_name = 'ban'
  91. notfound_message = _('Requested Ban could not be found.')
  92. submit_fallback = True
  93. def get_url(self, model):
  94. return reverse('admin_bans_edit', model)
  95. def get_edit_url(self, model):
  96. return self.get_url(model)
  97. def get_initial_data(self, model):
  98. return {
  99. 'type': model.type,
  100. 'ban': model.ban,
  101. 'reason_user': model.reason_user,
  102. 'reason_admin': model.reason_admin,
  103. 'expires': model.expires,
  104. }
  105. def submit_form(self, form, target):
  106. target.type = form.cleaned_data['type']
  107. target.ban = form.cleaned_data['ban']
  108. target.reason_user = form.cleaned_data['reason_user']
  109. target.reason_admin = form.cleaned_data['reason_admin']
  110. target.expires = form.cleaned_data['expires']
  111. target.save(force_update=True)
  112. self.request.monitor['bans_version'] = int(self.request.monitor['bans_version']) + 1
  113. return target, Message(_('Changes in ban have been saved.'), 'success')
  114. class Delete(ButtonWidget):
  115. """
  116. Delete Ban
  117. """
  118. admin = site.get_action('bans')
  119. id = 'delete'
  120. fallback = 'admin_bans'
  121. notfound_message = _('Requested Ban could not be found.')
  122. def action(self, target):
  123. target.delete()
  124. self.request.monitor['bans_version'] = int(self.request.monitor['bans_version']) + 1
  125. if target.type == 0:
  126. return Message(_('E-mail and username Ban "%(ban)s" has been lifted.') % {'ban': target.ban}, 'success'), False
  127. if target.type == 1:
  128. return Message(_('Username Ban "%(ban)s" has been lifted.') % {'ban': target.ban}, 'success'), False
  129. if target.type == 2:
  130. return Message(_('E-mail Ban "%(ban)s" has been lifted.') % {'ban': target.ban}, 'success'), False
  131. if target.type == 3:
  132. return Message(_('IP Ban "%(ban)s" has been lifted.') % {'ban': target.ban}, 'success'), False