Create ingest/export.py

ingest/export.py

@@ -1,9 +1,21 @@
+import io
 import zipfile
-import json
-import tempfile
 
-def export_zip(results):
-    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".zip")
-    with zipfile.ZipFile(tmp.name, "w") as z:
-        z.writestr("results.json", json.dumps(results, indent=2))
-    return tmp.name
+def export_results_zip(results):
+    """
+    results: list[dict]
+    returns: BytesIO zip
+    """
+    buffer = io.BytesIO()
+    with zipfile.ZipFile(buffer, "w", zipfile.ZIP_DEFLATED) as zf:
+        for i, r in enumerate(results, start=1):
+            content = (
+                f"Source: {r.get('source')}\n"
+                f"Title: {r.get('title')}\n"
+                f"URL: {r.get('url')}\n\n"
+                f"Snippet:\n{r.get('snippet')}"
+            )
+            zf.writestr(f"{i:03d}_{r.get('source')}.txt", content)
+
+    buffer.seek(0)
+    return buffer