Python backend: import class into module

1 files changed, 5 insertions, 4 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/TypeName.java b/src/main/java/org/zwobble/couscous/ast/TypeName.java
index ce80f82..cbc0fe0 100644
--- a/src/main/java/org/zwobble/couscous/ast/TypeName.java
+++ b/src/main/java/org/zwobble/couscous/ast/TypeName.java
@@ -8,11 +8,12 @@ import lombok.Value;
 @AllArgsConstructor(access=AccessLevel.PRIVATE)
 public class TypeName {
     public static TypeName of(String qualifiedName) {
-        return new TypeName(
-            qualifiedName,
-            qualifiedName.substring(qualifiedName.lastIndexOf(".") + 1));
+        return new TypeName(qualifiedName);
     }
     
     String qualifiedName;
-    String simpleName;
+    
+    public String getSimpleName() {
+        return qualifiedName.substring(qualifiedName.lastIndexOf(".") + 1);
+    }
 }