fix(exports): Fix xlsx export extension (#20629)

PostHog · Feb 29, 2024 · c033a7c · c033a7c
1 parent 3796fba
commit c033a7c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/posthog/models/exported_asset.py b/posthog/models/exported_asset.py
@@ -83,7 +83,7 @@ def has_content(self):
 
     @property
     def filename(self):
-        ext = self.export_format.split("/")[1]
+        ext = self.ExportFormat(self.export_format).name.lower()
         filename = "export"
 
         if self.export_context and self.export_context.get("filename"):

diff --git a/posthog/tasks/exports/test/test_csv_exporter.py b/posthog/tasks/exports/test/test_csv_exporter.py
@@ -268,6 +268,7 @@ def test_csv_exporter_excel(self, mocked_object_storage_write: Any, mocked_uuidt
         with self.settings(OBJECT_STORAGE_ENABLED=True, OBJECT_STORAGE_EXPORTS_FOLDER="Test-Exports"):
             csv_exporter.export_tabular(exported_asset)
 
+            assert exported_asset.filename == "export.xlsx"
             assert exported_asset.content_location is None
 
             wb = load_workbook(filename=BytesIO(exported_asset.content))