mirror of https://github.com/astral-sh/ruff
16 lines
512 B
Rust
16 lines
512 B
Rust
pub use generator::Generator;
|
|
use ruff_python_parser::{ParseError, parse_module};
|
|
pub use stylist::{Indentation, 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())
|
|
}
|