chalkpy 2.93.5__py3-none-any.whl → 2.93.7__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.
- chalk/_gen/chalk/arrow/v1/arrow_pb2.py +7 -5
- chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi +6 -0
- chalk/_gen/chalk/common/v1/offline_query_pb2.py +17 -15
- chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +25 -0
- chalk/_gen/chalk/common/v1/script_task_pb2.py +3 -3
- chalk/_gen/chalk/common/v1/script_task_pb2.pyi +2 -0
- chalk/_gen/chalk/dataframe/__init__.py +0 -0
- chalk/_gen/chalk/dataframe/v1/__init__.py +0 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py +48 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi +119 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py +4 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/graph/v1/graph_pb2.py +150 -149
- chalk/_gen/chalk/graph/v1/graph_pb2.pyi +5 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.py +79 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.pyi +377 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.py +4 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.py +43 -7
- chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi +252 -2
- chalk/_gen/chalk/server/v1/benchmark_pb2.py +45 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +74 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +78 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +36 -0
- chalk/_gen/chalk/server/v1/builder_pb2.py +218 -210
- chalk/_gen/chalk/server/v1/builder_pb2.pyi +59 -1
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +70 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +219 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py +170 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi +62 -0
- chalk/_gen/chalk/server/v1/files_pb2.py +65 -0
- chalk/_gen/chalk/server/v1/files_pb2.pyi +167 -0
- chalk/_gen/chalk/server/v1/files_pb2_grpc.py +4 -0
- chalk/_gen/chalk/server/v1/files_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/server/v1/kube_pb2.py +29 -19
- chalk/_gen/chalk/server/v1/kube_pb2.pyi +28 -0
- chalk/_gen/chalk/server/v1/kube_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/kube_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/queries_pb2.py +66 -66
- chalk/_gen/chalk/server/v1/queries_pb2.pyi +20 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2.py +75 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi +142 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py +349 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi +114 -0
- chalk/_gen/chalk/server/v1/trace_pb2.py +44 -40
- chalk/_gen/chalk/server/v1/trace_pb2.pyi +12 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2.py +62 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi +75 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.py +221 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi +88 -0
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +8 -8
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +4 -1
- chalk/_version.py +1 -1
- chalk/df/LazyFramePlaceholder.py +1129 -0
- chalk/features/_encoding/converter.py +23 -0
- chalk/features/resolver.py +9 -2
- chalk/operators/_utils.py +10 -3
- chalk/parsed/to_proto.py +7 -1
- {chalkpy-2.93.5.dist-info → chalkpy-2.93.7.dist-info}/METADATA +1 -1
- {chalkpy-2.93.5.dist-info → chalkpy-2.93.7.dist-info}/RECORD +63 -32
- {chalkpy-2.93.5.dist-info → chalkpy-2.93.7.dist-info}/WHEEL +0 -0
- {chalkpy-2.93.5.dist-info → chalkpy-2.93.7.dist-info}/entry_points.txt +0 -0
- {chalkpy-2.93.5.dist-info → chalkpy-2.93.7.dist-info}/top_level.txt +0 -0
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: chalk/streaming/v1/debug_service.proto
|
|
4
|
+
# Protobuf Python Version: 4.25.3
|
|
5
|
+
"""Generated protocol buffer code."""
|
|
6
|
+
|
|
7
|
+
from google.protobuf import descriptor as _descriptor
|
|
8
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
|
9
|
+
from google.protobuf import symbol_database as _symbol_database
|
|
10
|
+
from google.protobuf.internal import builder as _builder
|
|
11
|
+
# @@protoc_insertion_point(imports)
|
|
12
|
+
|
|
13
|
+
_sym_db = _symbol_database.Default()
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
from chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
|
|
17
|
+
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
|
|
21
|
+
b'\n&chalk/streaming/v1/debug_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1fgoogle/protobuf/timestamp.proto"`\n\x16\x45nableDebugModeRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn\x12#\n\rdeployment_id\x18\x02 \x01(\tR\x0c\x64\x65ploymentId"n\n\x17\x45nableDebugModeResponse\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled\x12\x39\n\nenabled_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tenabledAt"<\n\x17\x44isableDebugModeRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn"4\n\x18\x44isableDebugModeResponse\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled">\n\x19GetDebugModeStatusRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn"\xe8\x01\n\x1aGetDebugModeStatusResponse\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled\x12>\n\nenabled_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tenabledAt\x88\x01\x01\x12%\n\x0estorage_bucket\x18\x03 \x01(\tR\rstorageBucket\x12(\n\rdeployment_id\x18\x04 \x01(\tH\x01R\x0c\x64\x65ploymentId\x88\x01\x01\x42\r\n\x0b_enabled_atB\x10\n\x0e_deployment_id"<\n\x17GetDebugMessagesRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn"Y\n\x18GetDebugMessagesResponse\x12\x18\n\x07parquet\x18\x01 \x01(\x0cR\x07parquet\x12\x19\n\x05\x65rror\x18\x02 \x01(\tH\x00R\x05\x65rror\x88\x01\x01\x42\x08\n\x06_error2\xf0\x03\n\x15StreamingDebugService\x12o\n\x0f\x45nableDebugMode\x12*.chalk.streaming.v1.EnableDebugModeRequest\x1a+.chalk.streaming.v1.EnableDebugModeResponse"\x03\x80}\x0c\x12r\n\x10\x44isableDebugMode\x12+.chalk.streaming.v1.DisableDebugModeRequest\x1a,.chalk.streaming.v1.DisableDebugModeResponse"\x03\x80}\x0c\x12{\n\x12GetDebugModeStatus\x12-.chalk.streaming.v1.GetDebugModeStatusRequest\x1a..chalk.streaming.v1.GetDebugModeStatusResponse"\x06\x90\x02\x01\x80}\x0b\x12u\n\x10GetDebugMessages\x12+.chalk.streaming.v1.GetDebugMessagesRequest\x1a,.chalk.streaming.v1.GetDebugMessagesResponse"\x06\x90\x02\x01\x80}\x0b\x42\x95\x01\n\x16\x63om.chalk.streaming.v1B\x11\x44\x65\x62ugServiceProtoP\x01\xa2\x02\x03\x43SX\xaa\x02\x12\x43halk.Streaming.V1\xca\x02\x12\x43halk\\Streaming\\V1\xe2\x02\x1e\x43halk\\Streaming\\V1\\GPBMetadata\xea\x02\x14\x43halk::Streaming::V1b\x06proto3'
|
|
22
|
+
)
|
|
23
|
+
|
|
24
|
+
_globals = globals()
|
|
25
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
26
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.streaming.v1.debug_service_pb2", _globals)
|
|
27
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
|
28
|
+
_globals["DESCRIPTOR"]._options = None
|
|
29
|
+
_globals[
|
|
30
|
+
"DESCRIPTOR"
|
|
31
|
+
]._serialized_options = b"\n\026com.chalk.streaming.v1B\021DebugServiceProtoP\001\242\002\003CSX\252\002\022Chalk.Streaming.V1\312\002\022Chalk\\Streaming\\V1\342\002\036Chalk\\Streaming\\V1\\GPBMetadata\352\002\024Chalk::Streaming::V1"
|
|
32
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["EnableDebugMode"]._options = None
|
|
33
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["EnableDebugMode"]._serialized_options = b"\200}\014"
|
|
34
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["DisableDebugMode"]._options = None
|
|
35
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["DisableDebugMode"]._serialized_options = b"\200}\014"
|
|
36
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["GetDebugModeStatus"]._options = None
|
|
37
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name[
|
|
38
|
+
"GetDebugModeStatus"
|
|
39
|
+
]._serialized_options = b"\220\002\001\200}\013"
|
|
40
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["GetDebugMessages"]._options = None
|
|
41
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name[
|
|
42
|
+
"GetDebugMessages"
|
|
43
|
+
]._serialized_options = b"\220\002\001\200}\013"
|
|
44
|
+
_globals["_ENABLEDEBUGMODEREQUEST"]._serialized_start = 128
|
|
45
|
+
_globals["_ENABLEDEBUGMODEREQUEST"]._serialized_end = 224
|
|
46
|
+
_globals["_ENABLEDEBUGMODERESPONSE"]._serialized_start = 226
|
|
47
|
+
_globals["_ENABLEDEBUGMODERESPONSE"]._serialized_end = 336
|
|
48
|
+
_globals["_DISABLEDEBUGMODEREQUEST"]._serialized_start = 338
|
|
49
|
+
_globals["_DISABLEDEBUGMODEREQUEST"]._serialized_end = 398
|
|
50
|
+
_globals["_DISABLEDEBUGMODERESPONSE"]._serialized_start = 400
|
|
51
|
+
_globals["_DISABLEDEBUGMODERESPONSE"]._serialized_end = 452
|
|
52
|
+
_globals["_GETDEBUGMODESTATUSREQUEST"]._serialized_start = 454
|
|
53
|
+
_globals["_GETDEBUGMODESTATUSREQUEST"]._serialized_end = 516
|
|
54
|
+
_globals["_GETDEBUGMODESTATUSRESPONSE"]._serialized_start = 519
|
|
55
|
+
_globals["_GETDEBUGMODESTATUSRESPONSE"]._serialized_end = 751
|
|
56
|
+
_globals["_GETDEBUGMESSAGESREQUEST"]._serialized_start = 753
|
|
57
|
+
_globals["_GETDEBUGMESSAGESREQUEST"]._serialized_end = 813
|
|
58
|
+
_globals["_GETDEBUGMESSAGESRESPONSE"]._serialized_start = 815
|
|
59
|
+
_globals["_GETDEBUGMESSAGESRESPONSE"]._serialized_end = 904
|
|
60
|
+
_globals["_STREAMINGDEBUGSERVICE"]._serialized_start = 907
|
|
61
|
+
_globals["_STREAMINGDEBUGSERVICE"]._serialized_end = 1403
|
|
62
|
+
# @@protoc_insertion_point(module_scope)
|
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
|
|
2
|
+
from google.protobuf import timestamp_pb2 as _timestamp_pb2
|
|
3
|
+
from google.protobuf import descriptor as _descriptor
|
|
4
|
+
from google.protobuf import message as _message
|
|
5
|
+
from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
|
|
6
|
+
|
|
7
|
+
DESCRIPTOR: _descriptor.FileDescriptor
|
|
8
|
+
|
|
9
|
+
class EnableDebugModeRequest(_message.Message):
|
|
10
|
+
__slots__ = ("resolver_fqn", "deployment_id")
|
|
11
|
+
RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
|
|
12
|
+
DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
|
|
13
|
+
resolver_fqn: str
|
|
14
|
+
deployment_id: str
|
|
15
|
+
def __init__(self, resolver_fqn: _Optional[str] = ..., deployment_id: _Optional[str] = ...) -> None: ...
|
|
16
|
+
|
|
17
|
+
class EnableDebugModeResponse(_message.Message):
|
|
18
|
+
__slots__ = ("enabled", "enabled_at")
|
|
19
|
+
ENABLED_FIELD_NUMBER: _ClassVar[int]
|
|
20
|
+
ENABLED_AT_FIELD_NUMBER: _ClassVar[int]
|
|
21
|
+
enabled: bool
|
|
22
|
+
enabled_at: _timestamp_pb2.Timestamp
|
|
23
|
+
def __init__(
|
|
24
|
+
self, enabled: bool = ..., enabled_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...
|
|
25
|
+
) -> None: ...
|
|
26
|
+
|
|
27
|
+
class DisableDebugModeRequest(_message.Message):
|
|
28
|
+
__slots__ = ("resolver_fqn",)
|
|
29
|
+
RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
|
|
30
|
+
resolver_fqn: str
|
|
31
|
+
def __init__(self, resolver_fqn: _Optional[str] = ...) -> None: ...
|
|
32
|
+
|
|
33
|
+
class DisableDebugModeResponse(_message.Message):
|
|
34
|
+
__slots__ = ("enabled",)
|
|
35
|
+
ENABLED_FIELD_NUMBER: _ClassVar[int]
|
|
36
|
+
enabled: bool
|
|
37
|
+
def __init__(self, enabled: bool = ...) -> None: ...
|
|
38
|
+
|
|
39
|
+
class GetDebugModeStatusRequest(_message.Message):
|
|
40
|
+
__slots__ = ("resolver_fqn",)
|
|
41
|
+
RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
|
|
42
|
+
resolver_fqn: str
|
|
43
|
+
def __init__(self, resolver_fqn: _Optional[str] = ...) -> None: ...
|
|
44
|
+
|
|
45
|
+
class GetDebugModeStatusResponse(_message.Message):
|
|
46
|
+
__slots__ = ("enabled", "enabled_at", "storage_bucket", "deployment_id")
|
|
47
|
+
ENABLED_FIELD_NUMBER: _ClassVar[int]
|
|
48
|
+
ENABLED_AT_FIELD_NUMBER: _ClassVar[int]
|
|
49
|
+
STORAGE_BUCKET_FIELD_NUMBER: _ClassVar[int]
|
|
50
|
+
DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
|
|
51
|
+
enabled: bool
|
|
52
|
+
enabled_at: _timestamp_pb2.Timestamp
|
|
53
|
+
storage_bucket: str
|
|
54
|
+
deployment_id: str
|
|
55
|
+
def __init__(
|
|
56
|
+
self,
|
|
57
|
+
enabled: bool = ...,
|
|
58
|
+
enabled_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
|
|
59
|
+
storage_bucket: _Optional[str] = ...,
|
|
60
|
+
deployment_id: _Optional[str] = ...,
|
|
61
|
+
) -> None: ...
|
|
62
|
+
|
|
63
|
+
class GetDebugMessagesRequest(_message.Message):
|
|
64
|
+
__slots__ = ("resolver_fqn",)
|
|
65
|
+
RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
|
|
66
|
+
resolver_fqn: str
|
|
67
|
+
def __init__(self, resolver_fqn: _Optional[str] = ...) -> None: ...
|
|
68
|
+
|
|
69
|
+
class GetDebugMessagesResponse(_message.Message):
|
|
70
|
+
__slots__ = ("parquet", "error")
|
|
71
|
+
PARQUET_FIELD_NUMBER: _ClassVar[int]
|
|
72
|
+
ERROR_FIELD_NUMBER: _ClassVar[int]
|
|
73
|
+
parquet: bytes
|
|
74
|
+
error: str
|
|
75
|
+
def __init__(self, parquet: _Optional[bytes] = ..., error: _Optional[str] = ...) -> None: ...
|
|
@@ -0,0 +1,221 @@
|
|
|
1
|
+
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
|
|
2
|
+
"""Client and server classes corresponding to protobuf-defined services."""
|
|
3
|
+
|
|
4
|
+
import grpc
|
|
5
|
+
|
|
6
|
+
from chalk._gen.chalk.streaming.v1 import debug_service_pb2 as chalk_dot_streaming_dot_v1_dot_debug__service__pb2
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
class StreamingDebugServiceStub(object):
|
|
10
|
+
"""Streaming Debug Service
|
|
11
|
+
Provides APIs for managing and querying streaming resolver debug mode
|
|
12
|
+
"""
|
|
13
|
+
|
|
14
|
+
def __init__(self, channel):
|
|
15
|
+
"""Constructor.
|
|
16
|
+
|
|
17
|
+
Args:
|
|
18
|
+
channel: A grpc.Channel.
|
|
19
|
+
"""
|
|
20
|
+
self.EnableDebugMode = channel.unary_unary(
|
|
21
|
+
"/chalk.streaming.v1.StreamingDebugService/EnableDebugMode",
|
|
22
|
+
request_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeRequest.SerializeToString,
|
|
23
|
+
response_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeResponse.FromString,
|
|
24
|
+
)
|
|
25
|
+
self.DisableDebugMode = channel.unary_unary(
|
|
26
|
+
"/chalk.streaming.v1.StreamingDebugService/DisableDebugMode",
|
|
27
|
+
request_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeRequest.SerializeToString,
|
|
28
|
+
response_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeResponse.FromString,
|
|
29
|
+
)
|
|
30
|
+
self.GetDebugModeStatus = channel.unary_unary(
|
|
31
|
+
"/chalk.streaming.v1.StreamingDebugService/GetDebugModeStatus",
|
|
32
|
+
request_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusRequest.SerializeToString,
|
|
33
|
+
response_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusResponse.FromString,
|
|
34
|
+
)
|
|
35
|
+
self.GetDebugMessages = channel.unary_unary(
|
|
36
|
+
"/chalk.streaming.v1.StreamingDebugService/GetDebugMessages",
|
|
37
|
+
request_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesRequest.SerializeToString,
|
|
38
|
+
response_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesResponse.FromString,
|
|
39
|
+
)
|
|
40
|
+
|
|
41
|
+
|
|
42
|
+
class StreamingDebugServiceServicer(object):
|
|
43
|
+
"""Streaming Debug Service
|
|
44
|
+
Provides APIs for managing and querying streaming resolver debug mode
|
|
45
|
+
"""
|
|
46
|
+
|
|
47
|
+
def EnableDebugMode(self, request, context):
|
|
48
|
+
"""Enable debug mode for a streaming resolver"""
|
|
49
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
50
|
+
context.set_details("Method not implemented!")
|
|
51
|
+
raise NotImplementedError("Method not implemented!")
|
|
52
|
+
|
|
53
|
+
def DisableDebugMode(self, request, context):
|
|
54
|
+
"""Disable debug mode for a streaming resolver"""
|
|
55
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
56
|
+
context.set_details("Method not implemented!")
|
|
57
|
+
raise NotImplementedError("Method not implemented!")
|
|
58
|
+
|
|
59
|
+
def GetDebugModeStatus(self, request, context):
|
|
60
|
+
"""Get the current debug mode status for a resolver"""
|
|
61
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
62
|
+
context.set_details("Method not implemented!")
|
|
63
|
+
raise NotImplementedError("Method not implemented!")
|
|
64
|
+
|
|
65
|
+
def GetDebugMessages(self, request, context):
|
|
66
|
+
"""Get recent debug messages as parquet"""
|
|
67
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
68
|
+
context.set_details("Method not implemented!")
|
|
69
|
+
raise NotImplementedError("Method not implemented!")
|
|
70
|
+
|
|
71
|
+
|
|
72
|
+
def add_StreamingDebugServiceServicer_to_server(servicer, server):
|
|
73
|
+
rpc_method_handlers = {
|
|
74
|
+
"EnableDebugMode": grpc.unary_unary_rpc_method_handler(
|
|
75
|
+
servicer.EnableDebugMode,
|
|
76
|
+
request_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeRequest.FromString,
|
|
77
|
+
response_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeResponse.SerializeToString,
|
|
78
|
+
),
|
|
79
|
+
"DisableDebugMode": grpc.unary_unary_rpc_method_handler(
|
|
80
|
+
servicer.DisableDebugMode,
|
|
81
|
+
request_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeRequest.FromString,
|
|
82
|
+
response_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeResponse.SerializeToString,
|
|
83
|
+
),
|
|
84
|
+
"GetDebugModeStatus": grpc.unary_unary_rpc_method_handler(
|
|
85
|
+
servicer.GetDebugModeStatus,
|
|
86
|
+
request_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusRequest.FromString,
|
|
87
|
+
response_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusResponse.SerializeToString,
|
|
88
|
+
),
|
|
89
|
+
"GetDebugMessages": grpc.unary_unary_rpc_method_handler(
|
|
90
|
+
servicer.GetDebugMessages,
|
|
91
|
+
request_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesRequest.FromString,
|
|
92
|
+
response_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesResponse.SerializeToString,
|
|
93
|
+
),
|
|
94
|
+
}
|
|
95
|
+
generic_handler = grpc.method_handlers_generic_handler(
|
|
96
|
+
"chalk.streaming.v1.StreamingDebugService", rpc_method_handlers
|
|
97
|
+
)
|
|
98
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
|
99
|
+
|
|
100
|
+
|
|
101
|
+
# This class is part of an EXPERIMENTAL API.
|
|
102
|
+
class StreamingDebugService(object):
|
|
103
|
+
"""Streaming Debug Service
|
|
104
|
+
Provides APIs for managing and querying streaming resolver debug mode
|
|
105
|
+
"""
|
|
106
|
+
|
|
107
|
+
@staticmethod
|
|
108
|
+
def EnableDebugMode(
|
|
109
|
+
request,
|
|
110
|
+
target,
|
|
111
|
+
options=(),
|
|
112
|
+
channel_credentials=None,
|
|
113
|
+
call_credentials=None,
|
|
114
|
+
insecure=False,
|
|
115
|
+
compression=None,
|
|
116
|
+
wait_for_ready=None,
|
|
117
|
+
timeout=None,
|
|
118
|
+
metadata=None,
|
|
119
|
+
):
|
|
120
|
+
return grpc.experimental.unary_unary(
|
|
121
|
+
request,
|
|
122
|
+
target,
|
|
123
|
+
"/chalk.streaming.v1.StreamingDebugService/EnableDebugMode",
|
|
124
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeRequest.SerializeToString,
|
|
125
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeResponse.FromString,
|
|
126
|
+
options,
|
|
127
|
+
channel_credentials,
|
|
128
|
+
insecure,
|
|
129
|
+
call_credentials,
|
|
130
|
+
compression,
|
|
131
|
+
wait_for_ready,
|
|
132
|
+
timeout,
|
|
133
|
+
metadata,
|
|
134
|
+
)
|
|
135
|
+
|
|
136
|
+
@staticmethod
|
|
137
|
+
def DisableDebugMode(
|
|
138
|
+
request,
|
|
139
|
+
target,
|
|
140
|
+
options=(),
|
|
141
|
+
channel_credentials=None,
|
|
142
|
+
call_credentials=None,
|
|
143
|
+
insecure=False,
|
|
144
|
+
compression=None,
|
|
145
|
+
wait_for_ready=None,
|
|
146
|
+
timeout=None,
|
|
147
|
+
metadata=None,
|
|
148
|
+
):
|
|
149
|
+
return grpc.experimental.unary_unary(
|
|
150
|
+
request,
|
|
151
|
+
target,
|
|
152
|
+
"/chalk.streaming.v1.StreamingDebugService/DisableDebugMode",
|
|
153
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeRequest.SerializeToString,
|
|
154
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeResponse.FromString,
|
|
155
|
+
options,
|
|
156
|
+
channel_credentials,
|
|
157
|
+
insecure,
|
|
158
|
+
call_credentials,
|
|
159
|
+
compression,
|
|
160
|
+
wait_for_ready,
|
|
161
|
+
timeout,
|
|
162
|
+
metadata,
|
|
163
|
+
)
|
|
164
|
+
|
|
165
|
+
@staticmethod
|
|
166
|
+
def GetDebugModeStatus(
|
|
167
|
+
request,
|
|
168
|
+
target,
|
|
169
|
+
options=(),
|
|
170
|
+
channel_credentials=None,
|
|
171
|
+
call_credentials=None,
|
|
172
|
+
insecure=False,
|
|
173
|
+
compression=None,
|
|
174
|
+
wait_for_ready=None,
|
|
175
|
+
timeout=None,
|
|
176
|
+
metadata=None,
|
|
177
|
+
):
|
|
178
|
+
return grpc.experimental.unary_unary(
|
|
179
|
+
request,
|
|
180
|
+
target,
|
|
181
|
+
"/chalk.streaming.v1.StreamingDebugService/GetDebugModeStatus",
|
|
182
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusRequest.SerializeToString,
|
|
183
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusResponse.FromString,
|
|
184
|
+
options,
|
|
185
|
+
channel_credentials,
|
|
186
|
+
insecure,
|
|
187
|
+
call_credentials,
|
|
188
|
+
compression,
|
|
189
|
+
wait_for_ready,
|
|
190
|
+
timeout,
|
|
191
|
+
metadata,
|
|
192
|
+
)
|
|
193
|
+
|
|
194
|
+
@staticmethod
|
|
195
|
+
def GetDebugMessages(
|
|
196
|
+
request,
|
|
197
|
+
target,
|
|
198
|
+
options=(),
|
|
199
|
+
channel_credentials=None,
|
|
200
|
+
call_credentials=None,
|
|
201
|
+
insecure=False,
|
|
202
|
+
compression=None,
|
|
203
|
+
wait_for_ready=None,
|
|
204
|
+
timeout=None,
|
|
205
|
+
metadata=None,
|
|
206
|
+
):
|
|
207
|
+
return grpc.experimental.unary_unary(
|
|
208
|
+
request,
|
|
209
|
+
target,
|
|
210
|
+
"/chalk.streaming.v1.StreamingDebugService/GetDebugMessages",
|
|
211
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesRequest.SerializeToString,
|
|
212
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesResponse.FromString,
|
|
213
|
+
options,
|
|
214
|
+
channel_credentials,
|
|
215
|
+
insecure,
|
|
216
|
+
call_credentials,
|
|
217
|
+
compression,
|
|
218
|
+
wait_for_ready,
|
|
219
|
+
timeout,
|
|
220
|
+
metadata,
|
|
221
|
+
)
|
|
@@ -0,0 +1,88 @@
|
|
|
1
|
+
"""
|
|
2
|
+
@generated by mypy-protobuf. Do not edit manually!
|
|
3
|
+
isort:skip_file
|
|
4
|
+
"""
|
|
5
|
+
|
|
6
|
+
from abc import (
|
|
7
|
+
ABCMeta,
|
|
8
|
+
abstractmethod,
|
|
9
|
+
)
|
|
10
|
+
from chalk._gen.chalk.streaming.v1.debug_service_pb2 import (
|
|
11
|
+
DisableDebugModeRequest,
|
|
12
|
+
DisableDebugModeResponse,
|
|
13
|
+
EnableDebugModeRequest,
|
|
14
|
+
EnableDebugModeResponse,
|
|
15
|
+
GetDebugMessagesRequest,
|
|
16
|
+
GetDebugMessagesResponse,
|
|
17
|
+
GetDebugModeStatusRequest,
|
|
18
|
+
GetDebugModeStatusResponse,
|
|
19
|
+
)
|
|
20
|
+
from grpc import (
|
|
21
|
+
Channel,
|
|
22
|
+
Server,
|
|
23
|
+
ServicerContext,
|
|
24
|
+
UnaryUnaryMultiCallable,
|
|
25
|
+
)
|
|
26
|
+
|
|
27
|
+
class StreamingDebugServiceStub:
|
|
28
|
+
"""Streaming Debug Service
|
|
29
|
+
Provides APIs for managing and querying streaming resolver debug mode
|
|
30
|
+
"""
|
|
31
|
+
|
|
32
|
+
def __init__(self, channel: Channel) -> None: ...
|
|
33
|
+
EnableDebugMode: UnaryUnaryMultiCallable[
|
|
34
|
+
EnableDebugModeRequest,
|
|
35
|
+
EnableDebugModeResponse,
|
|
36
|
+
]
|
|
37
|
+
"""Enable debug mode for a streaming resolver"""
|
|
38
|
+
DisableDebugMode: UnaryUnaryMultiCallable[
|
|
39
|
+
DisableDebugModeRequest,
|
|
40
|
+
DisableDebugModeResponse,
|
|
41
|
+
]
|
|
42
|
+
"""Disable debug mode for a streaming resolver"""
|
|
43
|
+
GetDebugModeStatus: UnaryUnaryMultiCallable[
|
|
44
|
+
GetDebugModeStatusRequest,
|
|
45
|
+
GetDebugModeStatusResponse,
|
|
46
|
+
]
|
|
47
|
+
"""Get the current debug mode status for a resolver"""
|
|
48
|
+
GetDebugMessages: UnaryUnaryMultiCallable[
|
|
49
|
+
GetDebugMessagesRequest,
|
|
50
|
+
GetDebugMessagesResponse,
|
|
51
|
+
]
|
|
52
|
+
"""Get recent debug messages as parquet"""
|
|
53
|
+
|
|
54
|
+
class StreamingDebugServiceServicer(metaclass=ABCMeta):
|
|
55
|
+
"""Streaming Debug Service
|
|
56
|
+
Provides APIs for managing and querying streaming resolver debug mode
|
|
57
|
+
"""
|
|
58
|
+
|
|
59
|
+
@abstractmethod
|
|
60
|
+
def EnableDebugMode(
|
|
61
|
+
self,
|
|
62
|
+
request: EnableDebugModeRequest,
|
|
63
|
+
context: ServicerContext,
|
|
64
|
+
) -> EnableDebugModeResponse:
|
|
65
|
+
"""Enable debug mode for a streaming resolver"""
|
|
66
|
+
@abstractmethod
|
|
67
|
+
def DisableDebugMode(
|
|
68
|
+
self,
|
|
69
|
+
request: DisableDebugModeRequest,
|
|
70
|
+
context: ServicerContext,
|
|
71
|
+
) -> DisableDebugModeResponse:
|
|
72
|
+
"""Disable debug mode for a streaming resolver"""
|
|
73
|
+
@abstractmethod
|
|
74
|
+
def GetDebugModeStatus(
|
|
75
|
+
self,
|
|
76
|
+
request: GetDebugModeStatusRequest,
|
|
77
|
+
context: ServicerContext,
|
|
78
|
+
) -> GetDebugModeStatusResponse:
|
|
79
|
+
"""Get the current debug mode status for a resolver"""
|
|
80
|
+
@abstractmethod
|
|
81
|
+
def GetDebugMessages(
|
|
82
|
+
self,
|
|
83
|
+
request: GetDebugMessagesRequest,
|
|
84
|
+
context: ServicerContext,
|
|
85
|
+
) -> GetDebugMessagesResponse:
|
|
86
|
+
"""Get recent debug messages as parquet"""
|
|
87
|
+
|
|
88
|
+
def add_StreamingDebugServiceServicer_to_server(servicer: StreamingDebugServiceServicer, server: Server) -> None: ...
|
|
@@ -18,7 +18,7 @@ from chalk._gen.chalk.common.v1 import chalk_error_pb2 as chalk_dot_common_dot_v
|
|
|
18
18
|
|
|
19
19
|
|
|
20
20
|
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
|
|
21
|
-
b'\n1chalk/streaming/v1/simple_streaming_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto"\xa0\x01\n\x0eStreamingError\x12\x31\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x05\x65rror\x12\x38\n\x05phase\x18\x02 \x01(\x0e\x32".chalk.streaming.v1.ExecutionPhaseR\x05phase\x12!\n\x0coperation_id\x18\x03 \x01(\tR\x0boperationId"\
|
|
21
|
+
b'\n1chalk/streaming/v1/simple_streaming_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto"\xa0\x01\n\x0eStreamingError\x12\x31\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x05\x65rror\x12\x38\n\x05phase\x18\x02 \x01(\x0e\x32".chalk.streaming.v1.ExecutionPhaseR\x05phase\x12!\n\x0coperation_id\x18\x03 \x01(\tR\x0boperationId"\xd6\x01\n!SimpleStreamingUnaryInvokeRequest\x12\x34\n\x16streaming_resolver_fqn\x18\x01 \x01(\tR\x14streamingResolverFqn\x12\x1d\n\ninput_data\x18\x02 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x03 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x04 \x01(\x08H\x01R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_operation_idB\x08\n\x06_debug"\xe7\x02\n"SimpleStreamingUnaryInvokeResponse\x12(\n\x10num_rows_succeed\x18\x01 \x01(\x03R\x0enumRowsSucceed\x12&\n\x0fnum_rows_failed\x18\x02 \x01(\x03R\rnumRowsFailed\x12(\n\x10num_rows_skipped\x18\x03 \x01(\x03R\x0enumRowsSkipped\x12\x36\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorH\x00R\x05\x65rror\x88\x01\x01\x12$\n\x0boutput_data\x18\x05 \x01(\x0cH\x01R\noutputData\x88\x01\x01\x12M\n\x10\x65xecution_errors\x18\x06 \x03(\x0b\x32".chalk.streaming.v1.StreamingErrorR\x0f\x65xecutionErrorsB\x08\n\x06_errorB\x0e\n\x0c_output_data*\xc5\x01\n\x0e\x45xecutionPhase\x12\x1f\n\x1b\x45XECUTION_PHASE_UNSPECIFIED\x10\x00\x12\x19\n\x15\x45XECUTION_PHASE_PARSE\x10\x01\x12\x1b\n\x17\x45XECUTION_PHASE_MAPPING\x10\x02\x12\x1f\n\x1b\x45XECUTION_PHASE_PERSISTENCE\x10\x03\x12\x1f\n\x1b\x45XECUTION_PHASE_AGGREGATION\x10\x04\x12\x18\n\x14\x45XECUTION_PHASE_SINK\x10\x05\x32\xab\x01\n\x16SimpleStreamingService\x12\x90\x01\n\x1aSimpleStreamingUnaryInvoke\x12\x35.chalk.streaming.v1.SimpleStreamingUnaryInvokeRequest\x1a\x36.chalk.streaming.v1.SimpleStreamingUnaryInvokeResponse"\x03\x80}\x02\x42\x9f\x01\n\x16\x63om.chalk.streaming.v1B\x1bSimpleStreamingServiceProtoP\x01\xa2\x02\x03\x43SX\xaa\x02\x12\x43halk.Streaming.V1\xca\x02\x12\x43halk\\Streaming\\V1\xe2\x02\x1e\x43halk\\Streaming\\V1\\GPBMetadata\xea\x02\x14\x43halk::Streaming::V1b\x06proto3'
|
|
22
22
|
)
|
|
23
23
|
|
|
24
24
|
_globals = globals()
|
|
@@ -31,14 +31,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
31
31
|
]._serialized_options = b"\n\026com.chalk.streaming.v1B\033SimpleStreamingServiceProtoP\001\242\002\003CSX\252\002\022Chalk.Streaming.V1\312\002\022Chalk\\Streaming\\V1\342\002\036Chalk\\Streaming\\V1\\GPBMetadata\352\002\024Chalk::Streaming::V1"
|
|
32
32
|
_globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["SimpleStreamingUnaryInvoke"]._options = None
|
|
33
33
|
_globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["SimpleStreamingUnaryInvoke"]._serialized_options = b"\200}\002"
|
|
34
|
-
_globals["_EXECUTIONPHASE"]._serialized_start =
|
|
35
|
-
_globals["_EXECUTIONPHASE"]._serialized_end =
|
|
34
|
+
_globals["_EXECUTIONPHASE"]._serialized_start = 884
|
|
35
|
+
_globals["_EXECUTIONPHASE"]._serialized_end = 1081
|
|
36
36
|
_globals["_STREAMINGERROR"]._serialized_start = 142
|
|
37
37
|
_globals["_STREAMINGERROR"]._serialized_end = 302
|
|
38
38
|
_globals["_SIMPLESTREAMINGUNARYINVOKEREQUEST"]._serialized_start = 305
|
|
39
|
-
_globals["_SIMPLESTREAMINGUNARYINVOKEREQUEST"]._serialized_end =
|
|
40
|
-
_globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_start =
|
|
41
|
-
_globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_end =
|
|
42
|
-
_globals["_SIMPLESTREAMINGSERVICE"]._serialized_start =
|
|
43
|
-
_globals["_SIMPLESTREAMINGSERVICE"]._serialized_end =
|
|
39
|
+
_globals["_SIMPLESTREAMINGUNARYINVOKEREQUEST"]._serialized_end = 519
|
|
40
|
+
_globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_start = 522
|
|
41
|
+
_globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_end = 881
|
|
42
|
+
_globals["_SIMPLESTREAMINGSERVICE"]._serialized_start = 1084
|
|
43
|
+
_globals["_SIMPLESTREAMINGSERVICE"]._serialized_end = 1255
|
|
44
44
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -46,18 +46,21 @@ class StreamingError(_message.Message):
|
|
|
46
46
|
) -> None: ...
|
|
47
47
|
|
|
48
48
|
class SimpleStreamingUnaryInvokeRequest(_message.Message):
|
|
49
|
-
__slots__ = ("streaming_resolver_fqn", "input_data", "operation_id")
|
|
49
|
+
__slots__ = ("streaming_resolver_fqn", "input_data", "operation_id", "debug")
|
|
50
50
|
STREAMING_RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
|
|
51
51
|
INPUT_DATA_FIELD_NUMBER: _ClassVar[int]
|
|
52
52
|
OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
53
|
+
DEBUG_FIELD_NUMBER: _ClassVar[int]
|
|
53
54
|
streaming_resolver_fqn: str
|
|
54
55
|
input_data: bytes
|
|
55
56
|
operation_id: str
|
|
57
|
+
debug: bool
|
|
56
58
|
def __init__(
|
|
57
59
|
self,
|
|
58
60
|
streaming_resolver_fqn: _Optional[str] = ...,
|
|
59
61
|
input_data: _Optional[bytes] = ...,
|
|
60
62
|
operation_id: _Optional[str] = ...,
|
|
63
|
+
debug: bool = ...,
|
|
61
64
|
) -> None: ...
|
|
62
65
|
|
|
63
66
|
class SimpleStreamingUnaryInvokeResponse(_message.Message):
|
chalk/_version.py
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
__version__ = "2.93.
|
|
1
|
+
__version__ = "2.93.7"
|