wip: need to go back to wkb?

Author: gadomski

src/arrow.rs

@@ -1,7 +1,9 @@
 use crate::{Error, Json, Result};
 use geoarrow::table::Table;
-use pyo3::prelude::*;
+use pyo3::{prelude::*, IntoPyObjectExt};
 use pyo3_arrow::PyTable;
+use serde_json::Value;
+use stac::{Item, ItemCollection};
 
 #[pyfunction]
 pub fn from_arrow(py: Python<'_>, table: PyTable) -> PyResult<Bound<PyAny>> {
@@ -21,3 +23,23 @@ pub fn from_arrow(py: Python<'_>, table: PyTable) -> PyResult<Bound<PyAny>> {
     let item_collection = Json(item_collection).into_pyobject(py)?;
     Ok(item_collection)
 }
+
+#[pyfunction]
+pub fn to_arrow(py: Python<'_>, items: Bound<PyAny>) -> PyResult<PyObject> {
+    let value: Value = pythonize::depythonize(&items)?;
+    let item_collection = if let Value::Array(array) = value {
+        let items = array
+            .into_iter()
+            .map(|value| serde_json::from_value::<Item>(value).map_err(Error::from))
+            .collect::<Result<Vec<_>>>()?;
+        ItemCollection::from(items)
+    } else {
+        serde_json::from_value(value).map_err(Error::from)?
+    };
+    let (record_batches, schema) = stac::geoarrow::to_table(item_collection)
+        .map_err(Error::from)?
+        .into_inner();
+    let table = PyTable::try_new(record_batches, schema)?;
+    let table = table.to_arro3(py)?;
+    Ok(table.into_py_any(py)?)
+}

src/lib.rs

@@ -22,6 +22,7 @@ fn stacrs(py: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<duckdb::DuckdbClient>()?;
 
     m.add_function(wrap_pyfunction!(arrow::from_arrow, m)?)?;
+    m.add_function(wrap_pyfunction!(arrow::to_arrow, m)?)?;
     m.add_function(wrap_pyfunction!(cli::main, m)?)?;
     m.add_function(wrap_pyfunction!(migrate::migrate, m)?)?;
     m.add_function(wrap_pyfunction!(migrate::migrate_href, m)?)?;

stacrs.pyi

@@ -216,7 +216,7 @@ async def from_arrow(
     table: arro3.core.Table,
 ) -> dict[str, Any]:
     """
-    Converts an [arrow3.core.table][] to a STAC item collection.
+    Converts an [arro3.core.table][] to a STAC item collection.
 
     Requires **stacrs** to be installed with the `arrow` extra.
 
@@ -227,6 +227,21 @@ async def from_arrow(
         dict[str, Any]: The STAC item collection
     """
 
+async def to_arrow(
+    items: list[dict[str, Any]] | dict[str, Any],
+) -> arro3.core.Table:
+    """
+    Converts items to an [arro3.core.table][].
+
+    Requires **stacrs** to be installed with the `arrow` extra.
+
+    Args:
+        items: Either an iterable of items or a item collection
+
+    Returns:
+        arro3.core.Table: The table
+    """
+
 async def search(
     href: str,
     *,

tests/test_arrow.py

@@ -0,0 +1,13 @@
+from typing import Any
+
+import pytest
+import stacrs
+from geopandas import GeoDataFrame
+
+pytest.importorskip("arro3.core")
+
+
+def test_to_arrow(item: dict[str, Any]) -> None:
+    table = stacrs.to_arrow([item])
+    data_frame = GeoDataFrame.from_arrow(table)
+    assert len(data_frame) == 1