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

---
 .../couscous/backends/python/ast/PythonImportAliasNode.java       | 8 ++++++++
 .../zwobble/couscous/backends/python/ast/PythonImportNode.java    | 5 ++++-
 2 files changed, 12 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/org/zwobble/couscous/backends/python/ast/PythonImportAliasNode.java

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

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) {
-- 
cgit v1.2.3