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)