2

尝试使用 pyo3 和此代码构建模块层次结构

pub mod types;
pub mod sources;

use pyo3::prelude::*;
use pyo3::wrap_pymodule;

use sources::file::{find_days, read_many, read_one};


#[pymodule]
fn file(_py: Python, m: &PyModule) -> PyResult<()> {

    #[pyfn(m, "find_days")]
    fn find_days_py(_py: Python, dir: String) -> PyResult<Vec<String>> {
        let out = find_days(&dir)?;
        Ok(out.iter().map(|x| String::from(x.to_str().unwrap())).collect())
    }

    Ok(())
}

#[pymodule]
fn sources(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_wrapped(wrap_pymodule!(file))?;
    Ok(())
}

#[pymodule]
fn cstuff(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_wrapped(wrap_pymodule!(sources))?;
    // m.add("__path__", vec![""])?;
    Ok(())
}

代码构建和工作正常,除了当我尝试导入它时,我收到了这个错误

In [1]: import cstuff.sources.file                                                                      
---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-1-97430c5317d9> in <module>
----> 1 import cstuff.sources.file

ModuleNotFoundError: No module named 'cstuff.sources'; 'cstuff' is not a package

根据我从 python 文档中了解到的情况,如果模块具有__path__属性,它就是一个包。

我可以用m.add("__path__", vec![""])?;, 在这import按预期工作之后添加它,但仅当从带有.so文件的目录启动 python 时,这是预期的。

问题是我事先不知道__path__。如何解决这个问题,有没有办法强制 python 设置__path__

4

0 回答 0