databay-cli 0.1.0__py3-none-any.whl

databay/resources/source_stream.py

@@ -0,0 +1,38 @@
+"""``client.source_stream``: the ``source-stream`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from databay.operations import source_stream as _ops
+from databay.resource import Resource, UNSET, Unset
+
+if TYPE_CHECKING:
+    from databay.models.source_stream import StreamCatalogResponse, StreamShapeResponse
+
+
+class SourceStreamResource(Resource):
+
+    def get(
+        self,
+        *,
+        source_uuid: str,
+        namespace: str,
+        stream: str,
+        project: str | None = None,
+    ) -> StreamShapeResponse:
+        """Connected Source Stream Shape. ``databay source-stream get``."""
+        return self._call(_ops.SOURCE_STREAM_GET, {'source_uuid': source_uuid, 'namespace': namespace, 'stream': stream}, project=project)
+
+    def list(
+        self,
+        *,
+        source_uuid: str,
+        namespace: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> StreamCatalogResponse:
+        """Connected Source Streams. ``databay source-stream list``."""
+        return self._call(_ops.SOURCE_STREAM_LIST, {'source_uuid': source_uuid, 'namespace': namespace}, project=project)

databay/resources/source_table.py

@@ -0,0 +1,27 @@
+"""``client.source_table``: the ``source-table`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from databay.operations import source_table as _ops
+from databay.resource import Resource
+
+if TYPE_CHECKING:
+    from databay.models.source_table import TableReadinessReport
+
+
+class SourceTableResource(Resource):
+
+    def check(
+        self,
+        *,
+        source_uuid: str,
+        schema_name: str,
+        project: str | None = None,
+    ) -> TableReadinessReport:
+        """Connected Source Table Readiness. ``databay source-table check``."""
+        return self._call(_ops.SOURCE_TABLE_CHECK, {'source_uuid': source_uuid, 'schema_name': schema_name}, project=project)

databay/resources/sql.py

@@ -0,0 +1,27 @@
+"""``client.sql``: the ``sql`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from databay.operations import sql as _ops
+from databay.resource import Resource, UNSET, Unset
+
+if TYPE_CHECKING:
+    from databay.models.sql import SqlExecuteResponse
+
+
+class SqlResource(Resource):
+
+    def execute(
+        self,
+        *,
+        query: str,
+        db_uuid: str | None | Unset = UNSET,
+        project: str | None = None,
+    ) -> SqlExecuteResponse:
+        """Execute Sql. ``databay sql execute``."""
+        return self._call(_ops.SQL_EXECUTE, {'db_uuid': db_uuid, 'query': query}, project=project)

databay/resources/ssbi_job.py

@@ -0,0 +1,69 @@
+"""``client.ssbi_job``: the ``ssbi-job`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from databay.operations import ssbi_job as _ops
+from databay.resource import Resource, UNSET, Unset
+
+if TYPE_CHECKING:
+    from databay.models.ssbi_job import JobResponse, SsbiJobListResponse
+
+
+class SsbiJobResource(Resource):
+
+    def create(
+        self,
+        *,
+        destination_table: str,
+        name: str,
+        resource: Any,
+        schedule: Any,
+        transformation: Any,
+        project: str | None = None,
+        idempotency_key: str | None = None,
+    ) -> JobResponse:
+        """Create a new Job. ``databay ssbi-job create``."""
+        return self._call(_ops.SSBI_JOB_CREATE, {'destination_table': destination_table, 'name': name, 'resource': resource, 'schedule': schedule, 'transformation': transformation}, project=project, idempotency_key=idempotency_key)
+
+    def delete(
+        self,
+        *,
+        job_uuid: str,
+    ) -> None:
+        """Delete an Job. ``databay ssbi-job delete``."""
+        return self._call(_ops.SSBI_JOB_DELETE, {'job_uuid': job_uuid})
+
+    def get(
+        self,
+        *,
+        job_uuid: str,
+    ) -> JobResponse:
+        """Get a specific Job by UUID. ``databay ssbi-job get``."""
+        return self._call(_ops.SSBI_JOB_GET, {'job_uuid': job_uuid})
+
+    def list(
+        self,
+        *,
+        skip: int | Unset = UNSET,
+        limit: int | Unset = UNSET,
+        project: str | None = None,
+    ) -> SsbiJobListResponse:
+        """List all Jobs. ``databay ssbi-job list``."""
+        return self._call(_ops.SSBI_JOB_LIST, {'skip': skip, 'limit': limit}, project=project)
+
+    def update(
+        self,
+        *,
+        job_uuid: str,
+        name: str | None | Unset = UNSET,
+        resource: Any | None | Unset = UNSET,
+        schedule: Any | None | Unset = UNSET,
+        transformation: Any | None | Unset = UNSET,
+    ) -> JobResponse:
+        """Update an Job. ``databay ssbi-job update``."""
+        return self._call(_ops.SSBI_JOB_UPDATE, {'job_uuid': job_uuid, 'name': name, 'resource': resource, 'schedule': schedule, 'transformation': transformation})

databay/resources/transform_type.py

@@ -0,0 +1,21 @@
+"""``client.transform_type``: the ``transform-type`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from databay.operations import transform_type as _ops
+from databay.resource import Resource
+
+if TYPE_CHECKING:
+    from databay.models.transform_type import TransformTypeListResponse
+
+
+class TransformTypeResource(Resource):
+
+    def list(self) -> TransformTypeListResponse:
+        """Get Transform Types. ``databay transform-type list``."""
+        return self._call(_ops.TRANSFORM_TYPE_LIST, {})

databay/resources/usage.py

@@ -0,0 +1,21 @@
+"""``client.usage``: the ``usage`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from databay.operations import usage as _ops
+from databay.resource import Resource
+
+if TYPE_CHECKING:
+    from databay.models.usage import UsageView
+
+
+class UsageResource(Resource):
+
+    def get(self) -> UsageView:
+        """Get Usage. ``databay usage get``."""
+        return self._call(_ops.USAGE_GET, {})

databay/resources/warehouse.py

@@ -0,0 +1,136 @@
+"""``client.warehouse``: the ``warehouse`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Literal, overload
+
+from databay.operations import warehouse as _ops
+from databay.resource import Resource, Waited
+
+if TYPE_CHECKING:
+    from databay.models.shared import PublicDBResponse, WarehouseAcceptedResponse
+    from databay.models.warehouse import WarehouseImpactResponse, WarehouseListResponse, WarehouseStatsResponse
+
+
+class WarehouseResource(Resource):
+
+    def create(
+        self,
+        *,
+        name: str,
+        project: str | None = None,
+        idempotency_key: str | None = None,
+    ) -> WarehouseAcceptedResponse:
+        """Create Warehouse. ``databay warehouse create``."""
+        return self._call(_ops.WAREHOUSE_CREATE, {'name': name}, project=project, idempotency_key=idempotency_key)
+
+    def delete(
+        self,
+        *,
+        db_uuid: str,
+        project: str | None = None,
+    ) -> WarehouseAcceptedResponse:
+        """Delete Warehouse. ``databay warehouse delete``."""
+        return self._call(_ops.WAREHOUSE_DELETE, {'db_uuid': db_uuid}, project=project)
+
+    @overload
+    def get(
+        self,
+        *,
+        db_uuid: str,
+        project: str | None = None,
+        wait: Literal[False] = False,
+        timeout: None = None,
+    ) -> PublicDBResponse: ...
+
+
+    @overload
+    def get(
+        self,
+        *,
+        db_uuid: str,
+        project: str | None = None,
+        wait: Literal[True],
+        timeout: float | None = None,
+    ) -> Waited[PublicDBResponse]: ...
+
+    def get(
+        self,
+        *,
+        db_uuid: str,
+        project: str | None = None,
+        wait: bool = False,
+        timeout: float | None = None,
+    ) -> PublicDBResponse | Waited[PublicDBResponse]:
+        """Get Warehouse. ``databay warehouse get``."""
+        return self._call(_ops.WAREHOUSE_GET, {'db_uuid': db_uuid}, project=project, wait=wait, timeout=timeout)
+
+    def impact(
+        self,
+        *,
+        db_uuid: str,
+        project: str | None = None,
+    ) -> WarehouseImpactResponse:
+        """Warehouse Impact. ``databay warehouse impact``."""
+        return self._call(_ops.WAREHOUSE_IMPACT, {'db_uuid': db_uuid}, project=project)
+
+    def list(
+        self,
+        *,
+        project: str | None = None,
+    ) -> WarehouseListResponse:
+        """List Warehouses. ``databay warehouse list``."""
+        return self._call(_ops.WAREHOUSE_LIST, {}, project=project)
+
+    def rename(
+        self,
+        *,
+        db_uuid: str,
+        name: str,
+        project: str | None = None,
+    ) -> PublicDBResponse:
+        """Rename Warehouse. ``databay warehouse rename``."""
+        return self._call(_ops.WAREHOUSE_RENAME, {'db_uuid': db_uuid, 'name': name}, project=project)
+
+    @overload
+    def retry(
+        self,
+        *,
+        db_uuid: str,
+        project: str | None = None,
+        wait: Literal[False] = False,
+        timeout: None = None,
+    ) -> WarehouseAcceptedResponse: ...
+
+
+    @overload
+    def retry(
+        self,
+        *,
+        db_uuid: str,
+        project: str | None = None,
+        wait: Literal[True],
+        timeout: float | None = None,
+    ) -> Waited[WarehouseAcceptedResponse]: ...
+
+    def retry(
+        self,
+        *,
+        db_uuid: str,
+        project: str | None = None,
+        wait: bool = False,
+        timeout: float | None = None,
+    ) -> WarehouseAcceptedResponse | Waited[WarehouseAcceptedResponse]:
+        """Retry Warehouse. ``databay warehouse retry``."""
+        return self._call(_ops.WAREHOUSE_RETRY, {'db_uuid': db_uuid}, project=project, wait=wait, timeout=timeout)
+
+    def stats(
+        self,
+        *,
+        project: str | None = None,
+    ) -> WarehouseStatsResponse:
+        """Warehouse Stats. ``databay warehouse stats``."""
+        return self._call(_ops.WAREHOUSE_STATS, {}, project=project)

databay/resources/warehouse_access_assignment.py

@@ -0,0 +1,51 @@
+"""``client.warehouse_access_assignment``: the ``warehouse-access-assignment`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from databay.operations import warehouse_access_assignment as _ops
+from databay.resource import Resource, UNSET, Unset
+
+if TYPE_CHECKING:
+    from databay.models.warehouse_access_assignment import AccessPreviewOut, AssignmentOut, WarehouseAccessAssignmentListResponse
+
+
+class WarehouseAccessAssignmentResource(Resource):
+
+    def list(
+        self,
+        *,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> WarehouseAccessAssignmentListResponse:
+        """List Assignments. ``databay warehouse-access-assignment list``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ASSIGNMENT_LIST, {'project_id': project_id}, project=project)
+
+    def preview(
+        self,
+        *,
+        holder_id: str,
+        holder_kind: str,
+        role_ids: list[str],
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> AccessPreviewOut:
+        """Preview. ``databay warehouse-access-assignment preview``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ASSIGNMENT_PREVIEW, {'project_id': project_id, 'holder_id': holder_id, 'holder_kind': holder_kind, 'role_ids': role_ids}, project=project)
+
+    def set(
+        self,
+        *,
+        holder_id: str,
+        holder_kind: str,
+        revision: int,
+        role_ids: list[str],
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> AssignmentOut:
+        """Put Assignment. ``databay warehouse-access-assignment set``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ASSIGNMENT_SET, {'project_id': project_id, 'holder_id': holder_id, 'holder_kind': holder_kind, 'revision': revision, 'role_ids': role_ids}, project=project)

databay/resources/warehouse_access_role.py

@@ -0,0 +1,118 @@
+"""``client.warehouse_access_role``: the ``warehouse-access-role`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from databay.operations import warehouse_access_role as _ops
+from databay.resource import Resource, UNSET, Unset
+
+if TYPE_CHECKING:
+    from databay.models.warehouse_access_role import AccessRoleOut, RoleMetricsOut, WarehouseAccessRoleListResponse, WarehouseSchemaOut
+
+
+class WarehouseAccessRoleResource(Resource):
+
+    def create(
+        self,
+        *,
+        name: str,
+        project_id: str | Unset = UNSET,
+        description: str | Unset = UNSET,
+        project: str | None = None,
+        idempotency_key: str | None = None,
+    ) -> AccessRoleOut:
+        """Create Role. ``databay warehouse-access-role create``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ROLE_CREATE, {'project_id': project_id, 'description': description, 'name': name}, project=project, idempotency_key=idempotency_key)
+
+    def delete(
+        self,
+        *,
+        role_id: str,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> None:
+        """Delete Role. ``databay warehouse-access-role delete``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ROLE_DELETE, {'project_id': project_id, 'role_id': role_id}, project=project)
+
+    def get_metrics(
+        self,
+        *,
+        role_id: str,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> RoleMetricsOut:
+        """Get Metric Grant. ``databay warehouse-access-role get-metrics``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ROLE_GET_METRICS, {'project_id': project_id, 'role_id': role_id}, project=project)
+
+    def list(
+        self,
+        *,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> WarehouseAccessRoleListResponse:
+        """List Roles. ``databay warehouse-access-role list``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ROLE_LIST, {'project_id': project_id}, project=project)
+
+    def list_tables(
+        self,
+        *,
+        db_uuid: str,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> WarehouseSchemaOut:
+        """Warehouse Schema. ``databay warehouse-access-role list-tables``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ROLE_LIST_TABLES, {'project_id': project_id, 'db_uuid': db_uuid}, project=project)
+
+    def set_metrics(
+        self,
+        *,
+        role_id: str,
+        metrics: list[str],
+        mode: str,
+        revision: int,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> RoleMetricsOut:
+        """Put Metric Grant. ``databay warehouse-access-role set-metrics``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ROLE_SET_METRICS, {'project_id': project_id, 'role_id': role_id, 'metrics': metrics, 'mode': mode, 'revision': revision}, project=project)
+
+    def set_rules(
+        self,
+        *,
+        role_id: str,
+        revision: int,
+        rules: Any,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> AccessRoleOut:
+        """Put Rules. ``databay warehouse-access-role set-rules``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ROLE_SET_RULES, {'project_id': project_id, 'role_id': role_id, 'revision': revision, 'rules': rules}, project=project)
+
+    def set_scope_mode(
+        self,
+        *,
+        role_id: str,
+        revision: int,
+        scope_mode: str,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> AccessRoleOut:
+        """Set Scope Mode. ``databay warehouse-access-role set-scope-mode``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ROLE_SET_SCOPE_MODE, {'project_id': project_id, 'role_id': role_id, 'revision': revision, 'scope_mode': scope_mode}, project=project)
+
+    def update(
+        self,
+        *,
+        role_id: str,
+        revision: int,
+        project_id: str | Unset = UNSET,
+        description: str | None | Unset = UNSET,
+        name: str | None | Unset = UNSET,
+        project: str | None = None,
+    ) -> AccessRoleOut:
+        """Update Role. ``databay warehouse-access-role update``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_ROLE_UPDATE, {'project_id': project_id, 'role_id': role_id, 'description': description, 'name': name, 'revision': revision}, project=project)

databay/resources/warehouse_access_settings.py

@@ -0,0 +1,56 @@
+"""``client.warehouse_access_settings``: the ``warehouse-access-settings`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from databay.operations import warehouse_access_settings as _ops
+from databay.resource import Resource, UNSET, Unset
+
+if TYPE_CHECKING:
+    from databay.models.warehouse_access_settings import AccessSettingsOut, DefaultRoleOut
+
+
+class WarehouseAccessSettingsResource(Resource):
+
+    def get(
+        self,
+        *,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> AccessSettingsOut:
+        """Get Settings. ``databay warehouse-access-settings get``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_SETTINGS_GET, {'project_id': project_id}, project=project)
+
+    def get_default_role(
+        self,
+        *,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> DefaultRoleOut:
+        """Get Default Role. ``databay warehouse-access-settings get-default-role``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_SETTINGS_GET_DEFAULT_ROLE, {'project_id': project_id}, project=project)
+
+    def set(
+        self,
+        *,
+        personal_logins_enabled: bool,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> AccessSettingsOut:
+        """Put Settings. ``databay warehouse-access-settings set``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_SETTINGS_SET, {'project_id': project_id, 'personal_logins_enabled': personal_logins_enabled}, project=project)
+
+    def set_default_role(
+        self,
+        *,
+        revision: int,
+        role_id: str,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> DefaultRoleOut:
+        """Put Default Role. ``databay warehouse-access-settings set-default-role``."""
+        return self._call(_ops.WAREHOUSE_ACCESS_SETTINGS_SET_DEFAULT_ROLE, {'project_id': project_id, 'revision': revision, 'role_id': role_id}, project=project)

databay/resources/warehouse_login.py

@@ -0,0 +1,81 @@
+"""``client.warehouse_login``: the ``warehouse-login`` operations.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from databay.operations import warehouse_login as _ops
+from databay.resource import Resource, UNSET, Unset
+
+if TYPE_CHECKING:
+    from databay.models.shared import IssuedLoginOut, LoginOut
+    from databay.models.warehouse_login import LoginPolicyOut, WarehouseLoginListMineResponse, WarehouseLoginListResponse
+
+
+class WarehouseLoginResource(Resource):
+
+    def create(
+        self,
+        *,
+        db_uuid: str,
+        project_id: str | Unset = UNSET,
+        allowed_hosts: list[str] | None | Unset = UNSET,
+        valid_until: str | None | Unset = UNSET,
+        project: str | None = None,
+        idempotency_key: str | None = None,
+    ) -> IssuedLoginOut:
+        """Create Personal Login. ``databay warehouse-login create``."""
+        return self._call(_ops.WAREHOUSE_LOGIN_CREATE, {'project_id': project_id, 'allowed_hosts': allowed_hosts, 'db_uuid': db_uuid, 'valid_until': valid_until}, project=project, idempotency_key=idempotency_key)
+
+    def delete(
+        self,
+        *,
+        principal_id: str,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> None:
+        """Delete Personal Login. ``databay warehouse-login delete``."""
+        return self._call(_ops.WAREHOUSE_LOGIN_DELETE, {'project_id': project_id, 'principal_id': principal_id}, project=project)
+
+    def list(
+        self,
+        *,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+    ) -> WarehouseLoginListResponse:
+        """List Personal Logins. ``databay warehouse-login list``."""
+        return self._call(_ops.WAREHOUSE_LOGIN_LIST, {'project_id': project_id}, project=project)
+
+    def list_mine(self) -> WarehouseLoginListMineResponse:
+        """List Own Logins. ``databay warehouse-login list-mine``."""
+        return self._call(_ops.WAREHOUSE_LOGIN_LIST_MINE, {})
+
+    def policy(self) -> LoginPolicyOut:
+        """Get Login Policy. ``databay warehouse-login policy``."""
+        return self._call(_ops.WAREHOUSE_LOGIN_POLICY, {})
+
+    def rotate(
+        self,
+        *,
+        principal_id: str,
+        project_id: str | Unset = UNSET,
+        project: str | None = None,
+        idempotency_key: str | None = None,
+    ) -> IssuedLoginOut:
+        """Rotate Personal Login. ``databay warehouse-login rotate``."""
+        return self._call(_ops.WAREHOUSE_LOGIN_ROTATE, {'project_id': project_id, 'principal_id': principal_id}, project=project, idempotency_key=idempotency_key)
+
+    def update(
+        self,
+        *,
+        principal_id: str,
+        project_id: str | Unset = UNSET,
+        allowed_hosts: list[str] | None | Unset = UNSET,
+        valid_until: str | None | Unset = UNSET,
+        project: str | None = None,
+    ) -> LoginOut:
+        """Update Personal Login. ``databay warehouse-login update``."""
+        return self._call(_ops.WAREHOUSE_LOGIN_UPDATE, {'project_id': project_id, 'principal_id': principal_id, 'allowed_hosts': allowed_hosts, 'valid_until': valid_until}, project=project)