refactor: remove unused imports and simplify table provider extraction logic

Author: kosiew

src/utils.rs

@@ -20,15 +20,14 @@ use crate::{
     common::data_type::PyScalarValue,
     dataframe::PyDataFrame,
     dataset::Dataset,
-    errors::{py_datafusion_err, PyDataFusionError, PyDataFusionResult},
+    errors::{PyDataFusionError, PyDataFusionResult},
     table::PyTableProvider,
     TokioRuntime,
 };
 use datafusion::{
     common::ScalarValue, datasource::TableProvider, execution::context::SessionContext,
     logical_expr::Volatility,
 };
-use datafusion_ffi::table_provider::{FFI_TableProvider, ForeignTableProvider};
 use pyo3::prelude::*;
 use pyo3::{exceptions::PyValueError, types::PyCapsule};
 use std::{
@@ -134,12 +133,8 @@ pub(crate) fn table_provider_from_pycapsule(
 ) -> PyResult<Option<Arc<dyn TableProvider>>> {
     if obj.hasattr("__datafusion_table_provider__")? {
         let capsule = obj.getattr("__datafusion_table_provider__")?.call0()?;
-        let capsule = capsule.downcast::<PyCapsule>().map_err(py_datafusion_err)?;
-        validate_pycapsule(capsule, "datafusion_table_provider")?;
-
-        let provider = unsafe { capsule.reference::<FFI_TableProvider>() };
-        let provider: ForeignTableProvider = provider.into();
-        Ok(Some(Arc::new(provider) as Arc<dyn TableProvider>))
+        let provider = PyTableProvider::from_capsule(capsule)?;
+        Ok(Some(provider.into_inner()))
     } else {
         Ok(None)
     }