luxolus/pyo3
2
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

add pyfn to python2

Browse source
This commit is contained in:
Nikolay Kim 2017-06-14 14:14:34 -07:00
parent 706477f139
commit fdf971c54e
1 changed files with 18 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
pyo3cls/src/module.rs
View file

@ -88,7 +88,24 @@ pub fn build_py2_module_init(ast: &mut syn::Item, attr: String) -> Tokens {
let modname = &attr.to_string()[1..attr.to_string().len()-1].to_string(); let modname = &attr.to_string()[1..attr.to_string().len()-1].to_string();
match ast.node { match ast.node {
syn::ItemKind::Fn(_, _, _, _, _, _) => { syn::ItemKind::Fn(_, _, _, _, _, ref mut block) => {
let mut stmts = Vec::new();
for stmt in block.stmts.iter_mut() {
match stmt {
&mut syn::Stmt::Item(ref mut item) => {
if let Some(block) = wrap_fn(item) {
for stmt in block.stmts.iter() {
stmts.push(stmt.clone());
}
continue
}
}
_ => (),
}
stmts.push(stmt.clone());
}
block.stmts = stmts;
py2_init(&ast.ident, &modname) py2_init(&ast.ident, &modname)
}, },
_ => panic!("#[modinit] can only be used with fn block"), _ => panic!("#[modinit] can only be used with fn block"),
@ -278,8 +295,6 @@ fn wrap_fn(item: &mut syn::Item) -> Option<Box<syn::Block>> {
} }
}.to_string(); }.to_string();
println!("TEST {}", tokens);
let item = syn::parse_item(tokens.as_str()).unwrap(); let item = syn::parse_item(tokens.as_str()).unwrap();
match item.node { match item.node {
syn::ItemKind::Fn(_, _, _, _, _, ref block) => { syn::ItemKind::Fn(_, _, _, _, _, ref block) => {