diff src/ast_convert.rs @ 65:ce5e27a3f277

Add ClassDef.decorator_list support.
author Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date Mon, 13 Jun 2016 01:27:10 +0100
parents 53817b39f139
children 8ce78e2ba48c
line wrap: on
line diff
--- a/src/ast_convert.rs
+++ b/src/ast_convert.rs
@@ -553,13 +553,14 @@ fn parse_statement(py: Python, ast: PyOb
         let bases = ast.getattr(py, "bases").unwrap();
         //let keywords = ast.getattr(py, "keywords").unwrap();
         let body = ast.getattr(py, "body").unwrap();
-        //let decorator_list = ast.getattr(py, "decorator_list").unwrap();
+        let decorator_list = ast.getattr(py, "decorator_list").unwrap();
 
         let name = get_str(py, name);
         let bases = parse_list(py, bases, parse_expr);
         let body = parse_list(py, body, parse_statement);
+        let decorator_list = parse_list(py, decorator_list, parse_expr);
 
-        stmt::ClassDef(name, bases, vec!(), body, vec!())
+        stmt::ClassDef(name, bases, vec!(), body, decorator_list)
     } else if is_instance(&ast, &function_def_type) {
         let name = ast.getattr(py, "name").unwrap();
         let args = ast.getattr(py, "args").unwrap();