rust/hg-pyo3/src/lib.rs
author Rapha?l Gom?s <rgomes@octobus.net>
Mon, 06 Jan 2025 01:14:52 +0100
changeset 52837 01aff9437828
parent 52827 50c0c74ca266
child 52839 b8cd277d26f4
permissions -rw-r--r--
rust-pyo3: translate discovery module from hg-cpython This is a small module and the only one left that needs to know about the index. Building the conversion machinery would have taken longer and have been more dangerous.

use pyo3::prelude::*;

mod ancestors;
mod convert_cpython;
mod dagops;
mod discovery;
mod exceptions;
mod node;
mod revision;
mod revlog;
mod transaction;
mod store;
mod util;

#[pymodule]
fn pyo3_rustext(py: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> {
    m.add("__package__", "mercurial")?;
    m.add(
        "__doc__",
        "Mercurial core concepts - Rust implementation exposed via PyO3",
    )?;
    let dotted_name: String = m.getattr("__name__")?.extract()?;
    env_logger::init();

    m.add_submodule(&ancestors::init_module(py, &dotted_name)?)?;
    m.add_submodule(&dagops::init_module(py, &dotted_name)?)?;
    m.add_submodule(&discovery::init_module(py, &dotted_name)?)?;
    m.add_submodule(&revlog::init_module(py, &dotted_name)?)?;
    m.add("GraphError", py.get_type::<exceptions::GraphError>())?;
    Ok(())
}