diff options
Diffstat (limited to 'src/main/java')
3 files changed, 3 insertions, 8 deletions
diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java index 1524be7..4f54d45 100644 --- a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java +++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java @@ -80,7 +80,7 @@ public class PythonCodeGenerator { private static Stream<PythonImportNode> generateImports(ClassNode classNode) { val classes = findReferencedClasses(classNode); return classes.stream() - .map(name -> pythonImport(name)); + .map(name -> pythonImport(name.getQualifiedName())); } private static Set<TypeName> findReferencedClasses(ClassNode classNode) { 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 be53faa..9c82b5d 100644 --- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java +++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java @@ -166,7 +166,7 @@ public class PythonSerializer implements PythonNodeVisitor { writer.writeStatement(() -> { writer.writeKeyword("import"); writer.writeSpace(); - writer.writeIdentifier(importNode.getName().getQualifiedName()); + writer.writeIdentifier(importNode.getName()); }); } 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 eb6f743..2d38004 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,17 +1,12 @@ package org.zwobble.couscous.backends.python.ast; -import org.zwobble.couscous.ast.TypeName; import org.zwobble.couscous.backends.python.ast.visitors.PythonNodeVisitor; import lombok.Value; @Value(staticConstructor="pythonImport") public class PythonImportNode implements PythonStatementNode { - public static PythonImportNode pythonImport(String name) { - return pythonImport(TypeName.of(name)); - } - - TypeName name; + String name; @Override public void accept(PythonNodeVisitor visitor) { |