chalkpy 2.89.22__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 +36 -33
- chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi +41 -1
- 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 +19 -13
- chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +37 -0
- chalk/_gen/chalk/common/v1/online_query_pb2.py +54 -54
- chalk/_gen/chalk/common/v1/online_query_pb2.pyi +13 -1
- chalk/_gen/chalk/common/v1/script_task_pb2.py +13 -11
- chalk/_gen/chalk/common/v1/script_task_pb2.pyi +19 -1
- 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 +372 -272
- chalk/_gen/chalk/server/v1/builder_pb2.pyi +479 -12
- chalk/_gen/chalk/server/v1/builder_pb2_grpc.py +360 -0
- chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi +96 -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 +153 -107
- chalk/_gen/chalk/server/v1/cloud_components_pb2.pyi +146 -4
- chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.py +180 -0
- chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.pyi +48 -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 +59 -35
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.pyi +127 -1
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.py +135 -0
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.pyi +36 -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/datasets_pb2.py +36 -24
- chalk/_gen/chalk/server/v1/datasets_pb2.pyi +71 -2
- chalk/_gen/chalk/server/v1/datasets_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/datasets_pb2_grpc.pyi +12 -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 +20 -15
- chalk/_gen/chalk/server/v1/deployment_pb2.pyi +25 -0
- chalk/_gen/chalk/server/v1/environment_pb2.py +25 -15
- chalk/_gen/chalk/server/v1/environment_pb2.pyi +93 -1
- chalk/_gen/chalk/server/v1/eventbus_pb2.py +44 -0
- chalk/_gen/chalk/server/v1/eventbus_pb2.pyi +64 -0
- chalk/_gen/chalk/server/v1/eventbus_pb2_grpc.py +4 -0
- chalk/_gen/chalk/server/v1/eventbus_pb2_grpc.pyi +4 -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/graph_pb2.py +41 -3
- chalk/_gen/chalk/server/v1/graph_pb2.pyi +191 -0
- chalk/_gen/chalk/server/v1/graph_pb2_grpc.py +92 -0
- chalk/_gen/chalk/server/v1/graph_pb2_grpc.pyi +32 -0
- chalk/_gen/chalk/server/v1/incident_pb2.py +57 -0
- chalk/_gen/chalk/server/v1/incident_pb2.pyi +165 -0
- chalk/_gen/chalk/server/v1/incident_pb2_grpc.py +4 -0
- chalk/_gen/chalk/server/v1/incident_pb2_grpc.pyi +4 -0
- 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/metadataplanejobqueue_pb2.py +73 -0
- chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2.pyi +212 -0
- chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.py +217 -0
- chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.pyi +74 -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/monitoring_pb2.py +84 -75
- chalk/_gen/chalk/server/v1/monitoring_pb2.pyi +1 -0
- chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.py +136 -0
- chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.pyi +38 -0
- chalk/_gen/chalk/server/v1/offline_queries_pb2.py +32 -10
- chalk/_gen/chalk/server/v1/offline_queries_pb2.pyi +73 -0
- chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.pyi +24 -0
- 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 +76 -48
- chalk/_gen/chalk/server/v1/queries_pb2.pyi +155 -2
- chalk/_gen/chalk/server/v1/queries_pb2_grpc.py +180 -0
- chalk/_gen/chalk/server/v1/queries_pb2_grpc.pyi +48 -0
- chalk/_gen/chalk/server/v1/scheduled_query_pb2.py +4 -2
- chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +12 -6
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +75 -2
- 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 +26 -14
- chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi +33 -3
- 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 +156 -137
- chalk/_gen/chalk/server/v1/team_pb2.pyi +56 -10
- chalk/_gen/chalk/server/v1/team_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/team_pb2_grpc.pyi +24 -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 +50 -28
- chalk/_gen/chalk/server/v1/trace_pb2.pyi +121 -0
- chalk/_gen/chalk/server/v1/trace_pb2_grpc.py +135 -0
- chalk/_gen/chalk/server/v1/trace_pb2_grpc.pyi +42 -0
- chalk/_gen/chalk/server/v1/webhook_pb2.py +9 -3
- chalk/_gen/chalk/server/v1/webhook_pb2.pyi +18 -0
- chalk/_gen/chalk/server/v1/webhook_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/webhook_pb2_grpc.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 +19 -7
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +96 -3
- 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/_monitoring/Chart.py +1 -3
- chalk/_version.py +1 -1
- chalk/cli.py +5 -10
- chalk/client/client.py +178 -64
- chalk/client/client_async.py +154 -0
- chalk/client/client_async_impl.py +22 -0
- chalk/client/client_grpc.py +738 -112
- chalk/client/client_impl.py +541 -136
- chalk/client/dataset.py +27 -6
- chalk/client/models.py +99 -2
- chalk/client/serialization/model_serialization.py +126 -10
- chalk/config/project_config.py +1 -1
- chalk/df/LazyFramePlaceholder.py +1154 -0
- chalk/df/ast_parser.py +2 -10
- chalk/features/_class_property.py +7 -0
- chalk/features/_embedding/embedding.py +1 -0
- chalk/features/_embedding/sentence_transformer.py +1 -1
- chalk/features/_encoding/converter.py +83 -2
- chalk/features/_encoding/pyarrow.py +20 -4
- chalk/features/_encoding/rich.py +1 -3
- chalk/features/_tensor.py +1 -2
- chalk/features/dataframe/_filters.py +14 -5
- chalk/features/dataframe/_impl.py +91 -36
- chalk/features/dataframe/_validation.py +11 -7
- chalk/features/feature_field.py +40 -30
- chalk/features/feature_set.py +1 -2
- chalk/features/feature_set_decorator.py +1 -0
- chalk/features/feature_wrapper.py +42 -3
- chalk/features/hooks.py +81 -12
- chalk/features/inference.py +65 -10
- chalk/features/resolver.py +338 -56
- chalk/features/tag.py +1 -3
- chalk/features/underscore_features.py +2 -1
- chalk/functions/__init__.py +456 -21
- chalk/functions/holidays.py +1 -3
- chalk/gitignore/gitignore_parser.py +5 -1
- chalk/importer.py +186 -74
- chalk/ml/__init__.py +6 -2
- chalk/ml/model_hooks.py +368 -51
- chalk/ml/model_reference.py +68 -10
- chalk/ml/model_version.py +34 -21
- chalk/ml/utils.py +143 -40
- chalk/operators/_utils.py +14 -3
- chalk/parsed/_proto/export.py +22 -0
- chalk/parsed/duplicate_input_gql.py +4 -0
- chalk/parsed/expressions.py +1 -3
- chalk/parsed/json_conversions.py +21 -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 +13 -1
- chalk/serialization/parsed_annotation.py +25 -12
- chalk/sql/__init__.py +221 -0
- chalk/sql/_internal/integrations/athena.py +6 -1
- chalk/sql/_internal/integrations/bigquery.py +22 -2
- chalk/sql/_internal/integrations/databricks.py +61 -18
- chalk/sql/_internal/integrations/mssql.py +281 -0
- chalk/sql/_internal/integrations/postgres.py +11 -3
- chalk/sql/_internal/integrations/redshift.py +4 -0
- chalk/sql/_internal/integrations/snowflake.py +11 -2
- chalk/sql/_internal/integrations/util.py +2 -1
- chalk/sql/_internal/sql_file_resolver.py +55 -10
- chalk/sql/_internal/sql_source.py +36 -2
- chalk/streams/__init__.py +1 -3
- chalk/streams/_kafka_source.py +5 -1
- chalk/streams/_windows.py +16 -4
- chalk/streams/types.py +1 -2
- chalk/utils/__init__.py +1 -3
- chalk/utils/_otel_version.py +13 -0
- chalk/utils/async_helpers.py +14 -5
- chalk/utils/df_utils.py +2 -2
- chalk/utils/duration.py +1 -3
- chalk/utils/job_log_display.py +538 -0
- chalk/utils/missing_dependency.py +5 -4
- chalk/utils/notebook.py +255 -2
- chalk/utils/pl_helpers.py +190 -37
- chalk/utils/pydanticutil/pydantic_compat.py +1 -2
- chalk/utils/storage_client.py +246 -0
- chalk/utils/threading.py +1 -3
- chalk/utils/tracing.py +194 -86
- {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/METADATA +53 -21
- {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/RECORD +268 -198
- {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/WHEEL +0 -0
- {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/entry_points.txt +0 -0
- {chalkpy-2.89.22.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)
|