fix: refactor table provider registration logic for improved clarity

Author: kosiew

src/catalog.rs

@@ -199,20 +199,16 @@ impl PySchema {
     }
 
     fn register_table(&self, name: &str, table_provider: Bound<'_, PyAny>) -> PyResult<()> {
-        let provider = if let Some(provider) = table_provider_from_pycapsule(&table_provider)? {
+        let provider = if let Ok(py_table) = table_provider.extract::<PyTable>() {
+            py_table.table
+        } else if let Ok(py_provider) = table_provider.extract::<PyTableProvider>() {
+            py_provider.into_inner()
+        } else if let Some(provider) = table_provider_from_pycapsule(&table_provider)? {
             provider
         } else {
-            match table_provider.extract::<PyTable>() {
-                Ok(py_table) => py_table.table,
-                Err(_) => match table_provider.extract::<PyTableProvider>() {
-                    Ok(py_provider) => py_provider.into_inner(),
-                    Err(_) => {
-                        let py = table_provider.py();
-                        let provider = Dataset::new(&table_provider, py)?;
-                        Arc::new(provider) as Arc<dyn TableProvider + Send>
-                    }
-                },
-            }
+            let py = table_provider.py();
+            let provider = Dataset::new(&table_provider, py)?;
+            Arc::new(provider) as Arc<dyn TableProvider + Send>
         };
 
         let _ = self

src/context.rs

@@ -610,12 +610,12 @@ impl PySessionContext {
         name: &str,
         table_provider: Bound<'_, PyAny>,
     ) -> PyDataFusionResult<()> {
-        let provider = if let Some(provider) = table_provider_from_pycapsule(&table_provider)? {
-            provider
-        } else if let Ok(py_table) = table_provider.extract::<PyTable>() {
+        let provider = if let Ok(py_table) = table_provider.extract::<PyTable>() {
             py_table.table()
         } else if let Ok(py_provider) = table_provider.extract::<PyTableProvider>() {
             py_provider.into_inner()
+        } else if let Some(provider) = table_provider_from_pycapsule(&table_provider)? {
+            provider
         } else {
             return Err(crate::errors::PyDataFusionError::Common(
                 "Expected a Table or TableProvider. Convert DataFrames with \"DataFrame.into_view()\" or \"TableProvider.from_dataframe()\".".to_string(),