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