wip

Author: gadomski

src/arrow.rs

@@ -0,0 +1,8 @@
+use pyo3::prelude::*;
+use pyo3_arrow::PyTable;
+
+#[pyfunction]
+pub fn from_arrow(table: &Bound<PyTable>) -> () {
+    dbg!(table);
+    todo!()
+}

src/duckdb.rs

@@ -2,7 +2,7 @@ use crate::Result;
 use pyo3::{
     exceptions::PyException,
     prelude::*,
-    types::{PyCapsule, PyDict, PyList},
+    types::{PyDict, PyList},
     IntoPyObjectExt,
 };
 use pyo3_arrow::PyTable;
@@ -134,9 +134,3 @@ impl DuckdbClient {
         Ok(collections)
     }
 }
-
-// TODO this probably should live in a different mod
-#[pyfunction]
-pub fn from_arrow(table: &Bound<PyCapsule>) -> () {
-    todo!()
-}

src/lib.rs

@@ -1,5 +1,6 @@
 #![deny(unused_crate_dependencies)]
 
+mod arrow;
 mod cli;
 mod duckdb;
 mod error;
@@ -20,11 +21,11 @@ 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!(cli::main, m)?)?;
     m.add_function(wrap_pyfunction!(migrate::migrate, m)?)?;
     m.add_function(wrap_pyfunction!(migrate::migrate_href, m)?)?;
     m.add_function(wrap_pyfunction!(read::read, m)?)?;
-    m.add_function(wrap_pyfunction!(duckdb::from_arrow, m)?)?;
     m.add_function(wrap_pyfunction!(search::search, m)?)?;
     m.add_function(wrap_pyfunction!(search::search_to, m)?)?;
     m.add_function(wrap_pyfunction!(version::version, m)?)?;