Revert "Update Annotations to generate code"

This reverts commit 62e2933.
zinggAI · Mar 7, 2024 · 259fb60 · 259fb60
1 parent 6d87021
commit 259fb60
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 23 deletions.
diff --git a/common/py/src/main/java/zingg/common/py/processors/PythonClassProcessor.java b/common/py/src/main/java/zingg/common/py/processors/PythonClassProcessor.java
@@ -53,14 +53,14 @@ public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment
                 // __init__ method
                 System.out.println("    def __init__(self" +
                         generateConstructorParameters(classElement) + "):");
-                generateClassInitializationCode(classElement, element);
-
-                // for (VariableElement field : ElementFilter.fieldsIn(classElement.getEnclosedElements())) {
-                //     if (!field.getSimpleName().contentEquals("serialVersionUID")) {
-                //         generateFieldInitializationCode(field, element);
-                //     }
-                // }
-
+                if (element.getSimpleName().contentEquals("Pipe")) {
+                    generateClassInitializationCode(classElement, element);
+                }
+                for (VariableElement field : ElementFilter.fieldsIn(classElement.getEnclosedElements())) {
+                    if (!field.getSimpleName().contentEquals("serialVersionUID")) {
+                        generateFieldInitializationCode(field, element);
+                    }
+                }
                 for (ExecutableElement methodElement : ElementFilter.methodsIn(classElement.getEnclosedElements())) {
                     if (methodElement.getAnnotation(PythonMethod.class) != null) {
                         methodNames.add(methodElement.getSimpleName().toString());
@@ -140,16 +140,17 @@ else if (element.getSimpleName().contentEquals("FieldDefinition")) {
             fileWriter.write("        return self.fielddefinition\n");
         }
         fileWriter.write("\n");
+        System.out.println("        self." + element.getSimpleName().toString().toLowerCase() + " = getJVM().zingg.spark.client.pipe.SparkPipe()");
     }
 
-    // private void generateFieldInitializationCode(VariableElement field, Element element) {
-    //     String fieldName = field.getSimpleName().toString();
-    //     String fieldAssignment = "self." + element.getSimpleName().toString().toLowerCase() + "." + fieldName + " = " + fieldName;
+    private void generateFieldInitializationCode(VariableElement field, Element element) {
+        String fieldName = field.getSimpleName().toString();
+        String fieldAssignment = "self." + element.getSimpleName().toString().toLowerCase() + "." + fieldName + " = " + fieldName;
 
-    //     if (!fieldName.startsWith("FORMAT_")) {
-    //         System.out.println("        " + fieldAssignment);
-    //     }
-    // }
+        if (!fieldName.startsWith("FORMAT_")) {
+            System.out.println("        " + fieldAssignment);
+        }
+    }
 
     private String generateConstructorParameters(TypeElement classElement) {
         StringBuilder parameters = new StringBuilder();

diff --git a/common/py/src/main/java/zingg/common/py/processors/PythonMethodProcessor.java b/common/py/src/main/java/zingg/common/py/processors/PythonMethodProcessor.java
@@ -7,7 +7,7 @@
 import javax.lang.model.type.TypeMirror;
 import javax.lang.model.type.TypeKind;
 import java.util.Set;
-// import java.util.logging.Logger;
+import java.util.logging.Logger;
 
 import javax.lang.model.element.*;
 import zingg.common.py.annotations.*;
@@ -38,8 +38,7 @@ public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment
                     if (methodNames.contains(methodElement.getSimpleName().toString())) {
                         // LOG.info("Generating Python method for: " + methodElement.getSimpleName());
                         System.out.println("    def " + methodElement.getSimpleName() + "(self" +
-                                generateMethodSignature(methodElement) + "):");
-                        generateMethodReturn(methodElement);
+                                generateMethodSignature(methodElement) + "):\n        " + generateMethodReturn(methodElement));
                         generateFieldAssignment(methodElement);
                     }
                 }
@@ -65,15 +64,14 @@ private String generateMethodParameters(ExecutableElement methodElement) {
         return parameters.toString();
     }
 
-    private void generateMethodReturn(ExecutableElement methodElement) {
+    private String generateMethodReturn(ExecutableElement methodElement) {
         TypeMirror returnType = methodElement.getReturnType();
         if (returnType.getKind() == TypeKind.VOID) {
-            return;
+            return "";
         } else {
             String returnTypeString = resolveType(returnType);
-            String methodName = methodElement.getSimpleName().toString();
-            String className = methodElement.getEnclosingElement().getSimpleName().toString();
-            System.out.println("        return self." + className.toLowerCase() + "." + methodName + "()");
+            String variableName = methodElement.getSimpleName().toString();
+            return "return " + variableName;
         }
     }