summaryrefslogtreecommitdiff
path: root/ishtar_common/admin.py
diff options
context:
space:
mode:
Diffstat (limited to 'ishtar_common/admin.py')
-rw-r--r--ishtar_common/admin.py66
1 files changed, 65 insertions, 1 deletions
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",