diff options
Diffstat (limited to 'src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java')
| -rw-r--r-- | src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java index 9c82b5d..249d9fa 100644 --- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java +++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java @@ -164,9 +164,21 @@ public class PythonSerializer implements PythonNodeVisitor { @Override public void visit(PythonImportNode importNode) { writer.writeStatement(() -> { + writer.writeKeyword("from"); + writer.writeSpace(); + writer.writeIdentifier(importNode.getModuleName()); + writer.writeSpace(); writer.writeKeyword("import"); writer.writeSpace(); - writer.writeIdentifier(importNode.getName()); + writeWithSeparator( + importNode.getAliases(), + alias -> { + writer.writeIdentifier(alias.getName()); + }, + () -> { + writer.writeSymbol(","); + writer.writeSpace(); + }); }); } |
