refactor: move py_obj_to_scalar_value function to utils module

Author: kosiew

src/config.rs

@@ -22,6 +22,7 @@ use datafusion::common::ScalarValue;
 use datafusion::config::ConfigOptions;
 
 use crate::errors::PyDataFusionResult;
+use crate::utils::py_obj_to_scalar_value;
 
 #[pyclass(name = "Config", module = "datafusion", subclass)]
 #[derive(Clone)]
@@ -82,20 +83,3 @@ impl PyConfig {
         }
     }
 }
-
-/// Convert a python object to a ScalarValue
-fn py_obj_to_scalar_value(py: Python, obj: PyObject) -> ScalarValue {
-    if let Ok(value) = obj.extract::<bool>(py) {
-        ScalarValue::Boolean(Some(value))
-    } else if let Ok(value) = obj.extract::<i64>(py) {
-        ScalarValue::Int64(Some(value))
-    } else if let Ok(value) = obj.extract::<u64>(py) {
-        ScalarValue::UInt64(Some(value))
-    } else if let Ok(value) = obj.extract::<f64>(py) {
-        ScalarValue::Float64(Some(value))
-    } else if let Ok(value) = obj.extract::<String>(py) {
-        ScalarValue::Utf8(Some(value))
-    } else {
-        panic!("Unsupported value type")
-    }
-}

src/utils.rs

@@ -87,3 +87,20 @@ pub(crate) fn validate_pycapsule(capsule: &Bound<PyCapsule>, name: &str) -> PyRe
 
     Ok(())
 }
+
+/// Convert a python object to a ScalarValue
+pub(crate) fn py_obj_to_scalar_value(py: Python, obj: PyObject) -> ScalarValue {
+    if let Ok(value) = obj.extract::<bool>(py) {
+        ScalarValue::Boolean(Some(value))
+    } else if let Ok(value) = obj.extract::<i64>(py) {
+        ScalarValue::Int64(Some(value))
+    } else if let Ok(value) = obj.extract::<u64>(py) {
+        ScalarValue::UInt64(Some(value))
+    } else if let Ok(value) = obj.extract::<f64>(py) {
+        ScalarValue::Float64(Some(value))
+    } else if let Ok(value) = obj.extract::<String>(py) {
+        ScalarValue::Utf8(Some(value))
+    } else {
+        panic!("Unsupported value type")
+    }
+}