added topojson conversion option

geospatial-data-converter/app.py

@@ -68,8 +68,6 @@ elif st.session_state.uploaded_file is not None:
         os.path.basename(st.session_state.uploaded_file.name),
     )
     st.session_state.gdf = read_file(st.session_state.uploaded_file)
-    # except AttributeError:
-    #     pass # there is a lingering file from the previous instance, but it can be ignored
 
 if st.session_state.gdf is not None:
     st.selectbox(

geospatial-data-converter/utils.py

@@ -1,18 +1,20 @@
 import io
 import os
 import zipfile
-from tempfile import TemporaryDirectory
-from typing import BinaryIO
 import geopandas as gpd
+import topojson
 
+from tempfile import TemporaryDirectory
+from typing import BinaryIO
 from kml_tricks import load_ge_data
 
 output_format_dict = {
-    "ESRI Shapefile": ("shp", "zip", "application/zip"),  # must be zipped
-    "OpenFileGDB": ("gdb", "zip", "application/zip"),  # must be zipped
-    "GeoJSON": ("geojson", "geojson", "application/geo+json"),
     "CSV": ("csv", "csv", "text/csv"),
     "KML": ("kml", "kml", "application/vnd.google-earth.kml+xml"),
+    "GeoJSON": ("geojson", "geojson", "application/geo+json"),
+    "TopoJSON": ("topojson", "topojson", "application/json"),
+    "ESRI Shapefile": ("shp", "zip", "application/zip"),  # must be zipped
+    "OpenFileGDB": ("gdb", "zip", "application/zip"),  # must be zipped
 }
 
 
@@ -34,9 +36,6 @@ def read_file(file: BinaryIO, *args, **kwargs) -> gpd.GeoDataFrame:
     elif ext in ("kml", "kmz"):
         with TemporaryDirectory() as tmp_dir:
             tmp_file_path = os.path.join(tmp_dir, f"{basename}.{ext}")
-            print(file.name)
-            print(basename)
-            print(tmp_file_path)
             with open(tmp_file_path, "wb") as tmp_file:
                 tmp_file.write(file.read())
             return load_ge_data(tmp_file_path)
@@ -65,6 +64,9 @@ def convert(gdf: gpd.GeoDataFrame, output_name: str, output_format: str) -> byte
         out_path = os.path.join(tmpdir, output_name)
         if output_format == "CSV":
             gdf.to_csv(out_path)
+        elif output_format == "TopoJSON":
+            topojson_data = topojson.Topology(gdf)
+            topojson_data.to_json(out_path)
         else:
             gdf.to_file(out_path, driver=output_format, engine="pyogrio")
 

requirements.txt

@@ -5,3 +5,4 @@ lxml==4.9.3
 pyogrio==0.7.2
 restgdf==0.9.7
 streamlit==1.29.0
+topojson

tests/test_conversions.py

@@ -1,26 +1,19 @@
 import os
 import pytest
-from utils import convert, read_file
+from utils import convert, read_file, output_format_dict
 
 input_exts = ["kml", "kmz", "geojson", "zip"]
-output_exts = ["KML", "ESRI Shapefile", "GeoJSON", "CSV", "OpenFileGDB"]
-output_format_dict = {
-    "ESRI Shapefile": "shp",
-    "GeoJSON": "geojson",
-    "CSV": "csv",
-    "KML": "kml",
-    "OpenFileGDB": "gdb",
-}
+output_exts = output_format_dict.keys()
 
 
 @pytest.mark.parametrize("in_ext", input_exts)
 @pytest.mark.parametrize("out_ext", output_exts)
-def test_kml_coversion(in_ext: str, out_ext: str) -> None:
+def test_coversion(in_ext: str, out_ext: str) -> None:
     test_file = f"test.{in_ext}"
     test_file_path = os.path.join(os.getcwd(), "tests", "test_data", test_file)
     with open(test_file_path, "rb") as f:
         kml = read_file(f)
-    out_file = f"test.{output_format_dict[out_ext]}"
+    out_file = f"test.{output_format_dict[out_ext][0]}"
     converted_data = convert(kml, out_file, out_ext)
     with open("test.kml", "wb") as f:
         f.write(converted_data)