view rust/hg-pyo3/src/lib.rs @ 52853:c6707e112d96

rust-pyo3-dirstate: exceptions for dirstate errors
author Georges Racinet <georges.racinet@cloudcrane.io>
date Wed, 29 Jan 2025 14:03:24 +0100
parents d0c0ad938eb9
children c60f69556924
line wrap: on
line source

use pyo3::prelude::*;

mod ancestors;
mod dagops;
mod dirstate;
mod discovery;
mod exceptions;
mod node;
mod revision;
mod revlog;
mod store;
mod transaction;
mod utils;

#[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(&dirstate::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(())
}