fix court entity profile lookup

laws-africa · Sep 5, 2024 · 32a194e · 32a194e
1 parent 787ae48
commit 32a194e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/peachjam/templates/peachjam/court_detail.html b/peachjam/templates/peachjam/court_detail.html
@@ -34,13 +34,6 @@
     </nav>
   </div>
 {% endblock %}
-{% block entity-profile %}
-  {% with entity_profile=court.entity_profile.first entity_profile_title=court.name %}
-    {% if entity_profile %}
-      {% include 'peachjam/_entity_profile.html' %}
-    {% endif %}
-  {% endwith %}
-{% endblock %}
 {% block page-title %}
   {% if not entity_profile %}<h1 class="mt-4">{{ page_title }}</h1>{% endif %}
 {% endblock %}

diff --git a/peachjam/views/courts.py b/peachjam/views/courts.py
@@ -168,6 +168,10 @@ def get_context_data(self, **kwargs):
         context["all_years_url"] = self.court.get_absolute_url()
         return context
 
+    def add_entity_profile(self, context):
+        context["entity_profile"] = self.court.entity_profile.first()
+        context["entity_profile_title"] = self.court.name
+
 
 class CourtYearView(YearMixin, CourtDetailView):
     def get_context_data(self, **kwargs):