Added geoparquet format

hotosm · Nov 20, 2023 · b2932b1 · b2932b1
1 parent 3ad40c8
commit b2932b1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/app.py b/src/app.py
@@ -495,6 +495,18 @@ def ogr_export(query, outputtype, working_dir, dump_temp_path, params):
             )
             run_ogr2ogr_cmd(cmd)
 
+        if outputtype == RawDataOutputType.GEOPARQUET.value:
+            cmd = """ogr2ogr -overwrite -f Parquet {export_path} PG:"host={host} port={port} user={username} dbname={db} password={password}" -sql @"{pg_sql_select}" -lco ENCODING=UTF-8 -progress""".format(
+                export_path=dump_temp_path,
+                host=db_items.get("host"),
+                port=db_items.get("port"),
+                username=db_items.get("user"),
+                db=db_items.get("dbname"),
+                password=db_items.get("password"),
+                pg_sql_select=query_path,
+            )
+            run_ogr2ogr_cmd(cmd)
+
         if outputtype == RawDataOutputType.PGDUMP.value:
             cmd = """ogr2ogr -overwrite --config PG_USE_COPY YES -f PGDump {export_path} PG:"host={host} port={port} user={username} dbname={db} password={password}" -sql @"{pg_sql_select}" -lco SRID=4326 -progress""".format(
                 export_path=dump_temp_path,

diff --git a/src/validation/models.py b/src/validation/models.py
@@ -52,6 +52,7 @@ class RawDataOutputType(Enum):
     GEOPACKAGE = "gpkg"
     PGDUMP = "sql"
     CSV = "csv"
+    GEOPARQUET = "parquet"
     if ENABLE_TILES:
         MBTILES = "mbtiles"
         PMTILES = "pmtiles"  ## EXPERIMENTAL