Python backend: import class into module

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();
+                });
         });
     }