from django.core.urlresolvers import reverse as django_reverse import floppyforms as forms from django.utils.translation import ugettext as _ from misago.admin import site from misago.apps.admin.widgets import * from misago.forms import Form from misago.models import Rank from misago.utils.strings import slugify from misago.apps.admin.ranks.forms import RankForm def reverse(route, target=None): if target: return django_reverse(route, kwargs={'target': target.pk, 'slug': slugify(target.name)}) return django_reverse(route) """ Views """ class List(ListWidget): admin = site.get_action('ranks') id = 'list' columns = ( ('rank', _("Rank")), ) table_form_button = _('Reorder Ranks') nothing_checked_message = _('You have to check at least one rank.') actions = ( ('delete', _("Delete selected ranks"), _("Are you sure you want to delete selected ranks?")), ) def get_table_form(self, page_items): order_form = {} # Build choices list choices = [] for i in range(0, len(page_items)): choices.append([str(i), i + 1]) # Build selectors list position = 0 for item in page_items: order_form['pos_' + str(item.pk)] = forms.ChoiceField(choices=choices, initial=str(position)) position += 1 # Turn dict into object return type('OrderRanksForm', (Form,), order_form) def table_action(self, page_items, cleaned_data): for item in page_items: item.order = cleaned_data['pos_' + str(item.pk)] item.save(force_update=True) return Message(_('Ranks order has been changed'), 'success'), reverse('admin_ranks') def sort_items(self, page_items, sorting_method): return page_items.order_by('order') def get_item_actions(self, item): return ( self.action('pencil', _("Edit Rank"), reverse('admin_ranks_edit', item)), self.action('remove', _("Delete Rank"), reverse('admin_ranks_delete', item), post=True, prompt=_("Are you sure you want to delete this rank?")), ) def action_delete(self, items, checked): Rank.objects.filter(id__in=checked).delete() return Message(_('Selected ranks have been deleted successfully.'), 'success'), reverse('admin_ranks') class New(FormWidget): admin = site.get_action('ranks') id = 'new' fallback = 'admin_ranks' form = RankForm submit_button = _("Save Rank") def get_new_link(self, model): return reverse('admin_ranks_new') def get_edit_link(self, model): return reverse('admin_ranks_edit', model) def submit_form(self, form, target): position = 0 last_rank = Rank.objects.latest('order') new_rank = Rank( name=form.cleaned_data['name'], slug=slugify(form.cleaned_data['name']), description=form.cleaned_data['description'], style=form.cleaned_data['style'], title=form.cleaned_data['title'], special=form.cleaned_data['special'], as_tab=form.cleaned_data['as_tab'], on_index=form.cleaned_data['on_index'], order=(last_rank.order + 1 if last_rank else 0), criteria=form.cleaned_data['criteria'] ) new_rank.save(force_insert=True) for role in form.cleaned_data['roles']: new_rank.roles.add(role) return new_rank, Message(_('New Rank has been created.'), 'success') class Edit(FormWidget): admin = site.get_action('ranks') id = 'edit' name = _("Edit Rank") fallback = 'admin_ranks' form = RankForm target_name = 'name' notfound_message = _('Requested Rank could not be found.') translate_target_name = True submit_fallback = True def get_link(self, model): return reverse('admin_ranks_edit', model) def get_edit_link(self, model): return self.get_link(model) def get_initial_data(self, model): return { 'name': model.name, 'description': model.description, 'style': model.style, 'title': model.title, 'special': model.special, 'as_tab': model.as_tab, 'on_index': model.on_index, 'criteria': model.criteria, 'roles': model.roles.all(), } def submit_form(self, form, target): target.name = form.cleaned_data['name'] target.slug = slugify(form.cleaned_data['name']) target.description = form.cleaned_data['description'] target.style = form.cleaned_data['style'] target.title = form.cleaned_data['title'] target.special = form.cleaned_data['special'] target.as_tab = form.cleaned_data['as_tab'] target.on_index = form.cleaned_data['on_index'] target.criteria = form.cleaned_data['criteria'] target.save(force_update=True) if self.request.user.is_god(): target.roles.clear() else: target.roles.remove(*target.roles.filter(protected=False)) for role in form.cleaned_data['roles']: target.roles.add(role) target.user_set.update(acl_key=None) return target, Message(_('Changes in rank "%(name)s" have been saved.') % {'name': self.original_name}, 'success') class Delete(ButtonWidget): admin = site.get_action('ranks') id = 'delete' fallback = 'admin_ranks' notfound_message = _('Requested Rank could not be found.') def action(self, target): target.delete() return Message(_('Rank "%(name)s" has been deleted.') % {'name': _(target.name)}, 'success'), False