Python backend: import class into module

2 files changed, 12 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/PythonImportAliasNode.java b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonImportAliasNode.java
new file mode 100644
index 0000000..9fb531a
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonImportAliasNode.java
@@ -0,0 +1,8 @@
+package org.zwobble.couscous.backends.python.ast;
+
+import lombok.Value;
+
+@Value(staticConstructor="pythonImportAlias")
+public class PythonImportAliasNode {
+    String name;
+}
diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/PythonImportNode.java b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonImportNode.java
index 2d38004..c639829 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/ast/PythonImportNode.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonImportNode.java
@@ -1,12 +1,15 @@
 package org.zwobble.couscous.backends.python.ast;
 
+import java.util.List;
+
 import org.zwobble.couscous.backends.python.ast.visitors.PythonNodeVisitor;
 
 import lombok.Value;
 
 @Value(staticConstructor="pythonImport")
 public class PythonImportNode implements PythonStatementNode {
-    String name;
+    String moduleName;
+    List<PythonImportAliasNode> aliases;
     
     @Override
     public void accept(PythonNodeVisitor visitor) {