✨ sites - sheet: add heritage fields

1 files changed, 13 insertions, 12 deletions

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index d2f6861ed..62c0bcf93 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -1042,22 +1042,23 @@ class ArchaeologicalSite(
         return self.towns_label() or "-"
 
     def _generate_cached_remains(self):
-        q = self.remains
-        if not self.remains.exists():
-            return "-"
-        return " & ".join(list(q.values_list("label", flat=True)))
+        return self._regenerate_many_to_many_for_cache("remains")
 
     def _generate_cached_periods(self):
-        q = self.periods
-        if not self.periods.exists():
-            return "-"
-        return " & ".join(list(q.values_list("label", flat=True)))
+        return self._regenerate_many_to_many_for_cache("periods")
 
     def _generate_cached_types(self):
-        q = self.types
-        if not self.types.exists():
-            return "-"
-        return " & ".join(list(q.values_list("label", flat=True)))
+        return self._regenerate_many_to_many_for_cache("types")
+
+    def _generate_cached_current_states(self):
+        return self._regenerate_many_to_many_for_cache("current_states")
+
+    def _generate_cached_heritage_interests(self):
+        return self._regenerate_many_to_many_for_cache("heritage_interests")
+
+    def _generate_cached_heritage_environmental_protections(self):
+        return self._regenerate_many_to_many_for_cache(
+            "heritage_environmental_protections")
 
     def natural_key(self):
         return (self.reference,)