Mercurial > python-compiler.rs

diff src/ast_convert.rs @ 48:039f85b187f2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Also handle relative imports.
author Bastien Orivel <eijebong@bananium.fr>
date Tue, 07 Jun 2016 13:34:12 +0200
parents 38f59b0efc2c
children 141f1769e1f0
line wrap: on
line diff
--- a/src/ast_convert.rs
+++ b/src/ast_convert.rs
@@ -545,7 +545,12 @@ fn parse_statement(py: Python, ast: PyOb
         let module = get_str(py, module);
         let names = parse_list(py, names, parse_alias);
 
-        stmt::ImportFrom(module, names, None)
+        if level == py.None() {
+            stmt::ImportFrom(module, names, None)
+        } else {
+            let level = level.extract(py).unwrap();
+            stmt::ImportFrom(module, names, Some(level))
+        }
     } else if is_instance(&ast, &import_type) {
         let names = ast.getattr(py, "names").unwrap();
         let names = parse_list(py, names, parse_alias);