databay-cli 0.1.0__py3-none-any.whl

databay/models/ssbi_job.py

@@ -0,0 +1,97 @@
+"""Models of the ``ssbi_job`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+
+from databay.model_base import RequestModel, ResponseModel
+
+
+class ResourceProfileCreate(RequestModel):
+    driver_cpu: str | None = None
+    driver_memory: str | None = None
+    executor_count: int | None = None
+    executor_cpu: str | None = None
+    executor_memory: str | None = None
+    profile_name: str | None = None
+
+
+class ScheduleCreate(RequestModel):
+    cron_expression: str
+
+
+class TransformationCreate(RequestModel):
+    query: str
+    transform_type: str = 'query'
+
+
+class JobCreate(RequestModel):
+    destination_table: str
+    name: str
+    resource: ResourceProfileCreate
+    schedule: ScheduleCreate
+    transformation: TransformationCreate
+
+
+class ResourceProfileResponse(ResponseModel):
+    driver_cpu: str | None = None
+    driver_memory: str | None = None
+    executor_count: int | None = None
+    executor_cpu: str | None = None
+    executor_memory: str | None = None
+    profile_name: str | None = None
+    uuid: str
+
+
+class ScheduleResponse(ResponseModel):
+    cron_expression: str
+    uuid: str
+
+
+class TransformationResponse(ResponseModel):
+    query: str
+    transform_type: str = 'query'
+    uuid: str
+
+
+class JobResponse(ResponseModel):
+    created_at: datetime
+    destination_table: str
+    name: str
+    project_id: str | None = None
+    resource: ResourceProfileResponse
+    schedule: ScheduleResponse
+    transformation: TransformationResponse
+    updated_at: datetime
+    uuid: str
+
+
+class ResourceProfileUpdate(RequestModel):
+    driver_cpu: str | None = None
+    driver_memory: str | None = None
+    executor_count: int | None = None
+    executor_cpu: str | None = None
+    executor_memory: str | None = None
+    profile_name: str | None = None
+
+
+class ScheduleUpdate(RequestModel):
+    cron_expression: str
+
+
+class TransformationUpdate(RequestModel):
+    query: str
+    transform_type: str = 'query'
+
+
+class JobUpdate(RequestModel):
+    name: str | None = None
+    resource: ResourceProfileUpdate | None = None
+    schedule: ScheduleUpdate | None = None
+    transformation: TransformationUpdate | None = None
+
+
+SsbiJobListResponse = list[JobResponse]

databay/models/transform_type.py

@@ -0,0 +1,40 @@
+"""Models of the ``transform_type`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+from databay.model_base import OpenStrEnum, ResponseModel
+
+
+class TransformTypeResponseBand(OpenStrEnum):
+    PROJECT = 'project'
+    COLUMN_OPS = 'column_ops'
+    TABLE_STEPS = 'table_steps'
+    MASK = 'mask'
+
+
+class TransformTypeResponseScope(OpenStrEnum):
+    COLUMN = 'column'
+    TABLE = 'table'
+
+
+class TransformTypeResponseStatus(OpenStrEnum):
+    AVAILABLE = 'available'
+    COMING_SOON = 'coming_soon'
+
+
+class TransformTypeResponse(ResponseModel):
+    band: TransformTypeResponseBand
+    description: str
+    id: str
+    label: str
+    params_schema: dict[str, Any] | None = None
+    scope: TransformTypeResponseScope
+    status: TransformTypeResponseStatus
+
+
+TransformTypeListResponse = list[TransformTypeResponse]

databay/models/usage.py

@@ -0,0 +1,112 @@
+"""Models of the ``usage`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from databay.model_base import OpenStrEnum, ResponseModel
+from databay.models.shared import Meter
+
+
+class MeterKind(OpenStrEnum):
+    COUNT = 'count'
+    FLOW = 'flow'
+    BOUND = 'bound'
+
+
+class Mode(OpenStrEnum):
+    SHADOW = 'shadow'
+    ENFORCE = 'enforce'
+
+
+class OverageMode(OpenStrEnum):
+    PAUSE = 'pause'
+    PAYG = 'payg'
+
+
+class PlanSource(OpenStrEnum):
+    ENTITLEMENTS = 'entitlements'
+    SHOPIFY = 'shopify'
+
+
+class Rule(OpenStrEnum):
+    HARD = 'hard'
+    SOFT = 'soft'
+    GRACE = 'grace'
+
+
+class BreakdownLine(ResponseModel):
+    pipe_uuid: str | None = None
+    project_id: str | None
+    value: int
+
+
+class MeterUsage(ResponseModel):
+    blocking_limit: int | None = None
+    breakdown: list[BreakdownLine] | None = None
+    current: int | None = None
+    external_plan: bool = False
+    forecast_at: str | None = None
+    kind: MeterKind
+    level: str = 'ok'
+    limit: int | None = None
+    meter: Meter
+    mode: Mode
+    overage_units: int = 0
+    quota_resume: int = 0
+    rule: Rule
+    status: str
+    unit: str
+    unlimited: bool = False
+    updated_at: str | None = None
+
+
+class NoticeView(ResponseModel):
+    enforce_from: str
+    meter: Meter
+
+
+class PeriodView(ResponseModel):
+    end: str
+    key: str
+    start: str
+
+
+class PlanNoticeView(ResponseModel):
+    effective_at: str
+    from_plan: str
+    from_version: int
+    pending: bool
+    to_label: str
+    to_plan: str
+    to_version: int
+
+
+class PlanRef(ResponseModel):
+    key: str
+    label: str
+    metric_label: str
+    version: int
+
+
+class QuotaStateView(ResponseModel):
+    level: str
+    meter: Meter
+    period_end: str
+    scope_id: str
+    scope_kind: str
+
+
+class UsageView(ResponseModel):
+    is_paid: bool = False
+    meters: list[MeterUsage] | None = None
+    notices: list[NoticeView] | None = None
+    organization: str
+    overage_mode: OverageMode | None = None
+    period: PeriodView
+    plan: PlanRef | None = None
+    plan_notices: list[PlanNoticeView] | None = None
+    plan_source: PlanSource | None = None
+    quota_states: list[QuotaStateView] | None = None
+    state: str

databay/models/warehouse.py

@@ -0,0 +1,38 @@
+"""Models of the ``warehouse`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from databay.model_base import RequestModel, ResponseModel
+from databay.models.shared import PublicDBResponse
+
+
+class WarehouseImpactPipeline(ResponseModel):
+    name: str
+    pipe_uuid: str
+    status: str
+
+
+class WarehouseImpactResponse(ResponseModel):
+    hidden_pipeline_count: int = 0
+    pipelines: list[WarehouseImpactPipeline] | None = None
+
+
+class WarehouseNameBody(RequestModel):
+    name: str
+
+
+class WarehouseStats(ResponseModel):
+    size_bytes: int
+    table_count: int
+    uuid: str
+
+
+class WarehouseStatsResponse(ResponseModel):
+    available: bool
+    databases: list[WarehouseStats] | None = None
+
+
+WarehouseListResponse = list[PublicDBResponse]

databay/models/warehouse_access_assignment.py

@@ -0,0 +1,70 @@
+"""Models of the ``warehouse_access_assignment`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from databay.model_base import RequestModel, ResponseModel
+from databay.models.shared import AllOfOutput, AnyOfOutput, HolderKind, RowCondition
+
+
+class AccessPreviewIn(RequestModel):
+    holder_id: str
+    holder_kind: HolderKind
+    role_ids: list[str]
+
+
+class PreviewConflict(ResponseModel):
+    db_uuid: str
+    role_ids: list[str]
+    table_name: str
+
+
+class PreviewRule(ResponseModel):
+    columns: list[str] | None
+    role_id: str
+    row_filter: RowCondition | AllOfOutput | AnyOfOutput | None
+
+
+class PreviewTable(ResponseModel):
+    rules: list[PreviewRule]
+    table_name: str
+
+
+class PreviewWarehouse(ResponseModel):
+    db_uuid: str
+    name: str
+    tables: list[PreviewTable]
+    whole_warehouse_roles: list[str]
+
+
+class PreviewSide(ResponseModel):
+    conflicts: list[PreviewConflict]
+    has_project_access: bool
+    role_ids: list[str]
+    warehouses: list[PreviewWarehouse]
+
+
+class AccessPreviewOut(ResponseModel):
+    after: PreviewSide
+    before: PreviewSide
+    holder_id: str
+    holder_kind: HolderKind
+
+
+class AssignmentOut(ResponseModel):
+    holder_id: str
+    holder_kind: HolderKind
+    revision: int
+    role_ids: list[str]
+
+
+class AssignmentPut(RequestModel):
+    holder_id: str
+    holder_kind: HolderKind
+    revision: int
+    role_ids: list[str]
+
+
+WarehouseAccessAssignmentListResponse = list[AssignmentOut]

databay/models/warehouse_access_role.py

@@ -0,0 +1,148 @@
+"""Models of the ``warehouse_access_role`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+
+from databay.model_base import OpenStrEnum, RequestModel, ResponseModel
+from databay.models.shared import AccessScopeMode, AllOfOutput, AnyOfOutput, ReconcileOut, RowCondition
+
+
+class AccessRoleOrigin(OpenStrEnum):
+    USER = 'user'
+    PROJECT_DEFAULT = 'project_default'
+    LEGACY_SINK_USER = 'legacy_sink_user'
+
+
+class AccessRoleState(OpenStrEnum):
+    ACTIVE = 'active'
+    DROPPING = 'dropping'
+
+
+class MetricGrantMode(OpenStrEnum):
+    INCLUDE = 'include'
+    EXCLUDE = 'exclude'
+
+
+class RuleStatus(OpenStrEnum):
+    ACTIVE = 'active'
+    INACTIVE = 'inactive'
+    STALE = 'stale'
+
+
+class WriterKind(OpenStrEnum):
+    BUILTIN_WRITER = 'builtin_writer'
+    MODEL_WRITER = 'model_writer'
+
+
+class AccessRoleCreate(RequestModel):
+    description: str = ''
+    name: str
+
+
+class AccessRuleOut(ResponseModel):
+    auto_granted: bool
+    built_database: str | None
+    built_writer_kind: WriterKind | None
+    columns: list[str] | None
+    db_uuid: str
+    row_filter: RowCondition | AllOfOutput | AnyOfOutput | None
+    status: RuleStatus
+    status_reason: str | None
+    table_name: str | None
+    uuid: str
+
+
+class AccessRoleOut(ResponseModel):
+    created_at: datetime
+    created_by: str | None
+    description: str
+    is_default: bool
+    name: str
+    origin: AccessRoleOrigin
+    project_id: str
+    reconcile: ReconcileOut
+    revision: int
+    rules: list[AccessRuleOut]
+    scope_mode: AccessScopeMode
+    state: AccessRoleState
+    updated_at: datetime
+    uuid: str
+
+
+class AccessRolePatch(RequestModel):
+    description: str | None = None
+    name: str | None = None
+    revision: int
+
+
+class AnyOfInput(RequestModel):
+    any: list[RowCondition | AllOfInput | AnyOfInput]
+
+
+class AllOfInput(RequestModel):
+    all: list[RowCondition | AllOfInput | AnyOfInput]
+
+
+class AccessRuleIn(RequestModel):
+    built_database: str | None = None
+    columns: list[str] | None = None
+    db_uuid: str
+    row_filter: RowCondition | AllOfInput | AnyOfInput | None = None
+    table_name: str | None = None
+
+
+class AccessRulesPut(RequestModel):
+    revision: int
+    rules: list[AccessRuleIn]
+
+
+class AccessScopeModePut(RequestModel):
+    revision: int
+    scope_mode: AccessScopeMode
+
+
+class MetricGrant(ResponseModel):
+    metrics: list[str]
+    mode: MetricGrantMode
+
+
+class MetricGrantPut(RequestModel):
+    metrics: list[str]
+    mode: MetricGrantMode
+    revision: int
+
+
+class RoleMetricsOut(ResponseModel):
+    grant: MetricGrant | None
+    revision: int
+    role_id: str
+
+
+class SchemaColumnOut(ResponseModel):
+    name: str
+    type: str
+
+
+class SchemaBuiltRelationOut(ResponseModel):
+    columns: list[SchemaColumnOut]
+    database: str
+    name: str
+    writer_kind: WriterKind
+
+
+class SchemaTableOut(ResponseModel):
+    columns: list[SchemaColumnOut]
+    name: str
+
+
+class WarehouseSchemaOut(ResponseModel):
+    built_relations: list[SchemaBuiltRelationOut]
+    db_uuid: str
+    tables: list[SchemaTableOut]
+
+
+WarehouseAccessRoleListResponse = list[AccessRoleOut]

databay/models/warehouse_access_settings.py

@@ -0,0 +1,35 @@
+"""Models of the ``warehouse_access_settings`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from databay.model_base import RequestModel, ResponseModel
+from databay.models.shared import AccessScopeMode
+
+
+class DefaultRoleOut(ResponseModel):
+    project_id: str
+    revision: int
+    role_id: str | None
+    rule_count: int
+    scope_mode: AccessScopeMode | None
+    visible_warehouse_count: int
+
+
+class AccessSettingsOut(ResponseModel):
+    active_personal_logins: int
+    default_reads_all: bool
+    default_role: DefaultRoleOut
+    personal_logins_enabled: bool
+    project_id: str
+
+
+class AccessSettingsPut(RequestModel):
+    personal_logins_enabled: bool
+
+
+class DefaultRolePut(RequestModel):
+    revision: int
+    role_id: str | None

databay/models/warehouse_login.py

@@ -0,0 +1,20 @@
+"""Models of the ``warehouse_login`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from databay.model_base import ResponseModel
+from databay.models.shared import LoginOut
+
+
+class LoginPolicyOut(ResponseModel):
+    expiry_notice_days: int
+    max_valid_days: int
+
+
+WarehouseLoginListResponse = list[LoginOut]
+
+
+WarehouseLoginListMineResponse = list[LoginOut]

databay/models/warehouse_principal.py

@@ -0,0 +1,49 @@
+"""Models of the ``warehouse_principal`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+
+from databay.model_base import RequestModel, ResponseModel
+from databay.models.shared import HolderKind, PrincipalKind, PrincipalState, ReconcileOut
+
+
+class DisableIn(RequestModel):
+    reason: str
+
+
+class OrgLoginOut(ResponseModel):
+    allowed_hosts: list[str] | None
+    created_at: datetime
+    database: str | None
+    db_uuid: str
+    disabled_at: datetime | None
+    disabled_by: str | None
+    disabled_reason: str | None
+    holder_id: str
+    holder_kind: HolderKind
+    kind: PrincipalKind
+    last_rotated_at: datetime | None
+    last_used_at: datetime | None
+    native_username: str
+    principal_id: str
+    project_id: str
+    reconcile: ReconcileOut
+    service_account_name: str | None
+    state: PrincipalState
+    valid_until: datetime | None
+    warehouse_name: str | None
+
+
+class PrincipalStateOut(ResponseModel):
+    principal_id: str
+    state: str
+
+
+WarehousePrincipalListProjectResponse = list[OrgLoginOut]
+
+
+WarehousePrincipalListResponse = list[OrgLoginOut]

databay/models/warehouse_service_account.py

@@ -0,0 +1,35 @@
+"""Models of the ``warehouse_service_account`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+
+from databay.model_base import RequestModel, ResponseModel
+
+
+class ServiceAccountCreate(RequestModel):
+    description: str = ''
+    name: str
+    role_ids: list[str]
+
+
+class ServiceAccountOut(ResponseModel):
+    created_at: datetime
+    created_by: str
+    credentials: int
+    description: str
+    name: str
+    project_id: str
+    role_ids: list[str]
+    uuid: str
+
+
+class ServiceAccountPatch(RequestModel):
+    description: str | None = None
+    name: str | None = None
+
+
+WarehouseServiceAccountListResponse = list[ServiceAccountOut]

databay/models/warehouse_service_account_credential.py

@@ -0,0 +1,11 @@
+"""Models of the ``warehouse_service_account_credential`` noun.
+
+Generated by python/tools/gen_client.py from contract/core.json; do not edit.
+"""
+
+from __future__ import annotations
+
+from databay.models.shared import LoginOut
+
+
+WarehouseServiceAccountCredentialListResponse = list[LoginOut]