chalkpy 2.90.1__py3-none-any.whl → 2.95.3__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/__init__.py +2 -1
- chalk/_gen/chalk/arrow/v1/arrow_pb2.py +7 -5
- chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi +6 -0
- chalk/_gen/chalk/artifacts/v1/chart_pb2.py +16 -16
- chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi +4 -0
- chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py +8 -7
- chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi +5 -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 +123 -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 +25 -0
- chalk/_gen/chalk/graph/v1/sources_pb2.py +94 -84
- chalk/_gen/chalk/graph/v1/sources_pb2.pyi +56 -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/protosql/v1/sql_service_pb2.py +54 -27
- chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +131 -3
- chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py +45 -0
- chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi +14 -0
- chalk/_gen/chalk/python/v1/types_pb2.py +14 -14
- chalk/_gen/chalk/python/v1/types_pb2.pyi +8 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2.py +76 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +156 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +258 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +84 -0
- chalk/_gen/chalk/server/v1/billing_pb2.py +40 -38
- chalk/_gen/chalk/server/v1/billing_pb2.pyi +17 -1
- chalk/_gen/chalk/server/v1/branches_pb2.py +45 -0
- chalk/_gen/chalk/server/v1/branches_pb2.pyi +80 -0
- chalk/_gen/chalk/server/v1/branches_pb2_grpc.pyi +36 -0
- chalk/_gen/chalk/server/v1/builder_pb2.py +358 -288
- chalk/_gen/chalk/server/v1/builder_pb2.pyi +360 -10
- chalk/_gen/chalk/server/v1/builder_pb2_grpc.py +225 -0
- chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi +60 -0
- chalk/_gen/chalk/server/v1/chart_pb2.py +10 -10
- chalk/_gen/chalk/server/v1/chart_pb2.pyi +18 -2
- chalk/_gen/chalk/server/v1/clickhouse_pb2.py +42 -0
- chalk/_gen/chalk/server/v1/clickhouse_pb2.pyi +17 -0
- chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.py +78 -0
- chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.pyi +38 -0
- chalk/_gen/chalk/server/v1/cloud_components_pb2.py +141 -119
- chalk/_gen/chalk/server/v1/cloud_components_pb2.pyi +106 -4
- chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2.py +11 -3
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2.pyi +20 -0
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.py +52 -38
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.pyi +62 -1
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +90 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +264 -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/deploy_pb2.py +9 -3
- chalk/_gen/chalk/server/v1/deploy_pb2.pyi +12 -0
- chalk/_gen/chalk/server/v1/deploy_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/deploy_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/deployment_pb2.py +6 -6
- chalk/_gen/chalk/server/v1/deployment_pb2.pyi +20 -0
- chalk/_gen/chalk/server/v1/environment_pb2.py +14 -12
- chalk/_gen/chalk/server/v1/environment_pb2.pyi +19 -0
- chalk/_gen/chalk/server/v1/eventbus_pb2.py +4 -2
- 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/graph_pb2.py +38 -26
- chalk/_gen/chalk/server/v1/graph_pb2.pyi +58 -0
- chalk/_gen/chalk/server/v1/graph_pb2_grpc.py +47 -0
- chalk/_gen/chalk/server/v1/graph_pb2_grpc.pyi +18 -0
- chalk/_gen/chalk/server/v1/incident_pb2.py +23 -21
- chalk/_gen/chalk/server/v1/incident_pb2.pyi +15 -1
- chalk/_gen/chalk/server/v1/indexing_job_pb2.py +44 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi +38 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.py +78 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.pyi +38 -0
- chalk/_gen/chalk/server/v1/integrations_pb2.py +11 -9
- chalk/_gen/chalk/server/v1/integrations_pb2.pyi +34 -2
- 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/log_pb2.py +21 -3
- chalk/_gen/chalk/server/v1/log_pb2.pyi +68 -0
- chalk/_gen/chalk/server/v1/log_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/log_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/model_registry_pb2.py +10 -10
- chalk/_gen/chalk/server/v1/model_registry_pb2.pyi +4 -1
- chalk/_gen/chalk/server/v1/plandebug_pb2.py +53 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2.pyi +86 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.py +168 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.pyi +60 -0
- chalk/_gen/chalk/server/v1/queries_pb2.py +66 -66
- chalk/_gen/chalk/server/v1/queries_pb2.pyi +32 -2
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +12 -12
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +16 -3
- chalk/_gen/chalk/server/v1/scheduler_pb2.py +24 -12
- chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +61 -1
- chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/script_tasks_pb2.py +15 -3
- chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi +22 -0
- chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.pyi +24 -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/sql_queries_pb2.py +48 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +150 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py +123 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi +52 -0
- chalk/_gen/chalk/server/v1/team_pb2.py +154 -141
- chalk/_gen/chalk/server/v1/team_pb2.pyi +30 -2
- chalk/_gen/chalk/server/v1/team_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/team_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/topic_pb2.py +5 -3
- chalk/_gen/chalk/server/v1/topic_pb2.pyi +10 -1
- chalk/_gen/chalk/server/v1/trace_pb2.py +44 -40
- chalk/_gen/chalk/server/v1/trace_pb2.pyi +20 -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 +16 -10
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +52 -1
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py +48 -0
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi +20 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2.py +32 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2.pyi +42 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.py +4 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.pyi +4 -0
- chalk/_lsp/error_builder.py +11 -0
- chalk/_version.py +1 -1
- chalk/client/client.py +128 -43
- chalk/client/client_async.py +149 -0
- chalk/client/client_async_impl.py +22 -0
- chalk/client/client_grpc.py +539 -104
- chalk/client/client_impl.py +449 -122
- chalk/client/dataset.py +7 -1
- chalk/client/models.py +98 -0
- chalk/client/serialization/model_serialization.py +92 -9
- chalk/df/LazyFramePlaceholder.py +1154 -0
- chalk/features/_class_property.py +7 -0
- chalk/features/_embedding/embedding.py +1 -0
- chalk/features/_encoding/converter.py +83 -2
- chalk/features/feature_field.py +40 -30
- chalk/features/feature_set_decorator.py +1 -0
- chalk/features/feature_wrapper.py +42 -3
- chalk/features/hooks.py +81 -10
- chalk/features/inference.py +33 -31
- chalk/features/resolver.py +224 -24
- chalk/functions/__init__.py +65 -3
- chalk/gitignore/gitignore_parser.py +5 -1
- chalk/importer.py +142 -68
- chalk/ml/__init__.py +2 -0
- chalk/ml/model_hooks.py +194 -26
- chalk/ml/model_reference.py +56 -8
- chalk/ml/model_version.py +24 -15
- chalk/ml/utils.py +20 -17
- chalk/operators/_utils.py +10 -3
- chalk/parsed/_proto/export.py +22 -0
- chalk/parsed/duplicate_input_gql.py +3 -0
- chalk/parsed/json_conversions.py +20 -14
- chalk/parsed/to_proto.py +16 -4
- chalk/parsed/user_types_to_json.py +31 -10
- chalk/parsed/validation_from_registries.py +182 -0
- chalk/queries/named_query.py +16 -6
- chalk/queries/scheduled_query.py +9 -1
- chalk/serialization/parsed_annotation.py +24 -11
- chalk/sql/__init__.py +18 -0
- chalk/sql/_internal/integrations/databricks.py +55 -17
- chalk/sql/_internal/integrations/mssql.py +127 -62
- chalk/sql/_internal/integrations/redshift.py +4 -0
- chalk/sql/_internal/sql_file_resolver.py +53 -9
- chalk/sql/_internal/sql_source.py +35 -2
- chalk/streams/_kafka_source.py +5 -1
- chalk/streams/_windows.py +15 -2
- chalk/utils/_otel_version.py +13 -0
- chalk/utils/async_helpers.py +2 -2
- chalk/utils/missing_dependency.py +5 -4
- chalk/utils/tracing.py +185 -95
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/METADATA +4 -6
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/RECORD +202 -146
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/WHEEL +0 -0
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/entry_points.txt +0 -0
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/top_level.txt +0 -0
|
@@ -0,0 +1,142 @@
|
|
|
1
|
+
from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
|
|
2
|
+
from chalk._gen.chalk.server.v1 import files_pb2 as _files_pb2
|
|
3
|
+
from google.protobuf import timestamp_pb2 as _timestamp_pb2
|
|
4
|
+
from google.protobuf.internal import containers as _containers
|
|
5
|
+
from google.protobuf import descriptor as _descriptor
|
|
6
|
+
from google.protobuf import message as _message
|
|
7
|
+
from typing import (
|
|
8
|
+
ClassVar as _ClassVar,
|
|
9
|
+
Iterable as _Iterable,
|
|
10
|
+
Mapping as _Mapping,
|
|
11
|
+
Optional as _Optional,
|
|
12
|
+
Union as _Union,
|
|
13
|
+
)
|
|
14
|
+
|
|
15
|
+
DESCRIPTOR: _descriptor.FileDescriptor
|
|
16
|
+
|
|
17
|
+
class SqlWorksheet(_message.Message):
|
|
18
|
+
__slots__ = ("file_id", "content", "last_operation_id", "created_at", "updated_at", "file")
|
|
19
|
+
FILE_ID_FIELD_NUMBER: _ClassVar[int]
|
|
20
|
+
CONTENT_FIELD_NUMBER: _ClassVar[int]
|
|
21
|
+
LAST_OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
22
|
+
CREATED_AT_FIELD_NUMBER: _ClassVar[int]
|
|
23
|
+
UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
|
|
24
|
+
FILE_FIELD_NUMBER: _ClassVar[int]
|
|
25
|
+
file_id: str
|
|
26
|
+
content: str
|
|
27
|
+
last_operation_id: str
|
|
28
|
+
created_at: _timestamp_pb2.Timestamp
|
|
29
|
+
updated_at: _timestamp_pb2.Timestamp
|
|
30
|
+
file: _files_pb2.File
|
|
31
|
+
def __init__(
|
|
32
|
+
self,
|
|
33
|
+
file_id: _Optional[str] = ...,
|
|
34
|
+
content: _Optional[str] = ...,
|
|
35
|
+
last_operation_id: _Optional[str] = ...,
|
|
36
|
+
created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
|
|
37
|
+
updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
|
|
38
|
+
file: _Optional[_Union[_files_pb2.File, _Mapping]] = ...,
|
|
39
|
+
) -> None: ...
|
|
40
|
+
|
|
41
|
+
class CreateSqlWorksheetRequest(_message.Message):
|
|
42
|
+
__slots__ = ("name", "environment_id", "content")
|
|
43
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
44
|
+
ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
|
|
45
|
+
CONTENT_FIELD_NUMBER: _ClassVar[int]
|
|
46
|
+
name: str
|
|
47
|
+
environment_id: str
|
|
48
|
+
content: str
|
|
49
|
+
def __init__(
|
|
50
|
+
self, name: _Optional[str] = ..., environment_id: _Optional[str] = ..., content: _Optional[str] = ...
|
|
51
|
+
) -> None: ...
|
|
52
|
+
|
|
53
|
+
class CreateSqlWorksheetResponse(_message.Message):
|
|
54
|
+
__slots__ = ("worksheet",)
|
|
55
|
+
WORKSHEET_FIELD_NUMBER: _ClassVar[int]
|
|
56
|
+
worksheet: SqlWorksheet
|
|
57
|
+
def __init__(self, worksheet: _Optional[_Union[SqlWorksheet, _Mapping]] = ...) -> None: ...
|
|
58
|
+
|
|
59
|
+
class GetSqlWorksheetRequest(_message.Message):
|
|
60
|
+
__slots__ = ("file_id",)
|
|
61
|
+
FILE_ID_FIELD_NUMBER: _ClassVar[int]
|
|
62
|
+
file_id: str
|
|
63
|
+
def __init__(self, file_id: _Optional[str] = ...) -> None: ...
|
|
64
|
+
|
|
65
|
+
class GetSqlWorksheetResponse(_message.Message):
|
|
66
|
+
__slots__ = ("worksheet",)
|
|
67
|
+
WORKSHEET_FIELD_NUMBER: _ClassVar[int]
|
|
68
|
+
worksheet: SqlWorksheet
|
|
69
|
+
def __init__(self, worksheet: _Optional[_Union[SqlWorksheet, _Mapping]] = ...) -> None: ...
|
|
70
|
+
|
|
71
|
+
class UpdateSqlWorksheetRequest(_message.Message):
|
|
72
|
+
__slots__ = ("file_id", "content", "last_operation_id")
|
|
73
|
+
FILE_ID_FIELD_NUMBER: _ClassVar[int]
|
|
74
|
+
CONTENT_FIELD_NUMBER: _ClassVar[int]
|
|
75
|
+
LAST_OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
76
|
+
file_id: str
|
|
77
|
+
content: str
|
|
78
|
+
last_operation_id: str
|
|
79
|
+
def __init__(
|
|
80
|
+
self, file_id: _Optional[str] = ..., content: _Optional[str] = ..., last_operation_id: _Optional[str] = ...
|
|
81
|
+
) -> None: ...
|
|
82
|
+
|
|
83
|
+
class UpdateSqlWorksheetResponse(_message.Message):
|
|
84
|
+
__slots__ = ("worksheet",)
|
|
85
|
+
WORKSHEET_FIELD_NUMBER: _ClassVar[int]
|
|
86
|
+
worksheet: SqlWorksheet
|
|
87
|
+
def __init__(self, worksheet: _Optional[_Union[SqlWorksheet, _Mapping]] = ...) -> None: ...
|
|
88
|
+
|
|
89
|
+
class DeleteSqlWorksheetRequest(_message.Message):
|
|
90
|
+
__slots__ = ("file_id",)
|
|
91
|
+
FILE_ID_FIELD_NUMBER: _ClassVar[int]
|
|
92
|
+
file_id: str
|
|
93
|
+
def __init__(self, file_id: _Optional[str] = ...) -> None: ...
|
|
94
|
+
|
|
95
|
+
class DeleteSqlWorksheetResponse(_message.Message):
|
|
96
|
+
__slots__ = ()
|
|
97
|
+
def __init__(self) -> None: ...
|
|
98
|
+
|
|
99
|
+
class ListSqlWorksheetsRequest(_message.Message):
|
|
100
|
+
__slots__ = ("environment_id", "cursor", "limit", "search", "include_archived")
|
|
101
|
+
ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
|
|
102
|
+
CURSOR_FIELD_NUMBER: _ClassVar[int]
|
|
103
|
+
LIMIT_FIELD_NUMBER: _ClassVar[int]
|
|
104
|
+
SEARCH_FIELD_NUMBER: _ClassVar[int]
|
|
105
|
+
INCLUDE_ARCHIVED_FIELD_NUMBER: _ClassVar[int]
|
|
106
|
+
environment_id: str
|
|
107
|
+
cursor: str
|
|
108
|
+
limit: int
|
|
109
|
+
search: str
|
|
110
|
+
include_archived: bool
|
|
111
|
+
def __init__(
|
|
112
|
+
self,
|
|
113
|
+
environment_id: _Optional[str] = ...,
|
|
114
|
+
cursor: _Optional[str] = ...,
|
|
115
|
+
limit: _Optional[int] = ...,
|
|
116
|
+
search: _Optional[str] = ...,
|
|
117
|
+
include_archived: bool = ...,
|
|
118
|
+
) -> None: ...
|
|
119
|
+
|
|
120
|
+
class ListSqlWorksheetsResponse(_message.Message):
|
|
121
|
+
__slots__ = ("worksheets", "cursor")
|
|
122
|
+
WORKSHEETS_FIELD_NUMBER: _ClassVar[int]
|
|
123
|
+
CURSOR_FIELD_NUMBER: _ClassVar[int]
|
|
124
|
+
worksheets: _containers.RepeatedCompositeFieldContainer[SqlWorksheet]
|
|
125
|
+
cursor: str
|
|
126
|
+
def __init__(
|
|
127
|
+
self, worksheets: _Optional[_Iterable[_Union[SqlWorksheet, _Mapping]]] = ..., cursor: _Optional[str] = ...
|
|
128
|
+
) -> None: ...
|
|
129
|
+
|
|
130
|
+
class RenameSqlWorksheetRequest(_message.Message):
|
|
131
|
+
__slots__ = ("file_id", "name")
|
|
132
|
+
FILE_ID_FIELD_NUMBER: _ClassVar[int]
|
|
133
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
134
|
+
file_id: str
|
|
135
|
+
name: str
|
|
136
|
+
def __init__(self, file_id: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
|
|
137
|
+
|
|
138
|
+
class RenameSqlWorksheetResponse(_message.Message):
|
|
139
|
+
__slots__ = ("worksheet",)
|
|
140
|
+
WORKSHEET_FIELD_NUMBER: _ClassVar[int]
|
|
141
|
+
worksheet: SqlWorksheet
|
|
142
|
+
def __init__(self, worksheet: _Optional[_Union[SqlWorksheet, _Mapping]] = ...) -> None: ...
|
|
@@ -0,0 +1,349 @@
|
|
|
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.server.v1 import files_pb2 as chalk_dot_server_dot_v1_dot_files__pb2
|
|
7
|
+
from chalk._gen.chalk.server.v1 import sql_interface_pb2 as chalk_dot_server_dot_v1_dot_sql__interface__pb2
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
class SqlInterfaceServiceStub(object):
|
|
11
|
+
"""Missing associated documentation comment in .proto file."""
|
|
12
|
+
|
|
13
|
+
def __init__(self, channel):
|
|
14
|
+
"""Constructor.
|
|
15
|
+
|
|
16
|
+
Args:
|
|
17
|
+
channel: A grpc.Channel.
|
|
18
|
+
"""
|
|
19
|
+
self.CreateSqlWorksheet = channel.unary_unary(
|
|
20
|
+
"/chalk.server.v1.SqlInterfaceService/CreateSqlWorksheet",
|
|
21
|
+
request_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.CreateSqlWorksheetRequest.SerializeToString,
|
|
22
|
+
response_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.CreateSqlWorksheetResponse.FromString,
|
|
23
|
+
)
|
|
24
|
+
self.GetSqlWorksheet = channel.unary_unary(
|
|
25
|
+
"/chalk.server.v1.SqlInterfaceService/GetSqlWorksheet",
|
|
26
|
+
request_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.GetSqlWorksheetRequest.SerializeToString,
|
|
27
|
+
response_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.GetSqlWorksheetResponse.FromString,
|
|
28
|
+
)
|
|
29
|
+
self.UpdateSqlWorksheet = channel.unary_unary(
|
|
30
|
+
"/chalk.server.v1.SqlInterfaceService/UpdateSqlWorksheet",
|
|
31
|
+
request_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.UpdateSqlWorksheetRequest.SerializeToString,
|
|
32
|
+
response_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.UpdateSqlWorksheetResponse.FromString,
|
|
33
|
+
)
|
|
34
|
+
self.DeleteSqlWorksheet = channel.unary_unary(
|
|
35
|
+
"/chalk.server.v1.SqlInterfaceService/DeleteSqlWorksheet",
|
|
36
|
+
request_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.DeleteSqlWorksheetRequest.SerializeToString,
|
|
37
|
+
response_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.DeleteSqlWorksheetResponse.FromString,
|
|
38
|
+
)
|
|
39
|
+
self.ListSqlWorksheets = channel.unary_unary(
|
|
40
|
+
"/chalk.server.v1.SqlInterfaceService/ListSqlWorksheets",
|
|
41
|
+
request_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.ListSqlWorksheetsRequest.SerializeToString,
|
|
42
|
+
response_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.ListSqlWorksheetsResponse.FromString,
|
|
43
|
+
)
|
|
44
|
+
self.RenameSqlWorksheet = channel.unary_unary(
|
|
45
|
+
"/chalk.server.v1.SqlInterfaceService/RenameSqlWorksheet",
|
|
46
|
+
request_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.RenameSqlWorksheetRequest.SerializeToString,
|
|
47
|
+
response_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.RenameSqlWorksheetResponse.FromString,
|
|
48
|
+
)
|
|
49
|
+
self.ListFiles = channel.unary_unary(
|
|
50
|
+
"/chalk.server.v1.SqlInterfaceService/ListFiles",
|
|
51
|
+
request_serializer=chalk_dot_server_dot_v1_dot_files__pb2.ListFilesRequest.SerializeToString,
|
|
52
|
+
response_deserializer=chalk_dot_server_dot_v1_dot_files__pb2.ListFilesResponse.FromString,
|
|
53
|
+
)
|
|
54
|
+
|
|
55
|
+
|
|
56
|
+
class SqlInterfaceServiceServicer(object):
|
|
57
|
+
"""Missing associated documentation comment in .proto file."""
|
|
58
|
+
|
|
59
|
+
def CreateSqlWorksheet(self, request, context):
|
|
60
|
+
"""SQL Interface operations for ChalkSQL Explorer"""
|
|
61
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
62
|
+
context.set_details("Method not implemented!")
|
|
63
|
+
raise NotImplementedError("Method not implemented!")
|
|
64
|
+
|
|
65
|
+
def GetSqlWorksheet(self, request, context):
|
|
66
|
+
"""Missing associated documentation comment in .proto file."""
|
|
67
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
68
|
+
context.set_details("Method not implemented!")
|
|
69
|
+
raise NotImplementedError("Method not implemented!")
|
|
70
|
+
|
|
71
|
+
def UpdateSqlWorksheet(self, request, context):
|
|
72
|
+
"""Missing associated documentation comment in .proto file."""
|
|
73
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
74
|
+
context.set_details("Method not implemented!")
|
|
75
|
+
raise NotImplementedError("Method not implemented!")
|
|
76
|
+
|
|
77
|
+
def DeleteSqlWorksheet(self, request, context):
|
|
78
|
+
"""Missing associated documentation comment in .proto file."""
|
|
79
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
80
|
+
context.set_details("Method not implemented!")
|
|
81
|
+
raise NotImplementedError("Method not implemented!")
|
|
82
|
+
|
|
83
|
+
def ListSqlWorksheets(self, request, context):
|
|
84
|
+
"""Missing associated documentation comment in .proto file."""
|
|
85
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
86
|
+
context.set_details("Method not implemented!")
|
|
87
|
+
raise NotImplementedError("Method not implemented!")
|
|
88
|
+
|
|
89
|
+
def RenameSqlWorksheet(self, request, context):
|
|
90
|
+
"""Missing associated documentation comment in .proto file."""
|
|
91
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
92
|
+
context.set_details("Method not implemented!")
|
|
93
|
+
raise NotImplementedError("Method not implemented!")
|
|
94
|
+
|
|
95
|
+
def ListFiles(self, request, context):
|
|
96
|
+
"""File operations - only ListFiles is implemented (files are managed via SQL worksheets)"""
|
|
97
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
98
|
+
context.set_details("Method not implemented!")
|
|
99
|
+
raise NotImplementedError("Method not implemented!")
|
|
100
|
+
|
|
101
|
+
|
|
102
|
+
def add_SqlInterfaceServiceServicer_to_server(servicer, server):
|
|
103
|
+
rpc_method_handlers = {
|
|
104
|
+
"CreateSqlWorksheet": grpc.unary_unary_rpc_method_handler(
|
|
105
|
+
servicer.CreateSqlWorksheet,
|
|
106
|
+
request_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.CreateSqlWorksheetRequest.FromString,
|
|
107
|
+
response_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.CreateSqlWorksheetResponse.SerializeToString,
|
|
108
|
+
),
|
|
109
|
+
"GetSqlWorksheet": grpc.unary_unary_rpc_method_handler(
|
|
110
|
+
servicer.GetSqlWorksheet,
|
|
111
|
+
request_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.GetSqlWorksheetRequest.FromString,
|
|
112
|
+
response_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.GetSqlWorksheetResponse.SerializeToString,
|
|
113
|
+
),
|
|
114
|
+
"UpdateSqlWorksheet": grpc.unary_unary_rpc_method_handler(
|
|
115
|
+
servicer.UpdateSqlWorksheet,
|
|
116
|
+
request_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.UpdateSqlWorksheetRequest.FromString,
|
|
117
|
+
response_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.UpdateSqlWorksheetResponse.SerializeToString,
|
|
118
|
+
),
|
|
119
|
+
"DeleteSqlWorksheet": grpc.unary_unary_rpc_method_handler(
|
|
120
|
+
servicer.DeleteSqlWorksheet,
|
|
121
|
+
request_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.DeleteSqlWorksheetRequest.FromString,
|
|
122
|
+
response_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.DeleteSqlWorksheetResponse.SerializeToString,
|
|
123
|
+
),
|
|
124
|
+
"ListSqlWorksheets": grpc.unary_unary_rpc_method_handler(
|
|
125
|
+
servicer.ListSqlWorksheets,
|
|
126
|
+
request_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.ListSqlWorksheetsRequest.FromString,
|
|
127
|
+
response_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.ListSqlWorksheetsResponse.SerializeToString,
|
|
128
|
+
),
|
|
129
|
+
"RenameSqlWorksheet": grpc.unary_unary_rpc_method_handler(
|
|
130
|
+
servicer.RenameSqlWorksheet,
|
|
131
|
+
request_deserializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.RenameSqlWorksheetRequest.FromString,
|
|
132
|
+
response_serializer=chalk_dot_server_dot_v1_dot_sql__interface__pb2.RenameSqlWorksheetResponse.SerializeToString,
|
|
133
|
+
),
|
|
134
|
+
"ListFiles": grpc.unary_unary_rpc_method_handler(
|
|
135
|
+
servicer.ListFiles,
|
|
136
|
+
request_deserializer=chalk_dot_server_dot_v1_dot_files__pb2.ListFilesRequest.FromString,
|
|
137
|
+
response_serializer=chalk_dot_server_dot_v1_dot_files__pb2.ListFilesResponse.SerializeToString,
|
|
138
|
+
),
|
|
139
|
+
}
|
|
140
|
+
generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.SqlInterfaceService", rpc_method_handlers)
|
|
141
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
|
142
|
+
|
|
143
|
+
|
|
144
|
+
# This class is part of an EXPERIMENTAL API.
|
|
145
|
+
class SqlInterfaceService(object):
|
|
146
|
+
"""Missing associated documentation comment in .proto file."""
|
|
147
|
+
|
|
148
|
+
@staticmethod
|
|
149
|
+
def CreateSqlWorksheet(
|
|
150
|
+
request,
|
|
151
|
+
target,
|
|
152
|
+
options=(),
|
|
153
|
+
channel_credentials=None,
|
|
154
|
+
call_credentials=None,
|
|
155
|
+
insecure=False,
|
|
156
|
+
compression=None,
|
|
157
|
+
wait_for_ready=None,
|
|
158
|
+
timeout=None,
|
|
159
|
+
metadata=None,
|
|
160
|
+
):
|
|
161
|
+
return grpc.experimental.unary_unary(
|
|
162
|
+
request,
|
|
163
|
+
target,
|
|
164
|
+
"/chalk.server.v1.SqlInterfaceService/CreateSqlWorksheet",
|
|
165
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.CreateSqlWorksheetRequest.SerializeToString,
|
|
166
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.CreateSqlWorksheetResponse.FromString,
|
|
167
|
+
options,
|
|
168
|
+
channel_credentials,
|
|
169
|
+
insecure,
|
|
170
|
+
call_credentials,
|
|
171
|
+
compression,
|
|
172
|
+
wait_for_ready,
|
|
173
|
+
timeout,
|
|
174
|
+
metadata,
|
|
175
|
+
)
|
|
176
|
+
|
|
177
|
+
@staticmethod
|
|
178
|
+
def GetSqlWorksheet(
|
|
179
|
+
request,
|
|
180
|
+
target,
|
|
181
|
+
options=(),
|
|
182
|
+
channel_credentials=None,
|
|
183
|
+
call_credentials=None,
|
|
184
|
+
insecure=False,
|
|
185
|
+
compression=None,
|
|
186
|
+
wait_for_ready=None,
|
|
187
|
+
timeout=None,
|
|
188
|
+
metadata=None,
|
|
189
|
+
):
|
|
190
|
+
return grpc.experimental.unary_unary(
|
|
191
|
+
request,
|
|
192
|
+
target,
|
|
193
|
+
"/chalk.server.v1.SqlInterfaceService/GetSqlWorksheet",
|
|
194
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.GetSqlWorksheetRequest.SerializeToString,
|
|
195
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.GetSqlWorksheetResponse.FromString,
|
|
196
|
+
options,
|
|
197
|
+
channel_credentials,
|
|
198
|
+
insecure,
|
|
199
|
+
call_credentials,
|
|
200
|
+
compression,
|
|
201
|
+
wait_for_ready,
|
|
202
|
+
timeout,
|
|
203
|
+
metadata,
|
|
204
|
+
)
|
|
205
|
+
|
|
206
|
+
@staticmethod
|
|
207
|
+
def UpdateSqlWorksheet(
|
|
208
|
+
request,
|
|
209
|
+
target,
|
|
210
|
+
options=(),
|
|
211
|
+
channel_credentials=None,
|
|
212
|
+
call_credentials=None,
|
|
213
|
+
insecure=False,
|
|
214
|
+
compression=None,
|
|
215
|
+
wait_for_ready=None,
|
|
216
|
+
timeout=None,
|
|
217
|
+
metadata=None,
|
|
218
|
+
):
|
|
219
|
+
return grpc.experimental.unary_unary(
|
|
220
|
+
request,
|
|
221
|
+
target,
|
|
222
|
+
"/chalk.server.v1.SqlInterfaceService/UpdateSqlWorksheet",
|
|
223
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.UpdateSqlWorksheetRequest.SerializeToString,
|
|
224
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.UpdateSqlWorksheetResponse.FromString,
|
|
225
|
+
options,
|
|
226
|
+
channel_credentials,
|
|
227
|
+
insecure,
|
|
228
|
+
call_credentials,
|
|
229
|
+
compression,
|
|
230
|
+
wait_for_ready,
|
|
231
|
+
timeout,
|
|
232
|
+
metadata,
|
|
233
|
+
)
|
|
234
|
+
|
|
235
|
+
@staticmethod
|
|
236
|
+
def DeleteSqlWorksheet(
|
|
237
|
+
request,
|
|
238
|
+
target,
|
|
239
|
+
options=(),
|
|
240
|
+
channel_credentials=None,
|
|
241
|
+
call_credentials=None,
|
|
242
|
+
insecure=False,
|
|
243
|
+
compression=None,
|
|
244
|
+
wait_for_ready=None,
|
|
245
|
+
timeout=None,
|
|
246
|
+
metadata=None,
|
|
247
|
+
):
|
|
248
|
+
return grpc.experimental.unary_unary(
|
|
249
|
+
request,
|
|
250
|
+
target,
|
|
251
|
+
"/chalk.server.v1.SqlInterfaceService/DeleteSqlWorksheet",
|
|
252
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.DeleteSqlWorksheetRequest.SerializeToString,
|
|
253
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.DeleteSqlWorksheetResponse.FromString,
|
|
254
|
+
options,
|
|
255
|
+
channel_credentials,
|
|
256
|
+
insecure,
|
|
257
|
+
call_credentials,
|
|
258
|
+
compression,
|
|
259
|
+
wait_for_ready,
|
|
260
|
+
timeout,
|
|
261
|
+
metadata,
|
|
262
|
+
)
|
|
263
|
+
|
|
264
|
+
@staticmethod
|
|
265
|
+
def ListSqlWorksheets(
|
|
266
|
+
request,
|
|
267
|
+
target,
|
|
268
|
+
options=(),
|
|
269
|
+
channel_credentials=None,
|
|
270
|
+
call_credentials=None,
|
|
271
|
+
insecure=False,
|
|
272
|
+
compression=None,
|
|
273
|
+
wait_for_ready=None,
|
|
274
|
+
timeout=None,
|
|
275
|
+
metadata=None,
|
|
276
|
+
):
|
|
277
|
+
return grpc.experimental.unary_unary(
|
|
278
|
+
request,
|
|
279
|
+
target,
|
|
280
|
+
"/chalk.server.v1.SqlInterfaceService/ListSqlWorksheets",
|
|
281
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.ListSqlWorksheetsRequest.SerializeToString,
|
|
282
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.ListSqlWorksheetsResponse.FromString,
|
|
283
|
+
options,
|
|
284
|
+
channel_credentials,
|
|
285
|
+
insecure,
|
|
286
|
+
call_credentials,
|
|
287
|
+
compression,
|
|
288
|
+
wait_for_ready,
|
|
289
|
+
timeout,
|
|
290
|
+
metadata,
|
|
291
|
+
)
|
|
292
|
+
|
|
293
|
+
@staticmethod
|
|
294
|
+
def RenameSqlWorksheet(
|
|
295
|
+
request,
|
|
296
|
+
target,
|
|
297
|
+
options=(),
|
|
298
|
+
channel_credentials=None,
|
|
299
|
+
call_credentials=None,
|
|
300
|
+
insecure=False,
|
|
301
|
+
compression=None,
|
|
302
|
+
wait_for_ready=None,
|
|
303
|
+
timeout=None,
|
|
304
|
+
metadata=None,
|
|
305
|
+
):
|
|
306
|
+
return grpc.experimental.unary_unary(
|
|
307
|
+
request,
|
|
308
|
+
target,
|
|
309
|
+
"/chalk.server.v1.SqlInterfaceService/RenameSqlWorksheet",
|
|
310
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.RenameSqlWorksheetRequest.SerializeToString,
|
|
311
|
+
chalk_dot_server_dot_v1_dot_sql__interface__pb2.RenameSqlWorksheetResponse.FromString,
|
|
312
|
+
options,
|
|
313
|
+
channel_credentials,
|
|
314
|
+
insecure,
|
|
315
|
+
call_credentials,
|
|
316
|
+
compression,
|
|
317
|
+
wait_for_ready,
|
|
318
|
+
timeout,
|
|
319
|
+
metadata,
|
|
320
|
+
)
|
|
321
|
+
|
|
322
|
+
@staticmethod
|
|
323
|
+
def ListFiles(
|
|
324
|
+
request,
|
|
325
|
+
target,
|
|
326
|
+
options=(),
|
|
327
|
+
channel_credentials=None,
|
|
328
|
+
call_credentials=None,
|
|
329
|
+
insecure=False,
|
|
330
|
+
compression=None,
|
|
331
|
+
wait_for_ready=None,
|
|
332
|
+
timeout=None,
|
|
333
|
+
metadata=None,
|
|
334
|
+
):
|
|
335
|
+
return grpc.experimental.unary_unary(
|
|
336
|
+
request,
|
|
337
|
+
target,
|
|
338
|
+
"/chalk.server.v1.SqlInterfaceService/ListFiles",
|
|
339
|
+
chalk_dot_server_dot_v1_dot_files__pb2.ListFilesRequest.SerializeToString,
|
|
340
|
+
chalk_dot_server_dot_v1_dot_files__pb2.ListFilesResponse.FromString,
|
|
341
|
+
options,
|
|
342
|
+
channel_credentials,
|
|
343
|
+
insecure,
|
|
344
|
+
call_credentials,
|
|
345
|
+
compression,
|
|
346
|
+
wait_for_ready,
|
|
347
|
+
timeout,
|
|
348
|
+
metadata,
|
|
349
|
+
)
|
|
@@ -0,0 +1,114 @@
|
|
|
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.server.v1.files_pb2 import (
|
|
11
|
+
ListFilesRequest,
|
|
12
|
+
ListFilesResponse,
|
|
13
|
+
)
|
|
14
|
+
from chalk._gen.chalk.server.v1.sql_interface_pb2 import (
|
|
15
|
+
CreateSqlWorksheetRequest,
|
|
16
|
+
CreateSqlWorksheetResponse,
|
|
17
|
+
DeleteSqlWorksheetRequest,
|
|
18
|
+
DeleteSqlWorksheetResponse,
|
|
19
|
+
GetSqlWorksheetRequest,
|
|
20
|
+
GetSqlWorksheetResponse,
|
|
21
|
+
ListSqlWorksheetsRequest,
|
|
22
|
+
ListSqlWorksheetsResponse,
|
|
23
|
+
RenameSqlWorksheetRequest,
|
|
24
|
+
RenameSqlWorksheetResponse,
|
|
25
|
+
UpdateSqlWorksheetRequest,
|
|
26
|
+
UpdateSqlWorksheetResponse,
|
|
27
|
+
)
|
|
28
|
+
from grpc import (
|
|
29
|
+
Channel,
|
|
30
|
+
Server,
|
|
31
|
+
ServicerContext,
|
|
32
|
+
UnaryUnaryMultiCallable,
|
|
33
|
+
)
|
|
34
|
+
|
|
35
|
+
class SqlInterfaceServiceStub:
|
|
36
|
+
def __init__(self, channel: Channel) -> None: ...
|
|
37
|
+
CreateSqlWorksheet: UnaryUnaryMultiCallable[
|
|
38
|
+
CreateSqlWorksheetRequest,
|
|
39
|
+
CreateSqlWorksheetResponse,
|
|
40
|
+
]
|
|
41
|
+
"""SQL Interface operations for ChalkSQL Explorer"""
|
|
42
|
+
GetSqlWorksheet: UnaryUnaryMultiCallable[
|
|
43
|
+
GetSqlWorksheetRequest,
|
|
44
|
+
GetSqlWorksheetResponse,
|
|
45
|
+
]
|
|
46
|
+
UpdateSqlWorksheet: UnaryUnaryMultiCallable[
|
|
47
|
+
UpdateSqlWorksheetRequest,
|
|
48
|
+
UpdateSqlWorksheetResponse,
|
|
49
|
+
]
|
|
50
|
+
DeleteSqlWorksheet: UnaryUnaryMultiCallable[
|
|
51
|
+
DeleteSqlWorksheetRequest,
|
|
52
|
+
DeleteSqlWorksheetResponse,
|
|
53
|
+
]
|
|
54
|
+
ListSqlWorksheets: UnaryUnaryMultiCallable[
|
|
55
|
+
ListSqlWorksheetsRequest,
|
|
56
|
+
ListSqlWorksheetsResponse,
|
|
57
|
+
]
|
|
58
|
+
RenameSqlWorksheet: UnaryUnaryMultiCallable[
|
|
59
|
+
RenameSqlWorksheetRequest,
|
|
60
|
+
RenameSqlWorksheetResponse,
|
|
61
|
+
]
|
|
62
|
+
ListFiles: UnaryUnaryMultiCallable[
|
|
63
|
+
ListFilesRequest,
|
|
64
|
+
ListFilesResponse,
|
|
65
|
+
]
|
|
66
|
+
"""File operations - only ListFiles is implemented (files are managed via SQL worksheets)"""
|
|
67
|
+
|
|
68
|
+
class SqlInterfaceServiceServicer(metaclass=ABCMeta):
|
|
69
|
+
@abstractmethod
|
|
70
|
+
def CreateSqlWorksheet(
|
|
71
|
+
self,
|
|
72
|
+
request: CreateSqlWorksheetRequest,
|
|
73
|
+
context: ServicerContext,
|
|
74
|
+
) -> CreateSqlWorksheetResponse:
|
|
75
|
+
"""SQL Interface operations for ChalkSQL Explorer"""
|
|
76
|
+
@abstractmethod
|
|
77
|
+
def GetSqlWorksheet(
|
|
78
|
+
self,
|
|
79
|
+
request: GetSqlWorksheetRequest,
|
|
80
|
+
context: ServicerContext,
|
|
81
|
+
) -> GetSqlWorksheetResponse: ...
|
|
82
|
+
@abstractmethod
|
|
83
|
+
def UpdateSqlWorksheet(
|
|
84
|
+
self,
|
|
85
|
+
request: UpdateSqlWorksheetRequest,
|
|
86
|
+
context: ServicerContext,
|
|
87
|
+
) -> UpdateSqlWorksheetResponse: ...
|
|
88
|
+
@abstractmethod
|
|
89
|
+
def DeleteSqlWorksheet(
|
|
90
|
+
self,
|
|
91
|
+
request: DeleteSqlWorksheetRequest,
|
|
92
|
+
context: ServicerContext,
|
|
93
|
+
) -> DeleteSqlWorksheetResponse: ...
|
|
94
|
+
@abstractmethod
|
|
95
|
+
def ListSqlWorksheets(
|
|
96
|
+
self,
|
|
97
|
+
request: ListSqlWorksheetsRequest,
|
|
98
|
+
context: ServicerContext,
|
|
99
|
+
) -> ListSqlWorksheetsResponse: ...
|
|
100
|
+
@abstractmethod
|
|
101
|
+
def RenameSqlWorksheet(
|
|
102
|
+
self,
|
|
103
|
+
request: RenameSqlWorksheetRequest,
|
|
104
|
+
context: ServicerContext,
|
|
105
|
+
) -> RenameSqlWorksheetResponse: ...
|
|
106
|
+
@abstractmethod
|
|
107
|
+
def ListFiles(
|
|
108
|
+
self,
|
|
109
|
+
request: ListFilesRequest,
|
|
110
|
+
context: ServicerContext,
|
|
111
|
+
) -> ListFilesResponse:
|
|
112
|
+
"""File operations - only ListFiles is implemented (files are managed via SQL worksheets)"""
|
|
113
|
+
|
|
114
|
+
def add_SqlInterfaceServiceServicer_to_server(servicer: SqlInterfaceServiceServicer, server: Server) -> None: ...
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: chalk/server/v1/sql_queries.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/server/v1/sql_queries.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xe9\x06\n\x08SqlQuery\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1e\n\x08\x61gent_id\x18\x02 \x01(\tH\x00R\x07\x61gentId\x88\x01\x01\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x04 \x01(\tH\x01R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x1d\n\nquery_text\x18\x06 \x01(\tR\tqueryText\x12\x1f\n\x08\x64uration\x18\x07 \x01(\x01H\x02R\x08\x64uration\x88\x01\x01\x12\x16\n\x06status\x18\x08 \x01(\x05R\x06status\x12*\n\x0equery_metadata\x18\t \x01(\tH\x03R\rqueryMetadata\x88\x01\x01\x12\x1e\n\x08trace_id\x18\n \x01(\tH\x04R\x07traceId\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x0b \x01(\tH\x05R\nbranchName\x88\x01\x01\x12+\n\x0fhas_plan_stages\x18\x0c \x01(\x08H\x06R\rhasPlanStages\x88\x01\x01\x12$\n\x0b\x65xternal_id\x18\r \x01(\tH\x07R\nexternalId\x88\x01\x01\x12*\n\x0e\x63orrelation_id\x18\x0e \x01(\tH\x08R\rcorrelationId\x88\x01\x01\x12*\n\x0eresource_group\x18\x0f \x01(\tH\tR\rresourceGroup\x88\x01\x01\x12+\n\x0fquery_plan_json\x18\x10 \x01(\tH\nR\rqueryPlanJson\x88\x01\x01\x12/\n\x11output_uri_prefix\x18\x11 \x01(\tH\x0bR\x0foutputUriPrefix\x88\x01\x01\x42\x0b\n\t_agent_idB\x10\n\x0e_deployment_idB\x0b\n\t_durationB\x11\n\x0f_query_metadataB\x0b\n\t_trace_idB\x0e\n\x0c_branch_nameB\x12\n\x10_has_plan_stagesB\x0e\n\x0c_external_idB\x11\n\x0f_correlation_idB\x11\n\x0f_resource_groupB\x12\n\x10_query_plan_jsonB\x14\n\x12_output_uri_prefix"\x91\x03\n\x15ListSqlQueriesRequest\x12\x14\n\x05limit\x18\x01 \x01(\x05R\x05limit\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x12\x1e\n\x08\x61gent_id\x18\x03 \x01(\tH\x01R\x07\x61gentId\x88\x01\x01\x12(\n\rdeployment_id\x18\x04 \x01(\tH\x02R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1b\n\x06status\x18\x05 \x01(\x05H\x03R\x06status\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x06 \x01(\tH\x04R\nbranchName\x88\x01\x01\x12$\n\x0b\x65xternal_id\x18\x07 \x01(\tH\x05R\nexternalId\x88\x01\x01\x12*\n\x0e\x63orrelation_id\x18\x08 \x01(\tH\x06R\rcorrelationId\x88\x01\x01\x42\t\n\x07_cursorB\x0b\n\t_agent_idB\x10\n\x0e_deployment_idB\t\n\x07_statusB\x0e\n\x0c_branch_nameB\x0e\n\x0c_external_idB\x11\n\x0f_correlation_id"\x83\x01\n\x16ListSqlQueriesResponse\x12\x33\n\x07queries\x18\x01 \x03(\x0b\x32\x19.chalk.server.v1.SqlQueryR\x07queries\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"$\n\x12GetSqlQueryRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"F\n\x13GetSqlQueryResponse\x12/\n\x05query\x18\x01 \x01(\x0b\x32\x19.chalk.server.v1.SqlQueryR\x05query2\xe0\x01\n\x11SqlQueriesService\x12i\n\x0eListSqlQueries\x12&.chalk.server.v1.ListSqlQueriesRequest\x1a\'.chalk.server.v1.ListSqlQueriesResponse"\x06\x90\x02\x01\x80}\x03\x12`\n\x0bGetSqlQuery\x12#.chalk.server.v1.GetSqlQueryRequest\x1a$.chalk.server.v1.GetSqlQueryResponse"\x06\x90\x02\x01\x80}\x03\x42\x98\x01\n\x13\x63om.chalk.server.v1B\x0fSqlQueriesProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
|
|
22
|
+
)
|
|
23
|
+
|
|
24
|
+
_globals = globals()
|
|
25
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
26
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.server.v1.sql_queries_pb2", _globals)
|
|
27
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
|
28
|
+
_globals["DESCRIPTOR"]._options = None
|
|
29
|
+
_globals[
|
|
30
|
+
"DESCRIPTOR"
|
|
31
|
+
]._serialized_options = b"\n\023com.chalk.server.v1B\017SqlQueriesProtoP\001Z\022server/v1;serverv1\242\002\003CSX\252\002\017Chalk.Server.V1\312\002\017Chalk\\Server\\V1\342\002\033Chalk\\Server\\V1\\GPBMetadata\352\002\021Chalk::Server::V1"
|
|
32
|
+
_globals["_SQLQUERIESSERVICE"].methods_by_name["ListSqlQueries"]._options = None
|
|
33
|
+
_globals["_SQLQUERIESSERVICE"].methods_by_name["ListSqlQueries"]._serialized_options = b"\220\002\001\200}\003"
|
|
34
|
+
_globals["_SQLQUERIESSERVICE"].methods_by_name["GetSqlQuery"]._options = None
|
|
35
|
+
_globals["_SQLQUERIESSERVICE"].methods_by_name["GetSqlQuery"]._serialized_options = b"\220\002\001\200}\003"
|
|
36
|
+
_globals["_SQLQUERY"]._serialized_start = 121
|
|
37
|
+
_globals["_SQLQUERY"]._serialized_end = 994
|
|
38
|
+
_globals["_LISTSQLQUERIESREQUEST"]._serialized_start = 997
|
|
39
|
+
_globals["_LISTSQLQUERIESREQUEST"]._serialized_end = 1398
|
|
40
|
+
_globals["_LISTSQLQUERIESRESPONSE"]._serialized_start = 1401
|
|
41
|
+
_globals["_LISTSQLQUERIESRESPONSE"]._serialized_end = 1532
|
|
42
|
+
_globals["_GETSQLQUERYREQUEST"]._serialized_start = 1534
|
|
43
|
+
_globals["_GETSQLQUERYREQUEST"]._serialized_end = 1570
|
|
44
|
+
_globals["_GETSQLQUERYRESPONSE"]._serialized_start = 1572
|
|
45
|
+
_globals["_GETSQLQUERYRESPONSE"]._serialized_end = 1642
|
|
46
|
+
_globals["_SQLQUERIESSERVICE"]._serialized_start = 1645
|
|
47
|
+
_globals["_SQLQUERIESSERVICE"]._serialized_end = 1869
|
|
48
|
+
# @@protoc_insertion_point(module_scope)
|