From f83a193916bb52ddbd6f3dc458e39de54729c976 Mon Sep 17 00:00:00 2001
From: Stefan Wehrmeyer <mail@stefanwehrmeyer.com>
Date: Mon, 14 Mar 2022 14:29:02 +0100
Subject: [PATCH] Add sections progress plugin

---
 froide_govplan/models.py                                 | 9 +++++++++
 .../templates/froide_govplan/plugins/sections.html       | 5 +++--
 froide_govplan/templatetags/govplan.py                   | 5 +++++
 froide_govplan/views.py                                  | 8 +-------
 4 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/froide_govplan/models.py b/froide_govplan/models.py
index bafe900..2eedc91 100644
--- a/froide_govplan/models.py
+++ b/froide_govplan/models.py
@@ -354,6 +354,15 @@ class GovernmentPlanSection(models.Model):
     def get_absolute_domain_url(self):
         return settings.SITE_URL + self.get_absolute_url()
 
+    def get_plans(self):
+        return (
+            GovernmentPlan.objects.filter(
+                categories__in=self.categories.all(), government=self.government
+            )
+            .distinct()
+            .order_by("title")
+        )
+
 
 if CMSPlugin:
 
diff --git a/froide_govplan/templates/froide_govplan/plugins/sections.html b/froide_govplan/templates/froide_govplan/plugins/sections.html
index f1cfc02..3864174 100644
--- a/froide_govplan/templates/froide_govplan/plugins/sections.html
+++ b/froide_govplan/templates/froide_govplan/plugins/sections.html
@@ -1,4 +1,5 @@
 {% load thumbnail %}
+{% load govplan %}
 
 <style>
   {% for section in sections %}
@@ -27,11 +28,11 @@
             {% endif %}
           </div>
           <div class="flex-grow-1 tight-margin p-3 p-md-4 cms-plugin cms-plugin-87888">
-            progress bar
+            {% get_section_progress section %}
           </div>
         </div>
       </div>
     </a>
   </div>
   {% endfor %}
-</div>
\ No newline at end of file
+</div>
diff --git a/froide_govplan/templatetags/govplan.py b/froide_govplan/templatetags/govplan.py
index 94e7231..a731c00 100644
--- a/froide_govplan/templatetags/govplan.py
+++ b/froide_govplan/templatetags/govplan.py
@@ -35,3 +35,8 @@ def get_plan_progress(object_list):
 
     value, label
     return {"count": len(object_list), "sections": sections}
+
+
+@register.inclusion_tag("froide_govplan/plugins/progress.html")
+def get_section_progress(section):
+    return {"object_list": section.get_plans()}
diff --git a/froide_govplan/views.py b/froide_govplan/views.py
index 7a51f5e..1bf482b 100644
--- a/froide_govplan/views.py
+++ b/froide_govplan/views.py
@@ -27,13 +27,7 @@ class GovPlanSectionDetailView(GovernmentMixin, DetailView):
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
-        context["plans"] = (
-            GovernmentPlan.objects.filter(
-                categories__in=self.object.categories.all(), government=self.government
-            )
-            .distinct()
-            .order_by("title")
-        )
+        context["plans"] = self.context["object"].get_plans()
         return context