9 from thrift.Thrift
import TType, TMessageType, TFrozenDict, TException, TApplicationException
10 from thrift.protocol.TProtocol
import TProtocolException
11 from thrift.TRecursive
import fix_spec
15 from thrift.transport
import TTransport
63 ColumnTextEncodingDict = 43
64 ColumnListTextEncodingDict = 44
95 26:
"GeoMultiPolygon",
103 34:
"TextEncodingNone",
104 35:
"TextEncodingDict",
105 36:
"ColumnListInt8",
106 37:
"ColumnListInt16",
107 38:
"ColumnListInt32",
108 39:
"ColumnListInt64",
109 40:
"ColumnListFloat",
110 41:
"ColumnListDouble",
111 42:
"ColumnListBool",
112 43:
"ColumnTextEncodingDict",
113 44:
"ColumnListTextEncodingDict",
114 45:
"ColumnTimestamp",
145 "GeoMultiPolygon": 26,
153 "TextEncodingNone": 34,
154 "TextEncodingDict": 35,
155 "ColumnListInt8": 36,
156 "ColumnListInt16": 37,
157 "ColumnListInt32": 38,
158 "ColumnListInt64": 39,
159 "ColumnListFloat": 40,
160 "ColumnListDouble": 41,
161 "ColumnListBool": 42,
162 "ColumnTextEncodingDict": 43,
163 "ColumnListTextEncodingDict": 44,
164 "ColumnTimestamp": 45,
171 kUserSpecifiedConstantParameter = 1
172 kUserSpecifiedRowMultiplier = 2
173 kTableFunctionSpecifiedParameter = 3
174 kPreFlightParameter = 4
178 1:
"kUserSpecifiedConstantParameter",
179 2:
"kUserSpecifiedRowMultiplier",
180 3:
"kTableFunctionSpecifiedParameter",
181 4:
"kPreFlightParameter",
186 "kUserSpecifiedConstantParameter": 1,
187 "kUserSpecifiedRowMultiplier": 2,
188 "kTableFunctionSpecifiedParameter": 3,
189 "kPreFlightParameter": 4,
203 def __init__(self, name=None, argTypes=None, retType=None,):
209 if iprot._fast_decode
is not None and isinstance(iprot.trans, TTransport.CReadableTransport)
and self.thrift_spec
is not None:
210 iprot._fast_decode(self, iprot, [self.__class__, self.thrift_spec])
212 iprot.readStructBegin()
214 (fname, ftype, fid) = iprot.readFieldBegin()
215 if ftype == TType.STOP:
218 if ftype == TType.STRING:
219 self.
name = iprot.readString().decode(
'utf-8', errors=
'replace')
if sys.version_info[0] == 2
else iprot.readString()
223 if ftype == TType.LIST:
225 (_etype3, _size0) = iprot.readListBegin()
226 for _i4
in range(_size0):
227 _elem5 = iprot.readI32()
228 self.argTypes.append(_elem5)
233 if ftype == TType.I32:
240 iprot.readStructEnd()
243 if oprot._fast_encode
is not None and self.thrift_spec
is not None:
244 oprot.trans.write(oprot._fast_encode(self, [self.__class__, self.thrift_spec]))
246 oprot.writeStructBegin(
'TUserDefinedFunction')
247 if self.
name is not None:
248 oprot.writeFieldBegin(
'name', TType.STRING, 1)
249 oprot.writeString(self.name.encode(
'utf-8')
if sys.version_info[0] == 2
else self.
name)
250 oprot.writeFieldEnd()
252 oprot.writeFieldBegin(
'argTypes', TType.LIST, 2)
253 oprot.writeListBegin(TType.I32, len(self.
argTypes))
255 oprot.writeI32(iter6)
257 oprot.writeFieldEnd()
259 oprot.writeFieldBegin(
'retType', TType.I32, 3)
261 oprot.writeFieldEnd()
262 oprot.writeFieldStop()
263 oprot.writeStructEnd()
269 L = [
'%s=%r' % (key, value)
270 for key, value
in self.__dict__.items()]
271 return '%s(%s)' % (self.__class__.__name__,
', '.
join(L))
274 return isinstance(other, self.__class__)
and self.
__dict__ == other.__dict__
277 return not (self == other)
294 def __init__(self, name=None, sizerType=None, sizerArgPos=None, inputArgTypes=None, outputArgTypes=None, sqlArgTypes=None, annotations=None,):
304 if iprot._fast_decode
is not None and isinstance(iprot.trans, TTransport.CReadableTransport)
and self.thrift_spec
is not None:
305 iprot._fast_decode(self, iprot, [self.__class__, self.thrift_spec])
307 iprot.readStructBegin()
309 (fname, ftype, fid) = iprot.readFieldBegin()
310 if ftype == TType.STOP:
313 if ftype == TType.STRING:
314 self.
name = iprot.readString().decode(
'utf-8', errors=
'replace')
if sys.version_info[0] == 2
else iprot.readString()
318 if ftype == TType.I32:
323 if ftype == TType.I32:
328 if ftype == TType.LIST:
330 (_etype10, _size7) = iprot.readListBegin()
331 for _i11
in range(_size7):
332 _elem12 = iprot.readI32()
333 self.inputArgTypes.append(_elem12)
338 if ftype == TType.LIST:
340 (_etype16, _size13) = iprot.readListBegin()
341 for _i17
in range(_size13):
342 _elem18 = iprot.readI32()
343 self.outputArgTypes.append(_elem18)
348 if ftype == TType.LIST:
350 (_etype22, _size19) = iprot.readListBegin()
351 for _i23
in range(_size19):
352 _elem24 = iprot.readI32()
353 self.sqlArgTypes.append(_elem24)
358 if ftype == TType.LIST:
360 (_etype28, _size25) = iprot.readListBegin()
361 for _i29
in range(_size25):
363 (_ktype32, _vtype33, _size31) = iprot.readMapBegin()
364 for _i35
in range(_size31):
365 _key36 = iprot.readString().decode(
'utf-8', errors=
'replace')
if sys.version_info[0] == 2
else iprot.readString()
366 _val37 = iprot.readString().decode(
'utf-8', errors=
'replace')
if sys.version_info[0] == 2
else iprot.readString()
367 _elem30[_key36] = _val37
369 self.annotations.append(_elem30)
376 iprot.readStructEnd()
379 if oprot._fast_encode
is not None and self.thrift_spec
is not None:
380 oprot.trans.write(oprot._fast_encode(self, [self.__class__, self.thrift_spec]))
382 oprot.writeStructBegin(
'TUserDefinedTableFunction')
383 if self.
name is not None:
384 oprot.writeFieldBegin(
'name', TType.STRING, 1)
385 oprot.writeString(self.name.encode(
'utf-8')
if sys.version_info[0] == 2
else self.
name)
386 oprot.writeFieldEnd()
388 oprot.writeFieldBegin(
'sizerType', TType.I32, 2)
390 oprot.writeFieldEnd()
392 oprot.writeFieldBegin(
'sizerArgPos', TType.I32, 3)
394 oprot.writeFieldEnd()
396 oprot.writeFieldBegin(
'inputArgTypes', TType.LIST, 4)
399 oprot.writeI32(iter38)
401 oprot.writeFieldEnd()
403 oprot.writeFieldBegin(
'outputArgTypes', TType.LIST, 5)
406 oprot.writeI32(iter39)
408 oprot.writeFieldEnd()
410 oprot.writeFieldBegin(
'sqlArgTypes', TType.LIST, 6)
411 oprot.writeListBegin(TType.I32, len(self.
sqlArgTypes))
413 oprot.writeI32(iter40)
415 oprot.writeFieldEnd()
417 oprot.writeFieldBegin(
'annotations', TType.LIST, 7)
418 oprot.writeListBegin(TType.MAP, len(self.
annotations))
420 oprot.writeMapBegin(TType.STRING, TType.STRING, len(iter41))
421 for kiter42, viter43
in iter41.items():
422 oprot.writeString(kiter42.encode(
'utf-8')
if sys.version_info[0] == 2
else kiter42)
423 oprot.writeString(viter43.encode(
'utf-8')
if sys.version_info[0] == 2
else viter43)
426 oprot.writeFieldEnd()
427 oprot.writeFieldStop()
428 oprot.writeStructEnd()
434 L = [
'%s=%r' % (key, value)
435 for key, value
in self.__dict__.items()]
436 return '%s(%s)' % (self.__class__.__name__,
', '.
join(L))
439 return isinstance(other, self.__class__)
and self.
__dict__ == other.__dict__
442 return not (self == other)
443 all_structs.append(TUserDefinedFunction)
444 TUserDefinedFunction.thrift_spec = (
446 (1, TType.STRING,
'name',
'UTF8',
None, ),
447 (2, TType.LIST,
'argTypes', (TType.I32,
None,
False),
None, ),
448 (3, TType.I32,
'retType',
None,
None, ),
450 all_structs.append(TUserDefinedTableFunction)
451 TUserDefinedTableFunction.thrift_spec = (
453 (1, TType.STRING,
'name',
'UTF8',
None, ),
454 (2, TType.I32,
'sizerType',
None,
None, ),
455 (3, TType.I32,
'sizerArgPos',
None,
None, ),
456 (4, TType.LIST,
'inputArgTypes', (TType.I32,
None,
False),
None, ),
457 (5, TType.LIST,
'outputArgTypes', (TType.I32,
None,
False),
None, ),
458 (6, TType.LIST,
'sqlArgTypes', (TType.I32,
None,
False),
None, ),
459 (7, TType.LIST,
'annotations', (TType.MAP, (TType.STRING,
'UTF8', TType.STRING,
'UTF8',
False),
False),
None, ),
461 fix_spec(all_structs)