Merge pull request #204 from akrherz/geojson

Cleanup a geojson export

.pre-commit-config.yaml

@@ -2,7 +2,7 @@ ci:
   autoupdate_schedule: quarterly
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: "v0.4.7"
+    rev: "v0.4.10"
     hooks:
       - id: ruff
         args: [--fix, --exit-non-zero-on-fix]

environment.yml

@@ -9,6 +9,7 @@ dependencies:
   # For the future warning
   - pandas>=2.2
   - pygrib
+  - pyogrio
   - pyproj
   - pytest
   - pytest-runner

src/iemws/util.py

@@ -1,9 +1,9 @@
 """Helpers."""
 
 import logging
-import tempfile
 from contextlib import contextmanager
 from functools import wraps
+from io import BytesIO
 from typing import Callable
 
 from fastapi import Request, Response
@@ -67,10 +67,10 @@ def deliver_df(df: DataFrame, fmt: str):
         if df.empty:
             res = '{"type": "FeatureCollection", "features": []}'
         else:
-            with tempfile.NamedTemporaryFile("w", delete=True) as tmp:
-                df.to_file(tmp.name, driver="GeoJSON")
-                with open(tmp.name, encoding="utf8") as fh:
-                    res = fh.read()
+            with BytesIO() as tmp:
+                df.crs = "EPSG:4326"
+                df.to_file(tmp, driver="GeoJSON", engine="pyogrio")
+                res = tmp.getvalue()
     return Response(res, media_type=MEDIATYPES[fmt])