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!()