1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

#![cfg(not(test))]
#![cfg(feature = "pyo3")]
use pyo3::prelude::*;

/// Formats the sum of two numbers as string.
#[pyfunction]
fn sum_as_string(a: usize, b: usize) -> PyResult<String> {
  Ok((a + b).to_string())
}

/// A Python module implemented in Rust.
#[pymodule]
fn ast_grep(_py: Python, m: &PyModule) -> PyResult<()> {
  m.add_function(wrap_pyfunction!(sum_as_string, m)?)?;
  Ok(())
}