diff options
Diffstat (limited to 'ishtar_common/admin.py')
-rw-r--r-- | ishtar_common/admin.py | 58 |
1 files changed, 57 insertions, 1 deletions
diff --git a/ishtar_common/admin.py b/ishtar_common/admin.py index 03e736f4e..68e15404b 100644 --- a/ishtar_common/admin.py +++ b/ishtar_common/admin.py @@ -368,6 +368,62 @@ TokenAdmin.raw_id_fields = ("user",) admin_site.register(Token, TokenAdmin) +class ChangeParentForm(forms.Form): + change_parent = forms.ChoiceField(label=_("Change parent"), choices=[]) + + def __init__(self, choices, post=None, files=None): + if post: + super(ChangeParentForm, self).__init__(post, files) + else: + super(ChangeParentForm, self).__init__() + self.fields["change_parent"].choices = choices + +class ChangeParentAdmin: + def get_actions(self, request): + action_dct = super(ChangeParentAdmin, self).get_actions(request) + action_dct["change_parent_selected"] = ( + self.change_parent, + "change_parent_selected", + _("Change parent of selected item(s)"), + ) + return action_dct + + def change_parent_admin(self, modeladmin, request, queryset): + return_url = reverse( + "admin:%s_%s_changelist" + % (self.model._meta.app_label, self.model._meta.model_name) + ) + if not request.POST: + return HttpResponseRedirect(return_url) + + selected = request.POST.getlist("_selected_action", []) + items = {} + choices = [] + for pk in selected: + obj = self.model._get_parent_types(pk) + choices.append(obj.pk, str(obj)) + items[str(obj.pk)] = obj + + form = None + if "apply" in request.POST: + form = ChangeParentForm(choices, request.POST, request.FILES) + if form.is_valid(): + + messages.add_message( + request, + messages.INFO, + ) + return HttpResponseRedirect(return_url) + if not form: + form = ChangeParentForm(choices) + return render( + request, + "admin/change_parent.html", + { + "selected_item": selected, + }, + ) + class HistorizedObjectAdmin(admin.ModelAdmin): readonly_fields = [ "history_creator", @@ -968,7 +1024,7 @@ class TownAdmin(ImportGEOJSONActionAdmin, ImportActionAdmin): admin_site.register(models_common.Town, TownAdmin) -class GeneralTypeAdmin(ImportActionAdmin, ImportJSONActionAdmin): +class GeneralTypeAdmin(ImportActionAdmin, ImportJSONActionAdmin, ChangeParentAdmin): search_fields = ( "label", "txt_idx", |