awx-zipline-ai 0.2.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- agent/__init__.py +1 -0
- agent/constants.py +15 -0
- agent/ttypes.py +1684 -0
- ai/__init__.py +0 -0
- ai/chronon/__init__.py +0 -0
- ai/chronon/airflow_helpers.py +251 -0
- ai/chronon/api/__init__.py +1 -0
- ai/chronon/api/common/__init__.py +1 -0
- ai/chronon/api/common/constants.py +15 -0
- ai/chronon/api/common/ttypes.py +1844 -0
- ai/chronon/api/constants.py +15 -0
- ai/chronon/api/ttypes.py +3624 -0
- ai/chronon/cli/compile/column_hashing.py +313 -0
- ai/chronon/cli/compile/compile_context.py +177 -0
- ai/chronon/cli/compile/compiler.py +160 -0
- ai/chronon/cli/compile/conf_validator.py +590 -0
- ai/chronon/cli/compile/display/class_tracker.py +112 -0
- ai/chronon/cli/compile/display/compile_status.py +95 -0
- ai/chronon/cli/compile/display/compiled_obj.py +12 -0
- ai/chronon/cli/compile/display/console.py +3 -0
- ai/chronon/cli/compile/display/diff_result.py +46 -0
- ai/chronon/cli/compile/fill_templates.py +40 -0
- ai/chronon/cli/compile/parse_configs.py +141 -0
- ai/chronon/cli/compile/parse_teams.py +238 -0
- ai/chronon/cli/compile/serializer.py +115 -0
- ai/chronon/cli/git_utils.py +156 -0
- ai/chronon/cli/logger.py +61 -0
- ai/chronon/constants.py +3 -0
- ai/chronon/eval/__init__.py +122 -0
- ai/chronon/eval/query_parsing.py +19 -0
- ai/chronon/eval/sample_tables.py +100 -0
- ai/chronon/eval/table_scan.py +186 -0
- ai/chronon/fetcher/__init__.py +1 -0
- ai/chronon/fetcher/constants.py +15 -0
- ai/chronon/fetcher/ttypes.py +127 -0
- ai/chronon/group_by.py +692 -0
- ai/chronon/hub/__init__.py +1 -0
- ai/chronon/hub/constants.py +15 -0
- ai/chronon/hub/ttypes.py +1228 -0
- ai/chronon/join.py +566 -0
- ai/chronon/logger.py +24 -0
- ai/chronon/model.py +35 -0
- ai/chronon/observability/__init__.py +1 -0
- ai/chronon/observability/constants.py +15 -0
- ai/chronon/observability/ttypes.py +2192 -0
- ai/chronon/orchestration/__init__.py +1 -0
- ai/chronon/orchestration/constants.py +15 -0
- ai/chronon/orchestration/ttypes.py +4406 -0
- ai/chronon/planner/__init__.py +1 -0
- ai/chronon/planner/constants.py +15 -0
- ai/chronon/planner/ttypes.py +1686 -0
- ai/chronon/query.py +126 -0
- ai/chronon/repo/__init__.py +40 -0
- ai/chronon/repo/aws.py +298 -0
- ai/chronon/repo/cluster.py +65 -0
- ai/chronon/repo/compile.py +56 -0
- ai/chronon/repo/constants.py +164 -0
- ai/chronon/repo/default_runner.py +291 -0
- ai/chronon/repo/explore.py +421 -0
- ai/chronon/repo/extract_objects.py +137 -0
- ai/chronon/repo/gcp.py +585 -0
- ai/chronon/repo/gitpython_utils.py +14 -0
- ai/chronon/repo/hub_runner.py +171 -0
- ai/chronon/repo/hub_uploader.py +108 -0
- ai/chronon/repo/init.py +53 -0
- ai/chronon/repo/join_backfill.py +105 -0
- ai/chronon/repo/run.py +293 -0
- ai/chronon/repo/serializer.py +141 -0
- ai/chronon/repo/team_json_utils.py +46 -0
- ai/chronon/repo/utils.py +472 -0
- ai/chronon/repo/zipline.py +51 -0
- ai/chronon/repo/zipline_hub.py +105 -0
- ai/chronon/resources/gcp/README.md +174 -0
- ai/chronon/resources/gcp/group_bys/test/__init__.py +0 -0
- ai/chronon/resources/gcp/group_bys/test/data.py +34 -0
- ai/chronon/resources/gcp/joins/test/__init__.py +0 -0
- ai/chronon/resources/gcp/joins/test/data.py +30 -0
- ai/chronon/resources/gcp/sources/test/__init__.py +0 -0
- ai/chronon/resources/gcp/sources/test/data.py +23 -0
- ai/chronon/resources/gcp/teams.py +70 -0
- ai/chronon/resources/gcp/zipline-cli-install.sh +54 -0
- ai/chronon/source.py +88 -0
- ai/chronon/staging_query.py +185 -0
- ai/chronon/types.py +57 -0
- ai/chronon/utils.py +557 -0
- ai/chronon/windows.py +50 -0
- awx_zipline_ai-0.2.0.dist-info/METADATA +173 -0
- awx_zipline_ai-0.2.0.dist-info/RECORD +93 -0
- awx_zipline_ai-0.2.0.dist-info/WHEEL +5 -0
- awx_zipline_ai-0.2.0.dist-info/entry_points.txt +2 -0
- awx_zipline_ai-0.2.0.dist-info/licenses/LICENSE +202 -0
- awx_zipline_ai-0.2.0.dist-info/top_level.txt +3 -0
- jars/__init__.py +0 -0
|
@@ -0,0 +1,127 @@
|
|
|
1
|
+
#
|
|
2
|
+
# Autogenerated by Thrift Compiler (0.22.0)
|
|
3
|
+
#
|
|
4
|
+
# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
|
|
5
|
+
#
|
|
6
|
+
# options string: py
|
|
7
|
+
#
|
|
8
|
+
|
|
9
|
+
from thrift.Thrift import TType, TMessageType, TFrozenDict, TException, TApplicationException
|
|
10
|
+
from thrift.protocol.TProtocol import TProtocolException
|
|
11
|
+
from thrift.TRecursive import fix_spec
|
|
12
|
+
from uuid import UUID
|
|
13
|
+
|
|
14
|
+
import sys
|
|
15
|
+
|
|
16
|
+
from thrift.transport import TTransport
|
|
17
|
+
all_structs = []
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
class TileKey(object):
|
|
21
|
+
"""
|
|
22
|
+
Attributes:
|
|
23
|
+
- dataset
|
|
24
|
+
- keyBytes
|
|
25
|
+
- tileSizeMillis
|
|
26
|
+
- tileStartTimestampMillis
|
|
27
|
+
|
|
28
|
+
"""
|
|
29
|
+
thrift_spec = None
|
|
30
|
+
|
|
31
|
+
|
|
32
|
+
def __init__(self, dataset = None, keyBytes = None, tileSizeMillis = None, tileStartTimestampMillis = None,):
|
|
33
|
+
self.dataset = dataset
|
|
34
|
+
self.keyBytes = keyBytes
|
|
35
|
+
self.tileSizeMillis = tileSizeMillis
|
|
36
|
+
self.tileStartTimestampMillis = tileStartTimestampMillis
|
|
37
|
+
|
|
38
|
+
def read(self, iprot):
|
|
39
|
+
if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
|
|
40
|
+
iprot._fast_decode(self, iprot, [self.__class__, self.thrift_spec])
|
|
41
|
+
return
|
|
42
|
+
iprot.readStructBegin()
|
|
43
|
+
while True:
|
|
44
|
+
(fname, ftype, fid) = iprot.readFieldBegin()
|
|
45
|
+
if ftype == TType.STOP:
|
|
46
|
+
break
|
|
47
|
+
if fid == 1:
|
|
48
|
+
if ftype == TType.STRING:
|
|
49
|
+
self.dataset = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
|
|
50
|
+
else:
|
|
51
|
+
iprot.skip(ftype)
|
|
52
|
+
elif fid == 2:
|
|
53
|
+
if ftype == TType.LIST:
|
|
54
|
+
self.keyBytes = []
|
|
55
|
+
(_etype3, _size0) = iprot.readListBegin()
|
|
56
|
+
for _i4 in range(_size0):
|
|
57
|
+
_elem5 = iprot.readByte()
|
|
58
|
+
self.keyBytes.append(_elem5)
|
|
59
|
+
iprot.readListEnd()
|
|
60
|
+
else:
|
|
61
|
+
iprot.skip(ftype)
|
|
62
|
+
elif fid == 3:
|
|
63
|
+
if ftype == TType.I64:
|
|
64
|
+
self.tileSizeMillis = iprot.readI64()
|
|
65
|
+
else:
|
|
66
|
+
iprot.skip(ftype)
|
|
67
|
+
elif fid == 4:
|
|
68
|
+
if ftype == TType.I64:
|
|
69
|
+
self.tileStartTimestampMillis = iprot.readI64()
|
|
70
|
+
else:
|
|
71
|
+
iprot.skip(ftype)
|
|
72
|
+
else:
|
|
73
|
+
iprot.skip(ftype)
|
|
74
|
+
iprot.readFieldEnd()
|
|
75
|
+
iprot.readStructEnd()
|
|
76
|
+
|
|
77
|
+
def write(self, oprot):
|
|
78
|
+
self.validate()
|
|
79
|
+
if oprot._fast_encode is not None and self.thrift_spec is not None:
|
|
80
|
+
oprot.trans.write(oprot._fast_encode(self, [self.__class__, self.thrift_spec]))
|
|
81
|
+
return
|
|
82
|
+
oprot.writeStructBegin('TileKey')
|
|
83
|
+
if self.dataset is not None:
|
|
84
|
+
oprot.writeFieldBegin('dataset', TType.STRING, 1)
|
|
85
|
+
oprot.writeString(self.dataset.encode('utf-8') if sys.version_info[0] == 2 else self.dataset)
|
|
86
|
+
oprot.writeFieldEnd()
|
|
87
|
+
if self.keyBytes is not None:
|
|
88
|
+
oprot.writeFieldBegin('keyBytes', TType.LIST, 2)
|
|
89
|
+
oprot.writeListBegin(TType.BYTE, len(self.keyBytes))
|
|
90
|
+
for iter6 in self.keyBytes:
|
|
91
|
+
oprot.writeByte(iter6)
|
|
92
|
+
oprot.writeListEnd()
|
|
93
|
+
oprot.writeFieldEnd()
|
|
94
|
+
if self.tileSizeMillis is not None:
|
|
95
|
+
oprot.writeFieldBegin('tileSizeMillis', TType.I64, 3)
|
|
96
|
+
oprot.writeI64(self.tileSizeMillis)
|
|
97
|
+
oprot.writeFieldEnd()
|
|
98
|
+
if self.tileStartTimestampMillis is not None:
|
|
99
|
+
oprot.writeFieldBegin('tileStartTimestampMillis', TType.I64, 4)
|
|
100
|
+
oprot.writeI64(self.tileStartTimestampMillis)
|
|
101
|
+
oprot.writeFieldEnd()
|
|
102
|
+
oprot.writeFieldStop()
|
|
103
|
+
oprot.writeStructEnd()
|
|
104
|
+
|
|
105
|
+
def validate(self):
|
|
106
|
+
return
|
|
107
|
+
|
|
108
|
+
def __repr__(self):
|
|
109
|
+
L = ['%s=%r' % (key, value)
|
|
110
|
+
for key, value in self.__dict__.items()]
|
|
111
|
+
return '%s(%s)' % (self.__class__.__name__, ', '.join(L))
|
|
112
|
+
|
|
113
|
+
def __eq__(self, other):
|
|
114
|
+
return isinstance(other, self.__class__) and self.__dict__ == other.__dict__
|
|
115
|
+
|
|
116
|
+
def __ne__(self, other):
|
|
117
|
+
return not (self == other)
|
|
118
|
+
all_structs.append(TileKey)
|
|
119
|
+
TileKey.thrift_spec = (
|
|
120
|
+
None, # 0
|
|
121
|
+
(1, TType.STRING, 'dataset', 'UTF8', None, ), # 1
|
|
122
|
+
(2, TType.LIST, 'keyBytes', (TType.BYTE, None, False), None, ), # 2
|
|
123
|
+
(3, TType.I64, 'tileSizeMillis', None, None, ), # 3
|
|
124
|
+
(4, TType.I64, 'tileStartTimestampMillis', None, None, ), # 4
|
|
125
|
+
)
|
|
126
|
+
fix_spec(all_structs)
|
|
127
|
+
del all_structs
|