Improve support for generics

2 files changed, 27 insertions, 2 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonExpressionStatement.java b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonExpressionStatement.java
new file mode 100644
index 0000000..0ba1c38
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonExpressionStatement.java
@@ -0,0 +1,25 @@
+package org.zwobble.couscous.backends.python.ast.visitors;
+
+import org.zwobble.couscous.backends.python.ast.PythonExpressionNode;
+import org.zwobble.couscous.backends.python.ast.PythonStatementNode;
+
+public class PythonExpressionStatement implements PythonStatementNode {
+    public static PythonExpressionStatement pythonExpressionStatement(PythonExpressionNode expression) {
+        return new PythonExpressionStatement(expression);
+    }
+
+    private final PythonExpressionNode expression;
+
+    public PythonExpressionStatement(PythonExpressionNode expression) {
+        this.expression = expression;
+    }
+
+    public PythonExpressionNode getExpression() {
+        return expression;
+    }
+
+    @Override
+    public void accept(PythonNodeVisitor visitor) {
+        visitor.visit(this);
+    }
+}
diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
index 8f04787..d43691d 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
@@ -13,7 +13,8 @@ public interface PythonNodeVisitor {
     void visit(PythonGetSliceNode getSlice);
     void visit(PythonNotNode notOperation);
     void visit(PythonBinaryOperation operation);
-    
+
+    void visit(PythonExpressionStatement statement);
     void visit(PythonReturnNode pythonReturn);
     void visit(PythonPassNode pass);
     void visit(PythonClassNode pythonClass);
@@ -24,5 +25,4 @@ public interface PythonNodeVisitor {
     void visit(PythonWhileNode whileLoop);
 
     void visit(PythonModuleNode module);
-
 }