From 14d7f2b249855579eed8f71a597369777f375c3c Mon Sep 17 00:00:00 2001
From: Stefan Wehrmeyer <mail@stefanwehrmeyer.com>
Date: Thu, 17 Mar 2022 09:18:45 +0100
Subject: [PATCH] Pre-select plan when adding update from toolbar

---
 froide_govplan/cms_toolbars.py                        |  1 +
 .../governmentplanupdate/change_form.html             | 11 +++++++++++
 2 files changed, 12 insertions(+)
 create mode 100644 froide_govplan/templates/admin/froide_govplan/governmentplanupdate/change_form.html

diff --git a/froide_govplan/cms_toolbars.py b/froide_govplan/cms_toolbars.py
index c173cc7..4f46edc 100644
--- a/froide_govplan/cms_toolbars.py
+++ b/froide_govplan/cms_toolbars.py
@@ -33,6 +33,7 @@ class GovPlanToolbar(CMSToolbar):
                 "admin:froide_govplan_governmentplanupdate_add",
                 current_app="govplanadmin",
             )
+            url = "{}?plan={}".format(url, govplan.id)
             menu.add_modal_item(_("Add update"), url=url)
 
 
diff --git a/froide_govplan/templates/admin/froide_govplan/governmentplanupdate/change_form.html b/froide_govplan/templates/admin/froide_govplan/governmentplanupdate/change_form.html
new file mode 100644
index 0000000..755774d
--- /dev/null
+++ b/froide_govplan/templates/admin/froide_govplan/governmentplanupdate/change_form.html
@@ -0,0 +1,11 @@
+{% extends "admin/change_form.html" %}
+
+{% block after_field_sets %}
+  {% if not original.pk %}
+    <script>
+    (function(){
+      document.getElementById('id_plan').value = new URLSearchParams(window.location.search).get("plan") || ""
+    }())
+    </script>
+  {% endif %}
+{% endblock %}