Mercurial > python-compiler.rs
diff src/ast_convert.rs @ 77:7d1406181aae
Add ast.SetComp.
author | Bastien Orivel <eijebong@bananium.fr> |
---|---|
date | Mon, 13 Jun 2016 19:22:59 +0200 |
parents | efd42fc280e8 |
children | f1a845e4121b |
line wrap: on
line diff
--- a/src/ast_convert.rs +++ b/src/ast_convert.rs @@ -305,6 +305,7 @@ fn parse_expr(py: Python, ast: PyObject) let yield_type = ast_module.get(py, "Yield").unwrap(); let yield_from_type = ast_module.get(py, "YieldFrom").unwrap(); let set_type = ast_module.get(py, "Set").unwrap(); + let setcomp_type = ast_module.get(py, "SetComp").unwrap(); assert!(is_instance(&ast, &ast_type)); @@ -426,6 +427,14 @@ fn parse_expr(py: Python, ast: PyObject) let elements = parse_list(py, elts, parse_expr); expr::Set(elements) + } else if is_instance(&ast, &setcomp_type) { + let elt = ast.getattr(py, "elt").unwrap(); + let generators = ast.getattr(py, "generators").unwrap(); + + let elt = parse_expr(py, elt); + let generators = parse_list(py, generators, parse_comprehension); + + expr::SetComp(Box::new(elt), generators) } else { println!("expr {}", ast); unreachable!()