extract sys path to fn

Author: KSDaemon

packages/cubejs-backend-native/src/python/entry.rs

@@ -8,6 +8,18 @@ use pyo3::prelude::*;
 use pyo3::types::{PyDict, PyFunction, PyList, PyString, PyTuple};
 use std::path::Path;
 
+fn extend_sys_path(py: Python, file_name: &String) -> PyResult<()> {
+    let sys_path = py.import("sys")?.getattr("path")?.downcast::<PyList>()?;
+
+    let config_dir = Path::new(&file_name)
+        .parent()
+        .unwrap_or_else(|| Path::new("."));
+    let config_dir_str = config_dir.to_str().unwrap_or(".");
+
+    sys_path.insert(0, PyString::new(py, config_dir_str))?;
+    Ok(())
+}
+
 fn python_load_config(mut cx: FunctionContext) -> JsResult<JsPromise> {
     let file_content_arg = cx.argument::<JsString>(0)?.value(&mut cx);
     let options_arg = cx.argument::<JsObject>(1)?;
@@ -21,14 +33,7 @@ fn python_load_config(mut cx: FunctionContext) -> JsResult<JsPromise> {
     py_runtime_init(&mut cx, channel.clone())?;
 
     let conf_res = Python::with_gil(|py| -> PyResult<CubeConfigPy> {
-        let sys_path = py.import("sys")?.getattr("path")?.downcast::<PyList>()?;
-
-        let config_dir = Path::new(&options_file_name)
-            .parent()
-            .unwrap_or_else(|| Path::new("."));
-        let config_dir_str = config_dir.to_str().unwrap_or(".");
-
-        sys_path.insert(0, PyString::new(py, config_dir_str))?;
+        extend_sys_path(py, &options_file_name)?;
 
         let cube_code = include_str!(concat!(
             env!("CARGO_MANIFEST_DIR"),
@@ -71,14 +76,7 @@ fn python_load_model(mut cx: FunctionContext) -> JsResult<JsPromise> {
     py_runtime_init(&mut cx, channel.clone())?;
 
     let conf_res = Python::with_gil(|py| -> PyResult<CubePythonModel> {
-        let sys_path = py.import("sys")?.getattr("path")?.downcast::<PyList>()?;
-
-        let config_dir = Path::new(&model_file_name)
-            .parent()
-            .unwrap_or_else(|| Path::new("."));
-        let config_dir_str = config_dir.to_str().unwrap_or(".");
-
-        sys_path.insert(0, PyString::new(py, config_dir_str))?;
+        extend_sys_path(py, &model_file_name)?;
 
         let cube_code = include_str!(concat!(
             env!("CARGO_MANIFEST_DIR"),