diff src/ast_rewrite.rs @ 5:ddf372373a77

Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
author Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date Tue, 31 May 2016 04:15:00 +0100
parents 211b0df72e64
children 6f2bf13f4cb5
line wrap: on
line diff
--- a/src/ast_rewrite.rs
+++ b/src/ast_rewrite.rs
@@ -40,6 +40,16 @@ impl Visitor<()> for Rewrite {
                     self.visit_statement(statement);
                 }
             },
+            Statement::For(target, iter, body, orelse) => {
+                self.visit_expr(target);
+                self.visit_expr(iter);
+                for statement in body {
+                    self.visit_statement(statement);
+                }
+                for statement in orelse {
+                    self.visit_statement(statement);
+                }
+            },
             Statement::Assign(targets, value) => {
                 self.visit_expr(value);
                 for target in targets {