Mercurial > python-compiler.rs
annotate src/ast_dump.rs @ 6:6f2bf13f4cb5
Add ast.While and ast.Break.
| author | Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> |
|---|---|
| date | Tue, 31 May 2016 04:22:35 +0100 |
| parents | ddf372373a77 |
| children | 680d15073f55 |
| rev | line source |
|---|---|
|
5
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
1 use python_ast::{Module, Statement, Expr, BinOp, UnaryOp}; |
| 0 | 2 |
| 3 use std::iter; | |
| 4 | |
|
5
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
5 impl UnaryOp { |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
6 fn to_string(&self) -> &'static str { |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
7 match *self { |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
8 UnaryOp::UAdd => "+", |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
9 UnaryOp::USub => "-", |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
10 UnaryOp::Error => "BinOp::Error" |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
11 } |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
12 } |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
13 } |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
14 |
| 0 | 15 impl BinOp { |
| 16 fn to_string(&self) -> &'static str { | |
| 17 match *self { | |
| 18 BinOp::BinAdd => "+", | |
| 19 BinOp::BinMult => "*", | |
| 20 BinOp::BinEq => "==", | |
| 21 BinOp::BinLt => "<", | |
|
6
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
22 BinOp::BinGt => ">", |
|
5
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
23 BinOp::Sub => "-", |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
24 BinOp::Div => "/", |
| 0 | 25 BinOp::Error => "BinOp::Error" |
| 26 } | |
| 27 } | |
| 28 } | |
| 29 | |
| 30 impl Expr { | |
| 31 fn to_string(&self) -> String { | |
| 32 match self.clone() { | |
|
5
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
33 Expr::UnaryOp(op, operand) => format!("{}{}", op.to_string(), operand.to_string()), |
| 0 | 34 Expr::BinOp(a, op, b) => format!("{} {} {}", a.to_string(), op.to_string(), b.to_string()), |
| 35 Expr::Compare(left, ops, comparators) => format!("{} {}", left.to_string(), { | |
| 36 let mut arguments = vec!(); | |
| 37 | |
| 38 // XXX: wrong order! | |
| 39 for op in ops { | |
| 40 arguments.push(op.to_string().to_string()) | |
| 41 } | |
| 42 for comparator in comparators { | |
| 43 arguments.push(comparator.to_string()) | |
| 44 } | |
| 45 /* | |
| 46 for (op, comparator) in ops.zip(comparators) { | |
| 47 let op = op.unwrap(); | |
| 48 let comparator = comparator.unwrap(); | |
| 49 arguments.push(format!("{} {}", op.to_string(), comparator.to_string())) | |
| 50 } | |
| 51 */ | |
| 52 | |
| 53 arguments.join(" ") | |
| 54 }), | |
| 55 Expr::Call(func, args) => format!("{}({})", func.to_string(), { | |
| 56 let mut arguments = vec!(); | |
| 57 for arg in args { | |
| 58 arguments.push(arg.to_string()); | |
| 59 } | |
| 60 arguments.join(", ") | |
| 61 }), | |
| 62 Expr::Alias(name, asname) => { | |
|
1
b90e49ab734b
Factorise conversion of Python str into Rust String.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
0
diff
changeset
|
63 match asname { |
|
b90e49ab734b
Factorise conversion of Python str into Rust String.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
0
diff
changeset
|
64 None => format!("{}", name), |
|
b90e49ab734b
Factorise conversion of Python str into Rust String.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
0
diff
changeset
|
65 Some(asname) => format!("{} as {}", name, asname) |
| 0 | 66 } |
| 67 } | |
|
4
f27a4aee9dfa
Add ast.Attribute.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
3
diff
changeset
|
68 Expr::Attribute(lhs, rhs) => format!("{}.{}", lhs.to_string(), rhs), |
| 0 | 69 Expr::Name(name) => format!("{}", name), |
| 70 Expr::NameConstant(name) => format!("{}", name), | |
| 71 Expr::Str(s) => format!("\"{}\"", s), | |
| 72 Expr::Num(n) => format!("{}", n), | |
| 73 Expr::Error => "Expr::Error".to_string() | |
| 74 } | |
| 75 } | |
| 76 } | |
| 77 | |
| 78 fn make_indent(indent: usize) -> String { | |
| 79 let indent: String = iter::repeat(" ").take(indent).collect(); | |
| 80 indent | |
| 81 } | |
| 82 | |
| 83 impl Statement { | |
| 84 fn to_string(&self, indent: usize) -> String { | |
| 85 match self.clone() { | |
|
2
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
86 Statement::ClassDef(name, classes, body) => format!("{}def {}({}):\n{}", make_indent(indent), name, { |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
87 let mut bases = vec!(); |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
88 for class in classes { |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
89 bases.push(class.to_string()); |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
90 } |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
91 bases.join(", ") |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
92 }, { |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
93 let mut statements = vec!(); |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
94 for statement in body { |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
95 statements.push(statement.to_string(indent + 1)); |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
96 } |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
97 statements.join("\n") |
|
5fc7c2790d8c
Add class support.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
1
diff
changeset
|
98 }), |
|
3
326d7f2a94d4
Remove useless abstraction of function name as Expr.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
2
diff
changeset
|
99 Statement::FunctionDef(name, arguments, body) => format!("{}def {}({}):\n{}", make_indent(indent), name, { |
| 0 | 100 let mut args = vec!(); |
| 101 for arg in arguments { | |
| 102 args.push(arg.to_string()); | |
| 103 } | |
| 104 args.join(", ") | |
| 105 }, { | |
| 106 let mut statements = vec!(); | |
| 107 for statement in body { | |
| 108 statements.push(statement.to_string(indent + 1)); | |
| 109 } | |
| 110 statements.join("\n") | |
| 111 }), | |
| 112 Statement::Global(names) => format!("{}global {}", make_indent(indent), names.join(", ")), | |
| 113 Statement::If(test, body, orelse) => format!("{}if {}:\n{}", make_indent(indent), test.to_string(), { | |
| 114 let mut statements = vec!(); | |
| 115 for arg in body { | |
| 116 statements.push(arg.to_string(indent + 1)); | |
| 117 } | |
| 118 | |
| 119 let mut orelse_ = vec!(); | |
| 120 for arg in orelse { | |
| 121 orelse_.push(arg.to_string(indent + 1)); | |
| 122 } | |
| 123 | |
| 124 if orelse_.is_empty() { | |
| 125 statements.join("\n") | |
| 126 } else { | |
| 127 format!("{}\n{}else:\n{}", statements.join("\n"), make_indent(indent), orelse_.join("\n")) | |
| 128 } | |
| 129 }), | |
|
6
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
130 Statement::While(test, body, orelse) => format!("{}while {}:\n{}", make_indent(indent), test.to_string(), { |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
131 let mut statements = vec!(); |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
132 for arg in body { |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
133 statements.push(arg.to_string(indent + 1)); |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
134 } |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
135 |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
136 let mut orelse_ = vec!(); |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
137 for arg in orelse { |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
138 orelse_.push(arg.to_string(indent + 1)); |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
139 } |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
140 |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
141 if orelse_.is_empty() { |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
142 statements.join("\n") |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
143 } else { |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
144 format!("{}\n{}else:\n{}", statements.join("\n"), make_indent(indent), orelse_.join("\n")) |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
145 } |
|
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
146 }), |
|
5
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
147 Statement::For(target, iter, body, orelse) => format!("{}for {} in {}:\n{}", make_indent(indent), target.to_string(), iter.to_string(), { |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
148 let mut statements = vec!(); |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
149 for arg in body { |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
150 statements.push(arg.to_string(indent + 1)); |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
151 } |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
152 |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
153 let mut orelse_ = vec!(); |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
154 for arg in orelse { |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
155 orelse_.push(arg.to_string(indent + 1)); |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
156 } |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
157 |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
158 if orelse_.is_empty() { |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
159 statements.join("\n") |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
160 } else { |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
161 format!("{}\n{}else:\n{}", statements.join("\n"), make_indent(indent), orelse_.join("\n")) |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
162 } |
|
ddf372373a77
Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
4
diff
changeset
|
163 }), |
| 0 | 164 Statement::Assign(targets, value) => format!("{}{} = {}", make_indent(indent), { |
| 165 let mut exprs = vec!(); | |
| 166 for target in targets { | |
| 167 exprs.push(target.to_string()); | |
| 168 } | |
| 169 exprs.join(", ") | |
| 170 }, value.to_string()), | |
| 171 Statement::Return(expr) => format!("{}return {}", make_indent(indent), expr.to_string()), | |
| 172 Statement::ImportFrom(module, names) => format!("{}from {} import {}", make_indent(indent), module.to_string(), { | |
| 173 let mut exprs = vec!(); | |
| 174 for alias in names.iter() { | |
| 175 let alias = alias.to_string(); | |
| 176 exprs.push(alias); | |
| 177 } | |
| 178 exprs.join(", ") | |
| 179 }), | |
| 180 Statement::Expr(expr) => format!("{}{}", make_indent(indent), expr.to_string()), | |
|
6
6f2bf13f4cb5
Add ast.While and ast.Break.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
5
diff
changeset
|
181 Statement::Break => format!("{}break", make_indent(indent)), |
| 0 | 182 Statement::Error => format!("{}Statement::Error", make_indent(indent)) |
| 183 } | |
| 184 } | |
| 185 } | |
| 186 | |
| 187 #[allow(dead_code)] | |
| 188 pub fn dump_ast(ast: &Module) { | |
| 189 for statement in &ast.statements { | |
| 190 println!("{}", statement.to_string(0)); | |
| 191 } | |
| 192 } |