1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use std::path::Path;
use std::result::Result;
use ymlctx::context::{Context, CtxObj};
use crate::{TaskError, TaskErrorSource};
use std::io::Write; // for flushing stdio

#[cfg(feature = "lang_python")]
use pyo3::prelude::*;
#[cfg(feature = "lang_python")]
use pyo3::types::PyList;

/// Flush stdout & stderr
#[allow(unused_must_use)]
#[cfg(feature = "lang_python")]
fn flush_stdio(py: pyo3::Python) {
    py.run("sys.stderr.flush(); sys.stdout.flush()", None, None);
    std::io::stderr().flush();
    std::io::stdout().flush();
}

#[cfg(feature = "lang_python")]
pub fn invoke(src: Context, ctx_step: Context) -> Result<(), TaskError> {
    let gil = Python::acquire_gil();
    let py = gil.python();
    let syspath: &PyList = py.import("sys").unwrap().get("path").unwrap().try_into().unwrap();
    let ref src_path: String = src.unpack("src").unwrap();
    if let Some(CtxObj::Array(sys_paths)) = src.get("sys_path") {
        for item in sys_paths {
            if let CtxObj::Str(sys_path) = item {
                syspath.insert(0, sys_path).unwrap();
            }
        }
    }
    let mod_path;
    if let Some(parent) = Path::new(src_path).parent() {
        mod_path = parent;
    }
    else {
        mod_path = Path::new(".");
    }
    syspath.insert(0, mod_path.to_str().unwrap()).unwrap();

    let mod_name;
    if let Some(stem) = Path::new(src_path).file_stem() {
        mod_name = stem.to_str().unwrap();
    }
    else {
        unreachable!();
    }

    if let Ok(mod_py) = py.import(mod_name) {
        let ref action: String = ctx_step.unpack("action").unwrap();
        match mod_py.call_method1(action, (ctx_step.to_object(py), )) {
            Ok(_) => {
                flush_stdio(py);
                Ok(())
            },
            Err(e) => {
                flush_stdio(py);
                e.print_and_set_sys_last_vars(py);
                Err(TaskError { msg: String::from("There was an exception raised by the step action."), src: TaskErrorSource::Internal })
            }
        }
    }
    else {
        Err(TaskError { msg: String::from("Failed to import the step action source module."), src: TaskErrorSource::Internal })
    }
}