mirrors/RustPython-Parser
1
0
Fork 0
mirror of https://github.com/RustPython/Parser.git synced 2025-08-29 23:04:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Linear Locator (#46)

Browse source
This commit is contained in:
Jeong, YunWon 2023-06-01 13:53:31 +09:00 • committed by GitHub
parent fdec727f80
commit 5e9e8a7589
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 1080 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ast-pyo3/src/lib.rs
View file

@ -4,7 +4,7 @@ pub mod wrapper;
pub use py_ast::{init, PyNode, ToPyAst};
use pyo3::prelude::*;
use rustpython_parser::ast::{source_code::SourceLocator, Fold};
use rustpython_parser::ast::{source_code::LinearLocator, Fold};
#[pyfunction]
#[pyo3(signature = (source, filename="<unknown>", *, type_comments=false, locate=true))]
@ -21,7 +21,7 @@ pub fn parse<'py>(
let parsed = rustpython_parser::parse(source, rustpython_parser::Mode::Module, filename)
.map_err(|e| PyErr::new::<pyo3::exceptions::PySyntaxError, _>(e.to_string()))?;
if locate {
let parsed = SourceLocator::new(source).fold(parsed).unwrap();
let parsed = LinearLocator::new(source).fold(parsed).unwrap();
parsed.module().unwrap().to_py_ast(py)
} else {
parsed.module().unwrap().to_py_ast(py)