Cyclic0007/ruff
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Extract LineIndex independent methods from Locator (#13938)

Browse Source
This commit is contained in:
Micha Reiser
2024-10-28 08:53:41 +01:00
committed by GitHub
parent f8eb547fb4
commit 9f3a38d408
171 changed files with 1348 additions and 1284 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
crates/ruff_python_codegen/src/lib.rs
View File

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