chalkpy 2.93.5__py3-none-any.whl → 2.93.6__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.
Files changed (63) hide show
  1. chalk/_gen/chalk/arrow/v1/arrow_pb2.py +7 -5
  2. chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi +6 -0
  3. chalk/_gen/chalk/common/v1/offline_query_pb2.py +17 -15
  4. chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +25 -0
  5. chalk/_gen/chalk/common/v1/script_task_pb2.py +3 -3
  6. chalk/_gen/chalk/common/v1/script_task_pb2.pyi +2 -0
  7. chalk/_gen/chalk/dataframe/__init__.py +0 -0
  8. chalk/_gen/chalk/dataframe/v1/__init__.py +0 -0
  9. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py +48 -0
  10. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi +119 -0
  11. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py +4 -0
  12. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi +4 -0
  13. chalk/_gen/chalk/graph/v1/graph_pb2.py +150 -149
  14. chalk/_gen/chalk/graph/v1/graph_pb2.pyi +5 -0
  15. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.py +79 -0
  16. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.pyi +377 -0
  17. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.py +4 -0
  18. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.pyi +4 -0
  19. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.py +43 -7
  20. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi +252 -2
  21. chalk/_gen/chalk/server/v1/benchmark_pb2.py +45 -0
  22. chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +74 -0
  23. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +78 -0
  24. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +36 -0
  25. chalk/_gen/chalk/server/v1/builder_pb2.py +218 -210
  26. chalk/_gen/chalk/server/v1/builder_pb2.pyi +59 -1
  27. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +70 -0
  28. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +219 -0
  29. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py +170 -0
  30. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi +62 -0
  31. chalk/_gen/chalk/server/v1/files_pb2.py +65 -0
  32. chalk/_gen/chalk/server/v1/files_pb2.pyi +167 -0
  33. chalk/_gen/chalk/server/v1/files_pb2_grpc.py +4 -0
  34. chalk/_gen/chalk/server/v1/files_pb2_grpc.pyi +4 -0
  35. chalk/_gen/chalk/server/v1/kube_pb2.py +29 -19
  36. chalk/_gen/chalk/server/v1/kube_pb2.pyi +28 -0
  37. chalk/_gen/chalk/server/v1/kube_pb2_grpc.py +45 -0
  38. chalk/_gen/chalk/server/v1/kube_pb2_grpc.pyi +12 -0
  39. chalk/_gen/chalk/server/v1/queries_pb2.py +66 -66
  40. chalk/_gen/chalk/server/v1/queries_pb2.pyi +20 -0
  41. chalk/_gen/chalk/server/v1/sql_interface_pb2.py +75 -0
  42. chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi +142 -0
  43. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py +349 -0
  44. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi +114 -0
  45. chalk/_gen/chalk/server/v1/trace_pb2.py +44 -40
  46. chalk/_gen/chalk/server/v1/trace_pb2.pyi +12 -0
  47. chalk/_gen/chalk/streaming/v1/debug_service_pb2.py +62 -0
  48. chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi +75 -0
  49. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.py +221 -0
  50. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi +88 -0
  51. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +8 -8
  52. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +4 -1
  53. chalk/_version.py +1 -1
  54. chalk/df/LazyFramePlaceholder.py +1125 -0
  55. chalk/features/_encoding/converter.py +23 -0
  56. chalk/features/resolver.py +9 -2
  57. chalk/operators/_utils.py +10 -3
  58. chalk/parsed/to_proto.py +7 -1
  59. {chalkpy-2.93.5.dist-info → chalkpy-2.93.6.dist-info}/METADATA +1 -1
  60. {chalkpy-2.93.5.dist-info → chalkpy-2.93.6.dist-info}/RECORD +63 -32
  61. {chalkpy-2.93.5.dist-info → chalkpy-2.93.6.dist-info}/WHEEL +0 -0
  62. {chalkpy-2.93.5.dist-info → chalkpy-2.93.6.dist-info}/entry_points.txt +0 -0
  63. {chalkpy-2.93.5.dist-info → chalkpy-2.93.6.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"\xb1\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\x42\x0f\n\r_operation_id"\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'
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 = 847
35
- _globals["_EXECUTIONPHASE"]._serialized_end = 1044
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 = 482
40
- _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_start = 485
41
- _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_end = 844
42
- _globals["_SIMPLESTREAMINGSERVICE"]._serialized_start = 1047
43
- _globals["_SIMPLESTREAMINGSERVICE"]._serialized_end = 1218
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.5"
1
+ __version__ = "2.93.6"