1616 - Local patch to resolve CSV export issue

Author: bferguso

bcap/search/search_export.py

@@ -1,6 +1,7 @@
 import codecs
 from io import StringIO
 
+from arches.app.models import models as arches_models
 from arches.app.search.search_export import SearchResultsExporter
 
 
@@ -13,6 +14,24 @@ class BCAPSearchResultsExporter(SearchResultsExporter):
     def _prepend_bom(inner):
         return StringIO(codecs.BOM_UTF8.decode("utf-8") + inner.getvalue())
 
+    def return_ordered_header(self, graphid, export_type):
+        headers = super().return_ordered_header(graphid, export_type)
+        # CardXNodeXWidget only covers widget-bearing nodes, so collector nodes
+        # (e.g. resource-instance collectors like publication_reference) that are
+        # marked exportable are absent from the parent's result even though
+        # flatten_tiles() will emit them into every instance dict.  Appending
+        # them here prevents DictWriter from raising ValueError on export.
+        if export_type == "csv":
+            in_headers = set(headers)
+            extras = list(
+                arches_models.Node.objects.filter(graph_id=graphid, exportable=True)
+                .exclude(datatype="semantic")
+                .exclude(name__in=in_headers)
+                .values_list("name", flat=True)
+            )
+            headers.extend(extras)
+        return headers
+
     def to_csv(self, instances, headers, name):
         result = super().to_csv(instances, headers, name)
         result["outputfile"] = self._prepend_bom(result["outputfile"])