Fix python path error using path searching for both python3 and `py…

…thon` Fix #394
dora-rs · Dec 20, 2023 · adcfeaf · adcfeaf
1 parent a7244f0
commit adcfeaf
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/binaries/daemon/Cargo.toml b/binaries/daemon/Cargo.toml
@@ -39,3 +39,4 @@ bincode = "1.3.3"
 async-trait = "0.1.64"
 arrow-schema = { workspace = true }
 aligned-vec = "0.5.0"
+which = "4.3.0"
diff --git a/binaries/daemon/src/spawn.rs b/binaries/daemon/src/spawn.rs
@@ -137,7 +137,12 @@ pub async fn spawn_node(
 
             let mut command = if has_python_operator && !has_other_operator {
                 // Use python to spawn runtime if there is a python operator
-                let mut command = tokio::process::Command::new("python3");
+                let python = match which::which("python3") {
+                    Ok(python) => python,
+                    Err(_) => which::which("python")
+                        .context("failed to find `python` or `python3` in dora-daemon path. Make sure that python is available for the daemon.")?,
+                };
+                let mut command = tokio::process::Command::new(python);
                 command.args([
                     "-c",
                     format!("import dora; dora.start_runtime() # {}", node.id).as_str(),