From fda9d33c0d630906baad5c60e75db5cd8502bfe8 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Tue, 6 Oct 2015 11:13:43 +0100
Subject: Python backend: import class into module

---
 .../zwobble/couscous/backends/python/PythonSerializer.java | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java')

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();
+                });
         });
     }
 
-- 
cgit v1.2.3