call func from get_context_data - may override in tests

clinicedc · Nov 28, 2024 · 0970666 · 0970666
1 parent 5f8a06d
commit 0970666
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/edc_sites/view_mixins.py b/edc_sites/view_mixins.py
@@ -8,6 +8,10 @@
 
 class SiteViewMixin:
     def get_context_data(self, **kwargs) -> dict:
+        kwargs = self.get_context_data_for_sites(**kwargs)
+        return super().get_context_data(**kwargs)
+
+    def get_context_data_for_sites(self, **kwargs):
         try:
             site_profile = SiteProfile.objects.get(site__id=self.request.site.id)
         except ObjectDoesNotExist:
@@ -21,4 +25,4 @@ def get_context_data(self, **kwargs) -> dict:
                     "Unable to determine site profile 'title'. No sites have been registered! "
                 )
             raise
-        return super().get_context_data(**kwargs)
+        return kwargs