Function pythonize

src/python.rs:40–72

Transation between rust extraction type and Python dictionary

(py: Python, extraction: Extraction)

// Transation between rust extraction type and Python dictionary
fn pythonize(py: Python, extraction: Extraction) -> PyResult<Py<PyAny>> {
    let refs = PyList::empty(py);

    for r in extraction.refs.iter() {
        let pyref = PyDict::new(py);
        pyref.set_item("name", &r.name)?;
        if r.package.is_some() {
            pyref.set_item("package", &r.package)?;
        }
        match &r.version {
            Some(StringRV(s)) => pyref.set_item("version", s),
            Some(IntRV(i)) => pyref.set_item("version", i),
            Some(DoubleRV(d)) => pyref.set_item("version", d),
            _ => PyResult::Ok(()),
        }?;

        refs.append(pyref)?;
    }

    let sources = PySet::new(py, &extraction.sources[..])?;
    let py_configs: Vec<(String, Py<PyAny>)> = extraction
        .configs
        .into_iter()
        .map(|(k, v)| (k, convert_config(py, v)))
        .collect();
    let configs = PyList::new(py, &py_configs[..])?;

    let dict = PyDict::new(py);
    dict.set_item("refs", refs)?;
    dict.set_item("sources", sources)?;
    dict.set_item("configs", configs)?;
    Ok(dict.into_pyobject(py).unwrap().into())
}

// Naively converts a Result to a PyResult by using the string
// representation of the Rust exception as the message for ExtractionError.