Make more work

Author: Ghislain Fourny

src/jsoniq/sequence.py

@@ -0,0 +1,15 @@
+from pyspark.sql import SparkSession
+import json
+
+class SequenceOfItems:
+    def __init__(self, sequence):
+        self._jsequence = sequence
+
+    def getAsJSONList(self):
+        for s in self._jsequence.getAsList():
+            print(s.serialize());
+        return [json.loads(l.serialize()) for l in self._jsequence.getAsList()]
+
+    def __getattr__(self, item):
+        print(f"Accessing attribute: {item}")
+        return getattr(self._jsequence, item)

src/jsoniq/session.py

@@ -1,4 +1,5 @@
 from pyspark.sql import SparkSession
+from .sequence import SequenceOfItems
 
 class MetaRumbleSession(type):
     def __getattr__(cls, item):
@@ -60,7 +61,7 @@ def bindDataFrameAsVariable(self, name: str, df):
 
     def jsoniq(self, str):
         sequence = self._jrumblesession.runQuery(str);
-        return sequence;
+        return SequenceOfItems(sequence);
 
     def __getattr__(self, item):
         print(f"Accessing attribute: {item}")