awx-zipline-ai 0.2.1__py3-none-any.whl → 0.3.1__py3-none-any.whl

{ai/chronon → gen_thrift}/hub/ttypes.py

@@ -12,9 +12,8 @@ from thrift.TRecursive import fix_spec
 from uuid import UUID
 
 import sys
-import ai.chronon.api.common.ttypes
-import ai.chronon.api.ttypes
-import ai.chronon.orchestration.ttypes
+import gen_thrift.common.ttypes
+import gen_thrift.api.ttypes
 
 from thrift.transport import TTransport
 all_structs = []
@@ -38,21 +37,6 @@ class Direction(object):
     }
 
 
-class Mode(object):
-    ADHOC = 0
-    SCHEDULED = 1
-
-    _VALUES_TO_NAMES = {
-        0: "ADHOC",
-        1: "SCHEDULED",
-    }
-
-    _NAMES_TO_VALUES = {
-        "ADHOC": 0,
-        "SCHEDULED": 1,
-    }
-
-
 class Status(object):
     WAITING_FOR_UPSTREAM = 0
     WAITING_FOR_RESOURCES = 1
@@ -178,78 +162,6 @@ class LineageRequest(object):
         return not (self == other)
 
 
-class LineageResponse(object):
-    """
-    Attributes:
-     - nodeGraph
-     - mainNode
-
-    """
-    thrift_spec = None
-
-
-    def __init__(self, nodeGraph = None, mainNode = None,):
-        self.nodeGraph = nodeGraph
-        self.mainNode = mainNode
-
-    def read(self, iprot):
-        if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
-            iprot._fast_decode(self, iprot, [self.__class__, self.thrift_spec])
-            return
-        iprot.readStructBegin()
-        while True:
-            (fname, ftype, fid) = iprot.readFieldBegin()
-            if ftype == TType.STOP:
-                break
-            if fid == 1:
-                if ftype == TType.STRUCT:
-                    self.nodeGraph = ai.chronon.orchestration.ttypes.NodeGraph()
-                    self.nodeGraph.read(iprot)
-                else:
-                    iprot.skip(ftype)
-            elif fid == 2:
-                if ftype == TType.STRUCT:
-                    self.mainNode = ai.chronon.orchestration.ttypes.NodeKey()
-                    self.mainNode.read(iprot)
-                else:
-                    iprot.skip(ftype)
-            else:
-                iprot.skip(ftype)
-            iprot.readFieldEnd()
-        iprot.readStructEnd()
-
-    def write(self, oprot):
-        self.validate()
-        if oprot._fast_encode is not None and self.thrift_spec is not None:
-            oprot.trans.write(oprot._fast_encode(self, [self.__class__, self.thrift_spec]))
-            return
-        oprot.writeStructBegin('LineageResponse')
-        if self.nodeGraph is not None:
-            oprot.writeFieldBegin('nodeGraph', TType.STRUCT, 1)
-            self.nodeGraph.write(oprot)
-            oprot.writeFieldEnd()
-        if self.mainNode is not None:
-            oprot.writeFieldBegin('mainNode', TType.STRUCT, 2)
-            self.mainNode.write(oprot)
-            oprot.writeFieldEnd()
-        oprot.writeFieldStop()
-        oprot.writeStructEnd()
-
-    def validate(self):
-        return
-
-    def __repr__(self):
-        L = ['%s=%r' % (key, value)
-             for key, value in self.__dict__.items()]
-        return '%s(%s)' % (self.__class__.__name__, ', '.join(L))
-
-    def __eq__(self, other):
-        return isinstance(other, self.__class__) and self.__dict__ == other.__dict__
-
-    def __ne__(self, other):
-        return not (self == other)
-
-
 class JobTrackerRequest(object):
     """
     Attributes:
@@ -294,7 +206,7 @@ class JobTrackerRequest(object):
                     iprot.skip(ftype)
             elif fid == 10:
                 if ftype == TType.STRUCT:
-                    self.dateRange = ai.chronon.api.common.ttypes.DateRange()
+                    self.dateRange = gen_thrift.common.ttypes.DateRange()
                     self.dateRange.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -347,15 +259,13 @@ class JobTrackerResponse(object):
     """
     Attributes:
      - tasks
-     - mainNode
 
     """
     thrift_spec = None
 
 
-    def __init__(self, tasks = None, mainNode = None,):
+    def __init__(self, tasks = None,):
         self.tasks = tasks
-        self.mainNode = mainNode
 
     def read(self, iprot):
         if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
@@ -377,12 +287,6 @@ class JobTrackerResponse(object):
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
-            elif fid == 2:
-                if ftype == TType.STRUCT:
-                    self.mainNode = ai.chronon.orchestration.ttypes.NodeKey()
-                    self.mainNode.read(iprot)
-                else:
-                    iprot.skip(ftype)
             else:
                 iprot.skip(ftype)
             iprot.readFieldEnd()
@@ -401,10 +305,6 @@ class JobTrackerResponse(object):
                 iter6.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
-        if self.mainNode is not None:
-            oprot.writeFieldBegin('mainNode', TType.STRUCT, 2)
-            self.mainNode.write(oprot)
-            oprot.writeFieldEnd()
         oprot.writeFieldStop()
         oprot.writeStructEnd()
 
@@ -616,7 +516,7 @@ class TaskInfo(object):
                     iprot.skip(ftype)
             elif fid == 5:
                 if ftype == TType.STRUCT:
-                    self.dateRange = ai.chronon.api.common.ttypes.DateRange()
+                    self.dateRange = gen_thrift.common.ttypes.DateRange()
                     self.dateRange.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -917,7 +817,6 @@ class TaskResources(object):
 class Submission(object):
     """
     Attributes:
-     - node
      - submittedTs
      - finishedTs
      - dateRange
@@ -926,8 +825,7 @@ class Submission(object):
     thrift_spec = None
 
 
-    def __init__(self, node = None, submittedTs = None, finishedTs = None, dateRange = None,):
-        self.node = node
+    def __init__(self, submittedTs = None, finishedTs = None, dateRange = None,):
         self.submittedTs = submittedTs
         self.finishedTs = finishedTs
         self.dateRange = dateRange
@@ -941,13 +839,7 @@ class Submission(object):
             (fname, ftype, fid) = iprot.readFieldBegin()
             if ftype == TType.STOP:
                 break
-            if fid == 1:
-                if ftype == TType.STRUCT:
-                    self.node = ai.chronon.orchestration.ttypes.NodeKey()
-                    self.node.read(iprot)
-                else:
-                    iprot.skip(ftype)
-            elif fid == 10:
+            if fid == 10:
                 if ftype == TType.I64:
                     self.submittedTs = iprot.readI64()
                 else:
@@ -959,7 +851,7 @@ class Submission(object):
                     iprot.skip(ftype)
             elif fid == 21:
                 if ftype == TType.STRUCT:
-                    self.dateRange = ai.chronon.api.common.ttypes.DateRange()
+                    self.dateRange = gen_thrift.common.ttypes.DateRange()
                     self.dateRange.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -974,10 +866,6 @@ class Submission(object):
             oprot.trans.write(oprot._fast_encode(self, [self.__class__, self.thrift_spec]))
             return
         oprot.writeStructBegin('Submission')
-        if self.node is not None:
-            oprot.writeFieldBegin('node', TType.STRUCT, 1)
-            self.node.write(oprot)
-            oprot.writeFieldEnd()
         if self.submittedTs is not None:
             oprot.writeFieldBegin('submittedTs', TType.I64, 10)
             oprot.writeI64(self.submittedTs)
@@ -1106,12 +994,6 @@ LineageRequest.thrift_spec = (
     (3, TType.STRING, 'branch', 'UTF8', None, ),  # 3
     (4, TType.I32, 'direction', None, None, ),  # 4
 )
-all_structs.append(LineageResponse)
-LineageResponse.thrift_spec = (
-    None,  # 0
-    (1, TType.STRUCT, 'nodeGraph', [ai.chronon.orchestration.ttypes.NodeGraph, None], None, ),  # 1
-    (2, TType.STRUCT, 'mainNode', [ai.chronon.orchestration.ttypes.NodeKey, None], None, ),  # 2
-)
 all_structs.append(JobTrackerRequest)
 JobTrackerRequest.thrift_spec = (
     None,  # 0
@@ -1124,13 +1006,12 @@ JobTrackerRequest.thrift_spec = (
     None,  # 7
     None,  # 8
     None,  # 9
-    (10, TType.STRUCT, 'dateRange', [ai.chronon.api.common.ttypes.DateRange, None], None, ),  # 10
+    (10, TType.STRUCT, 'dateRange', [gen_thrift.common.ttypes.DateRange, None], None, ),  # 10
 )
 all_structs.append(JobTrackerResponse)
 JobTrackerResponse.thrift_spec = (
     None,  # 0
     (1, TType.LIST, 'tasks', (TType.STRUCT, [TaskInfo, None], False), None, ),  # 1
-    (2, TType.STRUCT, 'mainNode', [ai.chronon.orchestration.ttypes.NodeKey, None], None, ),  # 2
 )
 all_structs.append(SubmissionsRequest)
 SubmissionsRequest.thrift_spec = (
@@ -1149,7 +1030,7 @@ TaskInfo.thrift_spec = (
     (2, TType.STRING, 'logPath', 'UTF8', None, ),  # 2
     (3, TType.STRING, 'trackerUrl', 'UTF8', None, ),  # 3
     (4, TType.STRUCT, 'taskArgs', [TaskArgs, None], None, ),  # 4
-    (5, TType.STRUCT, 'dateRange', [ai.chronon.api.common.ttypes.DateRange, None], None, ),  # 5
+    (5, TType.STRUCT, 'dateRange', [gen_thrift.common.ttypes.DateRange, None], None, ),  # 5
     None,  # 6
     None,  # 7
     None,  # 8
@@ -1194,7 +1075,7 @@ TaskResources.thrift_spec = (
 all_structs.append(Submission)
 Submission.thrift_spec = (
     None,  # 0
-    (1, TType.STRUCT, 'node', [ai.chronon.orchestration.ttypes.NodeKey, None], None, ),  # 1
+    None,  # 1
     None,  # 2
     None,  # 3
     None,  # 4
@@ -1214,7 +1095,7 @@ Submission.thrift_spec = (
     None,  # 18
     None,  # 19
     (20, TType.I64, 'finishedTs', None, None, ),  # 20
-    (21, TType.STRUCT, 'dateRange', [ai.chronon.api.common.ttypes.DateRange, None], None, ),  # 21
+    (21, TType.STRUCT, 'dateRange', [gen_thrift.common.ttypes.DateRange, None], None, ),  # 21
 )
 all_structs.append(ConfRequest)
 ConfRequest.thrift_spec = (