Handle import as correctly.

This commit is contained in:
Kunal Parmar
2012-06-10 02:57:28 -07:00
parent 6a1eba1622
commit 2cc5e52201
3 changed files with 10 additions and 1 deletions

View File

@@ -2527,7 +2527,12 @@ void print_src(PycRef<ASTNode> node, PycModule* mod)
}
} else {
fprintf(pyc_output, "import ");
print_src(import->name(), mod);
PycRef<ASTNode> import_name = import->name();
print_src(import_name, mod);
if (!dest.cast<ASTName>()->name()->isEqual(import_name.cast<ASTName>()->name().cast<PycObject>())) {
fprintf(pyc_output, " as ");
print_src(dest, mod);
}
}
} else {
if (src->type() == ASTNode::NODE_BINARY &&

4
tests/test_import.py Normal file
View File

@@ -0,0 +1,4 @@
import sys
import unittest as ut
from logging import Logger, LogRecord
from os import *

BIN
tests/test_import.pyc Normal file

Binary file not shown.