Create notion of ClassName

1 files changed, 6 insertions, 1 deletions

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..d8b97ad 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,17 @@
 package org.zwobble.couscous.backends.python.ast;
 
+import org.zwobble.couscous.ast.ClassName;
 import org.zwobble.couscous.backends.python.ast.visitors.PythonNodeVisitor;
 
 import lombok.Value;
 
 @Value(staticConstructor="pythonImport")
 public class PythonImportNode implements PythonStatementNode {
-    String name;
+    public static PythonImportNode pythonImport(String name) {
+        return pythonImport(ClassName.of(name));
+    }
+    
+    ClassName name;
     
     @Override
     public void accept(PythonNodeVisitor visitor) {