python-compiler.rs: src/ast_convert.rs annotate

annotate src/ast_convert.rs @ 13:38b0d63697b1

Import the full AST grammar from CPython 3.5.

author	Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date	Thu, 02 Jun 2016 05:33:23 +0100
parents	fa7e285f88e7
children	719a27f1c1c7

rev	line source
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	1 use python_ast::{Module, stmt, expr, expr_context, cmpop, operator, unaryop, arguments, arg, alias};
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	2
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	3 use cpython::{Python, PyObject};
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	4 use cpython::ObjectProtocol; //for call method
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	5
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	6 fn get_str(py: Python, object: PyObject) -> String {
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	7 let pystring = object.str(py).unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	8 let mut string = pystring.to_string(py).unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	9 string.to_mut().to_string()
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	10 }
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	11
9 fa7e285f88e7 Add a scoping pass, associating each module/statement with a block. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 8 diff changeset	12 fn parse_expr_vec(py: Python, ast: PyObject) -> Vec<String> {
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	13 let builtins_module = py.import("builtins").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	14 let isinstance = builtins_module.get(py, "isinstance").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	15
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	16 let is_instance = \|object: &PyObject, type_: &PyObject\| {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	17 return isinstance.call(py, (object, type_), None).unwrap().is_true(py).unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	18 };
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	19
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	20 let ast_module = py.import("ast").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	21 let ast_type = ast_module.get(py, "AST").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	22 let arguments_type = ast_module.get(py, "arguments").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	23
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	24 assert!(is_instance(&ast, &ast_type));
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	25
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	26 if is_instance(&ast, &arguments_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	27 let args = ast.getattr(py, "args").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	28 let mut arguments = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	29 for arg in args.iter(py).unwrap() {
9 fa7e285f88e7 Add a scoping pass, associating each module/statement with a block. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 8 diff changeset	30 let arg = arg.unwrap();
fa7e285f88e7 Add a scoping pass, associating each module/statement with a block. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 8 diff changeset	31 let arg = parse_expr(py, arg);
fa7e285f88e7 Add a scoping pass, associating each module/statement with a block. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 8 diff changeset	32 arguments.push(match arg {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	33 expr::Name(arg, expr_context::Load) => arg,
9 fa7e285f88e7 Add a scoping pass, associating each module/statement with a block. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 8 diff changeset	34 _ => panic!()
fa7e285f88e7 Add a scoping pass, associating each module/statement with a block. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 8 diff changeset	35 });
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	36 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	37 arguments
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	38 } else {
9 fa7e285f88e7 Add a scoping pass, associating each module/statement with a block. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 8 diff changeset	39 panic!()
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	40 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	41 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	42
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	43 fn parse_unaryop(py: Python, ast: PyObject) -> unaryop {
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	44 let builtins_module = py.import("builtins").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	45 let isinstance = builtins_module.get(py, "isinstance").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	46
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	47 let is_instance = \|object: &PyObject, type_: &PyObject\| {
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	48 return isinstance.call(py, (object, type_), None).unwrap().is_true(py).unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	49 };
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	50
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	51 let ast_module = py.import("ast").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	52 let ast_type = ast_module.get(py, "AST").unwrap();
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	53 let invert_type = ast_module.get(py, "Invert").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	54 let not_type = ast_module.get(py, "Not").unwrap();
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	55 let uadd_type = ast_module.get(py, "UAdd").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	56 let usub_type = ast_module.get(py, "USub").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	57
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	58 assert!(is_instance(&ast, &ast_type));
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	59
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	60 if is_instance(&ast, &invert_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	61 unaryop::Invert
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	62 } else if is_instance(&ast, &not_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	63 unaryop::Not
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	64 } else if is_instance(&ast, &uadd_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	65 unaryop::UAdd
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	66 } else if is_instance(&ast, &usub_type) {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	67 unaryop::USub
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	68 } else {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	69 unreachable!()
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	70 }
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	71 }
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	72
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	73 fn parse_cmpop(py: Python, ast: PyObject) -> cmpop {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	74 let builtins_module = py.import("builtins").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	75 let isinstance = builtins_module.get(py, "isinstance").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	76
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	77 let is_instance = \|object: &PyObject, type_: &PyObject\| {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	78 return isinstance.call(py, (object, type_), None).unwrap().is_true(py).unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	79 };
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	80
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	81 let ast_module = py.import("ast").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	82 let ast_type = ast_module.get(py, "AST").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	83 let eq_type = ast_module.get(py, "Eq").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	84 let noteq_type = ast_module.get(py, "NotEq").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	85 let lt_type = ast_module.get(py, "Lt").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	86 let gt_type = ast_module.get(py, "Gt").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	87
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	88 assert!(is_instance(&ast, &ast_type));
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	89
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	90 if is_instance(&ast, &eq_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	91 cmpop::Eq
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	92 } else if is_instance(&ast, &noteq_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	93 cmpop::NotEq
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	94 } else if is_instance(&ast, &lt_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	95 cmpop::Lt
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	96 } else if is_instance(&ast, &gt_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	97 cmpop::Gt
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	98 } else {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	99 unreachable!()
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	100 }
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	101 }
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	102
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	103 fn parse_operator(py: Python, ast: PyObject) -> operator {
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	104 let builtins_module = py.import("builtins").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	105 let isinstance = builtins_module.get(py, "isinstance").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	106
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	107 let is_instance = \|object: &PyObject, type_: &PyObject\| {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	108 return isinstance.call(py, (object, type_), None).unwrap().is_true(py).unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	109 };
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	110
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	111 let ast_module = py.import("ast").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	112 let ast_type = ast_module.get(py, "AST").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	113 let add_type = ast_module.get(py, "Add").unwrap();
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	114 let sub_type = ast_module.get(py, "Sub").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	115 let mult_type = ast_module.get(py, "Mult").unwrap();
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	116 let matmult_type = ast_module.get(py, "MatMult").unwrap();
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	117 let div_type = ast_module.get(py, "Div").unwrap();
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	118 let mod_type = ast_module.get(py, "Mod").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	119 let pow_type = ast_module.get(py, "Pow").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	120 let lshift_type = ast_module.get(py, "LShift").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	121 let rshift_type = ast_module.get(py, "RShift").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	122 let bitor_type = ast_module.get(py, "BitOr").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	123 let bitxor_type = ast_module.get(py, "BitXor").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	124 let bitand_type = ast_module.get(py, "BitAnd").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	125 let floordiv_type = ast_module.get(py, "FloorDiv").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	126
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	127 assert!(is_instance(&ast, &ast_type));
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	128
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	129 if is_instance(&ast, &add_type) {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	130 operator::Add
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	131 } else if is_instance(&ast, &sub_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	132 operator::Sub
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	133 } else if is_instance(&ast, &mult_type) {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	134 operator::Mult
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	135 } else if is_instance(&ast, &matmult_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	136 operator::MatMult
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	137 } else if is_instance(&ast, &div_type) {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	138 operator::Div
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	139 } else if is_instance(&ast, &mod_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	140 operator::Mod
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	141 } else if is_instance(&ast, &pow_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	142 operator::Pow
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	143 } else if is_instance(&ast, &lshift_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	144 operator::LShift
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	145 } else if is_instance(&ast, &rshift_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	146 operator::RShift
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	147 } else if is_instance(&ast, &bitor_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	148 operator::BitOr
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	149 } else if is_instance(&ast, &bitxor_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	150 operator::BitXor
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	151 } else if is_instance(&ast, &bitand_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	152 operator::BitAnd
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	153 } else if is_instance(&ast, &floordiv_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	154 operator::FloorDiv
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	155 } else {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	156 println!("operator {}", ast);
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	157 panic!()
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	158 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	159 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	160
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	161 fn parse_expr(py: Python, ast: PyObject) -> expr {
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	162 let builtins_module = py.import("builtins").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	163 let isinstance = builtins_module.get(py, "isinstance").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	164
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	165 let is_instance = \|object: &PyObject, type_: &PyObject\| {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	166 return isinstance.call(py, (object, type_), None).unwrap().is_true(py).unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	167 };
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	168
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	169 let ast_module = py.import("ast").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	170 let ast_type = ast_module.get(py, "AST").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	171 let arg_type = ast_module.get(py, "arg").unwrap();
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	172 let unary_op_type = ast_module.get(py, "UnaryOp").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	173 let bin_op_type = ast_module.get(py, "BinOp").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	174 let name_constant_type = ast_module.get(py, "NameConstant").unwrap();
4 f27a4aee9dfa Add ast.Attribute. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 3 diff changeset	175 let attribute_type = ast_module.get(py, "Attribute").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	176 let name_type = ast_module.get(py, "Name").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	177 let num_type = ast_module.get(py, "Num").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	178 let str_type = ast_module.get(py, "Str").unwrap();
7 680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	179 let list_type = ast_module.get(py, "List").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	180 let compare_type = ast_module.get(py, "Compare").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	181 let call_type = ast_module.get(py, "Call").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	182
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	183 assert!(is_instance(&ast, &ast_type));
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	184
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	185 if is_instance(&ast, &arg_type) {
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	186 let arg = ast.getattr(py, "arg").unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	187 let arg = get_str(py, arg);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	188 expr::Name(arg, expr_context::Load)
4 f27a4aee9dfa Add ast.Attribute. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 3 diff changeset	189 } else if is_instance(&ast, &attribute_type) {
f27a4aee9dfa Add ast.Attribute. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 3 diff changeset	190 let value = ast.getattr(py, "value").unwrap();
f27a4aee9dfa Add ast.Attribute. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 3 diff changeset	191 let attr = ast.getattr(py, "attr").unwrap();
f27a4aee9dfa Add ast.Attribute. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 3 diff changeset	192
f27a4aee9dfa Add ast.Attribute. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 3 diff changeset	193 let value = parse_expr(py, value);
f27a4aee9dfa Add ast.Attribute. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 3 diff changeset	194 let attr = get_str(py, attr);
f27a4aee9dfa Add ast.Attribute. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 3 diff changeset	195
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	196 expr::Attribute(Box::new(value), attr, expr_context::Load)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	197 } else if is_instance(&ast, &name_type) {
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	198 let id = ast.getattr(py, "id").unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	199 let id = get_str(py, id);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	200 expr::Name(id, expr_context::Load)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	201 } else if is_instance(&ast, &name_constant_type) {
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	202 let value = ast.getattr(py, "value").unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	203 let value = get_str(py, value);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	204 expr::NameConstant(value)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	205 } else if is_instance(&ast, &num_type) {
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	206 let n = ast.getattr(py, "n").unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	207 let n = get_str(py, n);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	208 expr::Num(n)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	209 } else if is_instance(&ast, &str_type) {
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	210 let s = ast.getattr(py, "s").unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	211 let s = get_str(py, s);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	212 expr::Str(s)
7 680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	213 } else if is_instance(&ast, &list_type) {
680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	214 let elts = ast.getattr(py, "elts").unwrap();
680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	215
680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	216 let mut elements = vec!();
680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	217 for elt in elts.iter(py).unwrap() {
680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	218 let elt = elt.unwrap();
680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	219 elements.push(parse_expr(py, elt));
680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	220 }
680d15073f55 Add ast.List literal. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 6 diff changeset	221
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	222 expr::List(elements, expr_context::Load)
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	223 } else if is_instance(&ast, &unary_op_type) {
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	224 let op = ast.getattr(py, "op").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	225 let operand = ast.getattr(py, "operand").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	226
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	227 let op = parse_unaryop(py, op);
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	228 let operand = parse_expr(py, operand);
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	229
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	230 expr::UnaryOp(op, Box::new(operand))
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	231 } else if is_instance(&ast, &bin_op_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	232 let left = ast.getattr(py, "left").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	233 let op = ast.getattr(py, "op").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	234 let right = ast.getattr(py, "right").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	235
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	236 let left = parse_expr(py, left);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	237 let op = parse_operator(py, op);
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	238 let right = parse_expr(py, right);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	239
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	240 expr::BinOp(Box::new(left), op, Box::new(right))
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	241 } else if is_instance(&ast, &call_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	242 let func = ast.getattr(py, "func").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	243 let args = ast.getattr(py, "args").unwrap();
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	244 let keywords = ast.getattr(py, "keywords").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	245
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	246 let func = parse_expr(py, func);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	247
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	248 let mut arguments = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	249 for arg in args.iter(py).unwrap() {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	250 let arg = arg.unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	251 arguments.push(parse_expr(py, arg));
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	252 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	253
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	254 /*
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	255 let mut kwargs = vec!();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	256 for arg in keywords.iter(py).unwrap() {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	257 let arg = arg.unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	258 kwargs.push(parse_expr(py, arg));
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	259 }
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	260 */
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	261
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	262 expr::Call(Box::new(func), arguments, vec!())
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	263 } else if is_instance(&ast, &compare_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	264 let left = ast.getattr(py, "left").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	265 let ops = ast.getattr(py, "ops").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	266 let comparators = ast.getattr(py, "comparators").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	267
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	268 let left = parse_expr(py, left);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	269 let ops = ops.iter(py).unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	270 let comparators = comparators.iter(py).unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	271
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	272 let mut new_ops = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	273 for op in ops {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	274 let op = op.unwrap();
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	275 let op = parse_cmpop(py, op);
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	276 new_ops.push(op);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	277 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	278
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	279 let mut new_comparators = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	280 for comparator in comparators {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	281 let comparator = comparator.unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	282 let comparator = parse_expr(py, comparator);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	283 new_comparators.push(comparator);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	284 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	285
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	286 expr::Compare(Box::new(left), new_ops, new_comparators)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	287 } else {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	288 println!("expr {}", ast);
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	289 unreachable!()
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	290 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	291 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	292
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	293 fn parse_arguments(py: Python, ast: PyObject) -> arguments {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	294 let builtins_module = py.import("builtins").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	295 let isinstance = builtins_module.get(py, "isinstance").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	296
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	297 let is_instance = \|object: &PyObject, type_: &PyObject\| {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	298 return isinstance.call(py, (object, type_), None).unwrap().is_true(py).unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	299 };
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	300
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	301 let ast_module = py.import("ast").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	302 let ast_type = ast_module.get(py, "AST").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	303 let arguments_type = ast_module.get(py, "arguments").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	304 let arg_type = ast_module.get(py, "arg").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	305
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	306 assert!(is_instance(&ast, &ast_type));
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	307
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	308 if is_instance(&ast, &arguments_type) {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	309 let args = arguments{
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	310 //args: Vec<arg>,
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	311 args: {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	312 let args = ast.getattr(py, "args").unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	313 let mut arguments = vec!();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	314 for arg in args.iter(py).unwrap() {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	315 let arg = arg.unwrap();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	316 assert!(is_instance(&arg, &arg_type));
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	317 let arg = get_str(py, arg);
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	318 arguments.push(arg{arg: arg, annotation: None});
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	319 }
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	320 arguments
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	321 },
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	322 //vararg: Option<arg>,
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	323 vararg: None,
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	324 //kwonlyargs: Vec<arg>,
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	325 kwonlyargs: vec!(),
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	326 //kw_defaults: Vec<expr>,
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	327 kw_defaults: vec!(),
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	328 //kwarg: Option<arg>,
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	329 kwarg: None,
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	330 //defaults: Vec<expr>
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	331 defaults: vec!()
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	332 };
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	333 args
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	334 } else {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	335 println!("arguments {}", ast);
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	336 panic!()
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	337 }
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	338 }
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	339
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	340 fn parse_statement(py: Python, ast: PyObject) -> stmt {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	341 //stmt::FunctionDef(expr::Name("function".to_string()), vec!(expr::Name("a".to_string()), expr::Name("b".to_string())), vec!())
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	342 //stmt::If(expr::BinOp(BinOp::BinEq, Box::new(expr::Name("__name__".to_string())), Box::new(expr::Str("__main__".to_string()))), vec!(stmt::Expr(expr::Call(Box::new(expr::Name("function".to_string())), vec!(expr::Num(1), expr::Num(2))))))
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	343
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	344 let builtins_module = py.import("builtins").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	345 let isinstance = builtins_module.get(py, "isinstance").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	346
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	347 let is_instance = \|object: &PyObject, type_: &PyObject\| {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	348 return isinstance.call(py, (object, type_), None).unwrap().is_true(py).unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	349 };
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	350
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	351 let ast_module = py.import("ast").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	352 let ast_type = ast_module.get(py, "AST").unwrap();
2 5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	353 let class_def_type = ast_module.get(py, "ClassDef").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	354 let function_def_type = ast_module.get(py, "FunctionDef").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	355 let global_type = ast_module.get(py, "Global").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	356 let assign_type = ast_module.get(py, "Assign").unwrap();
8 94ff501bf336 Add ast.AugAssign. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 7 diff changeset	357 let aug_assign_type = ast_module.get(py, "AugAssign").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	358 let return_type = ast_module.get(py, "Return").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	359 let import_from_type = ast_module.get(py, "ImportFrom").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	360 let if_type = ast_module.get(py, "If").unwrap();
6 6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	361 let while_type = ast_module.get(py, "While").unwrap();
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	362 let for_type = ast_module.get(py, "For").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	363 let expr_type = ast_module.get(py, "Expr").unwrap();
6 6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	364 let break_type = ast_module.get(py, "Break").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	365
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	366 assert!(is_instance(&ast, &ast_type));
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	367
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	368 /*
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	369 // TODO: implement Hash for PyObject. (trivial)
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	370 let map = {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	371 let fields = ast.getattr(py, "_fields").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	372 let mut map = HashMap::new();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	373 for field in fields.iter(py).unwrap() {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	374 let field = field.unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	375 let value = ast.getattr(py, field).unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	376 map.insert(field, value);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	377 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	378 map
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	379 };
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	380 */
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	381
2 5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	382 if is_instance(&ast, &class_def_type) {
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	383 let name = ast.getattr(py, "name").unwrap();
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	384 let bases = ast.getattr(py, "bases").unwrap();
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	385 //let keywords = ast.getattr(py, "keywords").unwrap();
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	386 let body = ast.getattr(py, "body").unwrap();
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	387 //let decorator_list = ast.getattr(py, "decorator_list").unwrap();
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	388
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	389 let name = get_str(py, name);
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	390
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	391 let mut nodes = vec!();
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	392 for name_node in bases.iter(py).unwrap() {
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	393 let name_node = name_node.unwrap();
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	394 let name_node = parse_expr(py, name_node);
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	395 nodes.push(name_node);
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	396 }
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	397
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	398 let mut statements = vec!();
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	399 for statement in body.iter(py).unwrap() {
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	400 let statement = statement.unwrap();
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	401 let statement = parse_statement(py, statement);
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	402 statements.push(statement);
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	403 }
5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	404
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	405 stmt::ClassDef(name, nodes, vec!(), statements, vec!())
2 5fc7c2790d8c Add class support. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 1 diff changeset	406 } else if is_instance(&ast, &function_def_type) {
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	407 let name = ast.getattr(py, "name").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	408 let args = ast.getattr(py, "args").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	409 let body = ast.getattr(py, "body").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	410
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	411 let name = get_str(py, name);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	412 let args = parse_arguments(py, args);
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	413
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	414 let mut statements = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	415 for statement in body.iter(py).unwrap() {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	416 let statement = parse_statement(py, statement.unwrap());
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	417 statements.push(statement);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	418 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	419
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	420 let decorators = vec!();
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	421 let returns = None;
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	422
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	423 stmt::FunctionDef(name, args, statements, decorators, returns)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	424 } else if is_instance(&ast, &global_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	425 let names = ast.getattr(py, "names").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	426
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	427 let mut globals = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	428 for name in names.iter(py).unwrap() {
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	429 let name = name.unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	430 let name = get_str(py, name);
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	431 globals.push(name);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	432 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	433
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	434 stmt::Global(globals)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	435 } else if is_instance(&ast, &if_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	436 let test = ast.getattr(py, "test").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	437 let body = ast.getattr(py, "body").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	438 let orelse = ast.getattr(py, "orelse").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	439
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	440 let test = parse_expr(py, test);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	441
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	442 let mut statements = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	443 for statement in body.iter(py).unwrap() {
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	444 let statement = statement.unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	445 let statement = parse_statement(py, statement);
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	446 statements.push(statement);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	447 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	448
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	449 let mut orelse_ = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	450 for statement in orelse.iter(py).unwrap() {
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	451 let statement = statement.unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	452 let statement = parse_statement(py, statement);
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	453 orelse_.push(statement);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	454 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	455
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	456 stmt::If(test, statements, orelse_)
6 6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	457 } else if is_instance(&ast, &while_type) {
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	458 let test = ast.getattr(py, "test").unwrap();
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	459 let body = ast.getattr(py, "body").unwrap();
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	460 let orelse = ast.getattr(py, "orelse").unwrap();
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	461
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	462 let test = parse_expr(py, test);
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	463
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	464 let mut statements = vec!();
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	465 for statement in body.iter(py).unwrap() {
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	466 let statement = statement.unwrap();
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	467 let statement = parse_statement(py, statement);
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	468 statements.push(statement);
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	469 }
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	470
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	471 let mut orelse_ = vec!();
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	472 for statement in orelse.iter(py).unwrap() {
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	473 let statement = statement.unwrap();
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	474 let statement = parse_statement(py, statement);
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	475 orelse_.push(statement);
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	476 }
6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	477
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	478 stmt::While(test, statements, orelse_)
5 ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	479 } else if is_instance(&ast, &for_type) {
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	480 let target = ast.getattr(py, "target").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	481 let iter = ast.getattr(py, "iter").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	482 let body = ast.getattr(py, "body").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	483 let orelse = ast.getattr(py, "orelse").unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	484
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	485 let target = parse_expr(py, target);
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	486 let iter = parse_expr(py, iter);
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	487
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	488 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	489 for statement in body.iter(py).unwrap() {
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	490 let statement = statement.unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	491 let statement = parse_statement(py, statement);
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	492 statements.push(statement);
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	493 }
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	494
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	495 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	496 for statement in orelse.iter(py).unwrap() {
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	497 let statement = statement.unwrap();
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	498 let statement = parse_statement(py, statement);
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	499 orelse_.push(statement);
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	500 }
ddf372373a77 Add ast.For, ast.UnaryOp, and Sub and Div to ast.BinOp. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 4 diff changeset	501
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	502 stmt::For(target, iter, statements, orelse_)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	503 } else if is_instance(&ast, &assign_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	504 let targets = ast.getattr(py, "targets").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	505 let value = ast.getattr(py, "value").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	506
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	507 let mut arguments = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	508 for target in targets.iter(py).unwrap() {
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	509 let target = target.unwrap();
b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	510 let target = parse_expr(py, target);
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	511 arguments.push(target);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	512 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	513
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	514 let value = parse_expr(py, value);
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	515
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	516 stmt::Assign(arguments, value)
8 94ff501bf336 Add ast.AugAssign. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 7 diff changeset	517 } else if is_instance(&ast, &aug_assign_type) {
94ff501bf336 Add ast.AugAssign. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 7 diff changeset	518 let target = ast.getattr(py, "target").unwrap();
94ff501bf336 Add ast.AugAssign. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 7 diff changeset	519 let op = ast.getattr(py, "op").unwrap();
94ff501bf336 Add ast.AugAssign. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 7 diff changeset	520 let value = ast.getattr(py, "value").unwrap();
94ff501bf336 Add ast.AugAssign. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 7 diff changeset	521
94ff501bf336 Add ast.AugAssign. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 7 diff changeset	522 let target = parse_expr(py, target);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	523 let op = parse_operator(py, op);
8 94ff501bf336 Add ast.AugAssign. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 7 diff changeset	524 let value = parse_expr(py, value);
94ff501bf336 Add ast.AugAssign. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 7 diff changeset	525
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	526 stmt::AugAssign(target, op, value)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	527 } else if is_instance(&ast, &import_from_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	528 let module = ast.getattr(py, "module").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	529 let names = ast.getattr(py, "names").unwrap();
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	530 let level = ast.getattr(py, "level").unwrap();
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	531
1 b90e49ab734b Factorise conversion of Python str into Rust String. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 0 diff changeset	532 let module = get_str(py, module);
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	533
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	534 let mut names_ = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	535 for alias in names.iter(py).unwrap() {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	536 let alias = alias.unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	537 let alias = parse_expr(py, alias);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	538 println!("{:?}", alias);
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	539 // XXX
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	540 //names_.push(alias{name: alias, asname: alias});
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	541 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	542
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	543 stmt::ImportFrom(module, names_, None)
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	544 } else if is_instance(&ast, &return_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	545 let value = ast.getattr(py, "value").unwrap();
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	546 if value == py.None() {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	547 stmt::Return(None)
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	548 } else {
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	549 let value = parse_expr(py, value);
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	550 stmt::Return(Some(value))
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	551 }
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	552 } else if is_instance(&ast, &expr_type) {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	553 let value = ast.getattr(py, "value").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	554 let value = parse_expr(py, value);
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	555 stmt::Expr(value)
6 6f2bf13f4cb5 Add ast.While and ast.Break. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 5 diff changeset	556 } else if is_instance(&ast, &break_type) {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	557 stmt::Break
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	558 } else {
13 38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	559 println!("stmt {}", ast);
38b0d63697b1 Import the full AST grammar from CPython 3.5. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 9 diff changeset	560 panic!()
0 211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	561 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	562 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	563
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	564 #[allow(dead_code)]
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	565 pub fn convert_ast(name: String, module: &PyObject) -> Module {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	566 let gil = Python::acquire_gil();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	567 let py = gil.python();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	568
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	569 let builtins_module = py.import("builtins").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	570 let isinstance = builtins_module.get(py, "isinstance").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	571
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	572 let ast_module = py.import("ast").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	573 let module_type = ast_module.get(py, "Module").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	574
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	575 assert!(isinstance.call(py, (module, module_type), None).unwrap().is_true(py).unwrap());
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	576
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	577 let body = module.getattr(py, "body").unwrap();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	578 let mut statements = vec!();
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	579 for statement in body.iter(py).unwrap() {
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	580 let statement = parse_statement(py, statement.unwrap());
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	581 statements.push(statement)
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	582 }
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	583 Module{name: name, statements: statements}
211b0df72e64 Hello world! Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	584 }

Mercurial > python-compiler.rs

annotate src/ast_convert.rs @ 13:38b0d63697b1