From b4e9390b5672a1ccb12dd2b96dbf7a6aec0783f5 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 27 Oct 2021 14:20:12 +0000 Subject: Admin - list action: change parent (refs #4698) --- ishtar_common/admin.py | 66 +++++++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 65 insertions(+), 1 deletion(-) (limited to 'ishtar_common/admin.py') diff --git a/ishtar_common/admin.py b/ishtar_common/admin.py index 9011098c1..d1a598356 100644 --- a/ishtar_common/admin.py +++ b/ishtar_common/admin.py @@ -370,6 +370,70 @@ 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): + super(ChangeParentForm, self).__init__(post, files) + self.fields["change_parent"].choices = choices + + +class ChangeParentAdmin: + def get_actions(self, request): + action_dct = super(ChangeParentAdmin, self).get_actions(request) + if hasattr(self.model, "parent"): + action_dct["change_parent_selected"] = ( + self.change_parent_admin, + "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_children_pk = request.POST.getlist("_selected_action", []) + choices = [] + parents = self.model.objects.all() + for obj in parents: + choices.append((obj.pk, str(obj))) + + form = None + if "apply" in request.POST: + form = ChangeParentForm(choices, request.POST, request.FILES) + if form.is_valid(): + change_parent = form.cleaned_data["change_parent"] + change = [] + for child_id in selected_children_pk: + child = self.model.objects.get(pk=child_id) + child.parent_id = change_parent + child.save() + change.append(str(child)) + messages.add_message( + request, + messages.INFO, + str(_("{} parent(s) was change to {}.")).format( + " ; ".join(change), str(self.model.objects.get(pk=change_parent)) + ), + ) + return HttpResponseRedirect(return_url) + if not form: + form = ChangeParentForm(choices) + return render( + request, + "admin/change_parent.html", + { + "change_parent_form": form, + "current_action": "change_parent_selected", + "selected_items": selected_children_pk, + }, + ) + class HistorizedObjectAdmin(admin.ModelAdmin): readonly_fields = [ "history_creator", @@ -975,7 +1039,7 @@ class TownAdmin(ImportGEOJSONActionAdmin, ImportActionAdmin): admin_site.register(models_common.Town, TownAdmin) -class GeneralTypeAdmin(ImportActionAdmin, ImportJSONActionAdmin): +class GeneralTypeAdmin(ChangeParentAdmin, ImportActionAdmin, ImportJSONActionAdmin): search_fields = ( "label", "txt_idx", -- cgit v1.2.3