feat: add to search-to

Author: gadomski

python/rustac/rustac.pyi

@@ -350,6 +350,7 @@ async def search_to(
     filter: str | dict[str, Any] | None = None,
     query: dict[str, Any] | None = None,
     format: str | None = None,
+    parquet_compression: str | None = None,
     store: AnyObjectStore | None = None,
     use_duckdb: bool | None = None,
 ) -> int:
@@ -389,6 +390,10 @@ async def search_to(
             It is recommended to use filter instead, if possible.
         format: The output format. If none, will be inferred from
             the outfile extension, and if that fails will fall back to compact JSON.
+        parquet_compression: If writing stac-geoparquet, sets the compression
+            algorithm.
+            https://docs.rs/parquet/latest/parquet/basic/enum.Compression.html
+            is a list of what's available.
         store: An optional [ObjectStore][]
         use_duckdb: Query with DuckDB. If None and the href has a
             'parquet' or 'geoparquet' extension, will be set to True. Defaults

src/search.rs

@@ -58,7 +58,7 @@ pub fn search<'py>(
 }
 
 #[pyfunction]
-#[pyo3(signature = (outfile, href, *, intersects=None, ids=None, collections=None, max_items=None, limit=None, bbox=None, datetime=None, include=None, exclude=None, sortby=None, filter=None, query=None, format=None, store=None, use_duckdb=None, **kwargs))]
+#[pyo3(signature = (outfile, href, *, intersects=None, ids=None, collections=None, max_items=None, limit=None, bbox=None, datetime=None, include=None, exclude=None, sortby=None, filter=None, query=None, format=None, parquet_compression=None, store=None, use_duckdb=None, **kwargs))]
 #[allow(clippy::too_many_arguments)]
 pub fn search_to<'py>(
     py: Python<'py>,
@@ -77,6 +77,7 @@ pub fn search_to<'py>(
     filter: Option<StringOrDict>,
     query: Option<Bound<'py, PyDict>>,
     format: Option<String>,
+    parquet_compression: Option<String>,
     store: Option<AnyObjectStore>,
     use_duckdb: Option<bool>,
     kwargs: Option<Bound<'_, PyDict>>,
@@ -95,12 +96,18 @@ pub fn search_to<'py>(
         query,
         kwargs,
     )?;
-    let format = format
+    let mut format = format
         .map(|s| s.parse())
         .transpose()
         .map_err(Error::from)?
         .or_else(|| Format::infer_from_href(&outfile))
         .unwrap_or_default();
+    if matches!(format, Format::Geoparquet(_)) {
+        if let Some(parquet_compression) = parquet_compression {
+            tracing::debug!("setting parquet compression: {parquet_compression}");
+            format = Format::Geoparquet(Some(parquet_compression.parse().map_err(Error::from)?));
+        }
+    }
     if use_duckdb
         .unwrap_or_else(|| matches!(Format::infer_from_href(&href), Some(Format::Geoparquet(_))))
     {

src/write.rs

@@ -6,14 +6,14 @@ use stac::{Item, ItemCollection};
 use stac_io::{Format, StacStore};
 
 #[pyfunction]
-#[pyo3(signature = (href, value, *, format=None, store=None, parquet_compression=None))]
+#[pyo3(signature = (href, value, *, format=None, parquet_compression=None, store=None))]
 pub fn write<'py>(
     py: Python<'py>,
     href: String,
     value: Bound<'_, PyAny>,
     format: Option<String>,
-    store: Option<AnyObjectStore>,
     parquet_compression: Option<String>,
+    store: Option<AnyObjectStore>,
 ) -> PyResult<Bound<'py, PyAny>> {
     let value: Value = pythonize::depythonize(&value)?;
     let value = if let Value::Array(array) = value {