ruff/crates/ruff_python_codegen/src/lib.rs
2025-05-16 13:25:28 +02:00

15 lines
497 B
Rust

pub use generator::Generator;
use ruff_python_parser::{ParseError, parse_module};
pub use stylist::Stylist;
mod generator;
mod stylist;
/// Run round-trip source code generation on a given Python code.
pub fn round_trip(code: &str) -> Result<String, ParseError> {
let parsed = parse_module(code)?;
let stylist = Stylist::from_tokens(parsed.tokens(), code);
let mut generator: Generator = (&stylist).into();
generator.unparse_suite(parsed.suite());
Ok(generator.generate())
}