aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
diff options
context:
space:
mode:
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.java14
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();
+ });
});
}