--wip-- [skip ci]

Author: gadomski

pyproject.toml

@@ -43,6 +43,7 @@ asyncio_default_fixture_loop_scope = "function"
 
 [dependency-groups]
 dev = [
+    "geopandas>=1.0.1",
     "maturin>=1.7.4",
     "mypy>=1.11.2",
     "pyarrow>=19.0.1",

src/duckdb.rs

@@ -69,6 +69,48 @@ impl DuckdbClient {
         Ok(dict)
     }
 
+    #[pyo3(signature = (href, *, intersects=None, ids=None, collections=None, limit=None, bbox=None, datetime=None, include=None, exclude=None, sortby=None, filter=None, query=None, **kwargs))]
+    fn search_to_arrow<'py>(
+        &self,
+        py: Python<'py>,
+        href: String,
+        intersects: Option<StringOrDict>,
+        ids: Option<StringOrList>,
+        collections: Option<StringOrList>,
+        limit: Option<u64>,
+        bbox: Option<Vec<f64>>,
+        datetime: Option<String>,
+        include: Option<StringOrList>,
+        exclude: Option<StringOrList>,
+        sortby: Option<StringOrList>,
+        filter: Option<StringOrDict>,
+        query: Option<Bound<'py, PyDict>>,
+        kwargs: Option<Bound<'py, PyDict>>,
+    ) -> Result<Bound<'py, PyObject>> {
+        let search = stac_api::python::search(
+            intersects,
+            ids,
+            collections,
+            limit,
+            bbox,
+            datetime,
+            include,
+            exclude,
+            sortby,
+            filter,
+            query,
+            kwargs,
+        )?;
+        let record_batches = {
+            let client = self
+                .0
+                .lock()
+                .map_err(|err| PyException::new_err(err.to_string()))?;
+            client.search_to_arrow(&href, search)?
+        };
+        todo!()
+    }
+
     fn get_collections<'py>(&self, py: Python<'py>, href: String) -> Result<Bound<'py, PyList>> {
         let client = self
             .0

tests/test_duckdb.py

@@ -1,4 +1,5 @@
 import pytest
+from geopandas import GeoDataFrame
 from stacrs import DuckdbClient
 
 
@@ -37,3 +38,9 @@ def test_get_collections(client: DuckdbClient) -> None:
 
 def test_init_with_config() -> None:
     DuckdbClient(use_s3_credential_chain=True, use_hive_partitioning=True)
+
+
+def test_search_to_arrow(client: DuckdbClient) -> None:
+    table = client.search_to_arrow("data/100-sentinel-2-items.parquet")
+    data_frame = GeoDataFrame.from_arrow(table)
+    assert len(data_frame) == 100