chalkpy 2.89.22__py3-none-any.whl → 2.95.3__py3-none-any.whl

chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi

@@ -0,0 +1,150 @@
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class SqlQuery(_message.Message):
+    __slots__ = (
+        "id",
+        "agent_id",
+        "environment_id",
+        "deployment_id",
+        "created_at",
+        "query_text",
+        "duration",
+        "status",
+        "query_metadata",
+        "trace_id",
+        "branch_name",
+        "has_plan_stages",
+        "external_id",
+        "correlation_id",
+        "resource_group",
+        "query_plan_json",
+        "output_uri_prefix",
+    )
+    ID_FIELD_NUMBER: _ClassVar[int]
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    QUERY_TEXT_FIELD_NUMBER: _ClassVar[int]
+    DURATION_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    QUERY_METADATA_FIELD_NUMBER: _ClassVar[int]
+    TRACE_ID_FIELD_NUMBER: _ClassVar[int]
+    BRANCH_NAME_FIELD_NUMBER: _ClassVar[int]
+    HAS_PLAN_STAGES_FIELD_NUMBER: _ClassVar[int]
+    EXTERNAL_ID_FIELD_NUMBER: _ClassVar[int]
+    CORRELATION_ID_FIELD_NUMBER: _ClassVar[int]
+    RESOURCE_GROUP_FIELD_NUMBER: _ClassVar[int]
+    QUERY_PLAN_JSON_FIELD_NUMBER: _ClassVar[int]
+    OUTPUT_URI_PREFIX_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    agent_id: str
+    environment_id: str
+    deployment_id: str
+    created_at: _timestamp_pb2.Timestamp
+    query_text: str
+    duration: float
+    status: int
+    query_metadata: str
+    trace_id: str
+    branch_name: str
+    has_plan_stages: bool
+    external_id: str
+    correlation_id: str
+    resource_group: str
+    query_plan_json: str
+    output_uri_prefix: str
+    def __init__(
+        self,
+        id: _Optional[str] = ...,
+        agent_id: _Optional[str] = ...,
+        environment_id: _Optional[str] = ...,
+        deployment_id: _Optional[str] = ...,
+        created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        query_text: _Optional[str] = ...,
+        duration: _Optional[float] = ...,
+        status: _Optional[int] = ...,
+        query_metadata: _Optional[str] = ...,
+        trace_id: _Optional[str] = ...,
+        branch_name: _Optional[str] = ...,
+        has_plan_stages: bool = ...,
+        external_id: _Optional[str] = ...,
+        correlation_id: _Optional[str] = ...,
+        resource_group: _Optional[str] = ...,
+        query_plan_json: _Optional[str] = ...,
+        output_uri_prefix: _Optional[str] = ...,
+    ) -> None: ...
+
+class ListSqlQueriesRequest(_message.Message):
+    __slots__ = (
+        "limit",
+        "cursor",
+        "agent_id",
+        "deployment_id",
+        "status",
+        "branch_name",
+        "external_id",
+        "correlation_id",
+    )
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    CURSOR_FIELD_NUMBER: _ClassVar[int]
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    BRANCH_NAME_FIELD_NUMBER: _ClassVar[int]
+    EXTERNAL_ID_FIELD_NUMBER: _ClassVar[int]
+    CORRELATION_ID_FIELD_NUMBER: _ClassVar[int]
+    limit: int
+    cursor: str
+    agent_id: str
+    deployment_id: str
+    status: int
+    branch_name: str
+    external_id: str
+    correlation_id: str
+    def __init__(
+        self,
+        limit: _Optional[int] = ...,
+        cursor: _Optional[str] = ...,
+        agent_id: _Optional[str] = ...,
+        deployment_id: _Optional[str] = ...,
+        status: _Optional[int] = ...,
+        branch_name: _Optional[str] = ...,
+        external_id: _Optional[str] = ...,
+        correlation_id: _Optional[str] = ...,
+    ) -> None: ...
+
+class ListSqlQueriesResponse(_message.Message):
+    __slots__ = ("queries", "next_cursor")
+    QUERIES_FIELD_NUMBER: _ClassVar[int]
+    NEXT_CURSOR_FIELD_NUMBER: _ClassVar[int]
+    queries: _containers.RepeatedCompositeFieldContainer[SqlQuery]
+    next_cursor: str
+    def __init__(
+        self, queries: _Optional[_Iterable[_Union[SqlQuery, _Mapping]]] = ..., next_cursor: _Optional[str] = ...
+    ) -> None: ...
+
+class GetSqlQueryRequest(_message.Message):
+    __slots__ = ("id",)
+    ID_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    def __init__(self, id: _Optional[str] = ...) -> None: ...
+
+class GetSqlQueryResponse(_message.Message):
+    __slots__ = ("query",)
+    QUERY_FIELD_NUMBER: _ClassVar[int]
+    query: SqlQuery
+    def __init__(self, query: _Optional[_Union[SqlQuery, _Mapping]] = ...) -> None: ...

chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py

@@ -0,0 +1,123 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+
+import grpc
+
+from chalk._gen.chalk.server.v1 import sql_queries_pb2 as chalk_dot_server_dot_v1_dot_sql__queries__pb2
+
+
+class SqlQueriesServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.ListSqlQueries = channel.unary_unary(
+            "/chalk.server.v1.SqlQueriesService/ListSqlQueries",
+            request_serializer=chalk_dot_server_dot_v1_dot_sql__queries__pb2.ListSqlQueriesRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_sql__queries__pb2.ListSqlQueriesResponse.FromString,
+        )
+        self.GetSqlQuery = channel.unary_unary(
+            "/chalk.server.v1.SqlQueriesService/GetSqlQuery",
+            request_serializer=chalk_dot_server_dot_v1_dot_sql__queries__pb2.GetSqlQueryRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_sql__queries__pb2.GetSqlQueryResponse.FromString,
+        )
+
+
+class SqlQueriesServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def ListSqlQueries(self, request, context):
+        """List SQL queries with pagination and filtering"""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def GetSqlQuery(self, request, context):
+        """Get a specific SQL query by ID"""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+
+def add_SqlQueriesServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "ListSqlQueries": grpc.unary_unary_rpc_method_handler(
+            servicer.ListSqlQueries,
+            request_deserializer=chalk_dot_server_dot_v1_dot_sql__queries__pb2.ListSqlQueriesRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_sql__queries__pb2.ListSqlQueriesResponse.SerializeToString,
+        ),
+        "GetSqlQuery": grpc.unary_unary_rpc_method_handler(
+            servicer.GetSqlQuery,
+            request_deserializer=chalk_dot_server_dot_v1_dot_sql__queries__pb2.GetSqlQueryRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_sql__queries__pb2.GetSqlQueryResponse.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.SqlQueriesService", rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+# This class is part of an EXPERIMENTAL API.
+class SqlQueriesService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def ListSqlQueries(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.SqlQueriesService/ListSqlQueries",
+            chalk_dot_server_dot_v1_dot_sql__queries__pb2.ListSqlQueriesRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_sql__queries__pb2.ListSqlQueriesResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetSqlQuery(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.SqlQueriesService/GetSqlQuery",
+            chalk_dot_server_dot_v1_dot_sql__queries__pb2.GetSqlQueryRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_sql__queries__pb2.GetSqlQueryResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi

@@ -0,0 +1,52 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from abc import (
+    ABCMeta,
+    abstractmethod,
+)
+from chalk._gen.chalk.server.v1.sql_queries_pb2 import (
+    GetSqlQueryRequest,
+    GetSqlQueryResponse,
+    ListSqlQueriesRequest,
+    ListSqlQueriesResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class SqlQueriesServiceStub:
+    def __init__(self, channel: Channel) -> None: ...
+    ListSqlQueries: UnaryUnaryMultiCallable[
+        ListSqlQueriesRequest,
+        ListSqlQueriesResponse,
+    ]
+    """List SQL queries with pagination and filtering"""
+    GetSqlQuery: UnaryUnaryMultiCallable[
+        GetSqlQueryRequest,
+        GetSqlQueryResponse,
+    ]
+    """Get a specific SQL query by ID"""
+
+class SqlQueriesServiceServicer(metaclass=ABCMeta):
+    @abstractmethod
+    def ListSqlQueries(
+        self,
+        request: ListSqlQueriesRequest,
+        context: ServicerContext,
+    ) -> ListSqlQueriesResponse:
+        """List SQL queries with pagination and filtering"""
+    @abstractmethod
+    def GetSqlQuery(
+        self,
+        request: GetSqlQueryRequest,
+        context: ServicerContext,
+    ) -> GetSqlQueryResponse:
+        """Get a specific SQL query by ID"""
+
+def add_SqlQueriesServiceServicer_to_server(servicer: SqlQueriesServiceServicer, server: Server) -> None: ...