convexity-api-client 0.4.3.dev152__tar.gz → 0.5.0__tar.gz

{convexity_api_client-0.4.3.dev152 → convexity_api_client-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: convexity_api_client
-Version: 0.4.3.dev152
+Version: 0.5.0
 Summary: A client library for accessing Convexity API
 Requires-Dist: httpx>=0.23.0,<0.29.0
 Requires-Dist: attrs>=22.2.0

convexity_api_client-0.5.0/convexity_api_client/api/v1/cancel_refresh_task_v1_datasets_dataset_id_refresh_task_id_cancel_post.py

@@ -0,0 +1,183 @@
+from http import HTTPStatus
+from typing import Any
+from urllib.parse import quote
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.http_validation_error import HTTPValidationError
+from ...models.worker_task_response import WorkerTaskResponse
+from ...types import Response
+
+
+def _get_kwargs(
+    dataset_id: str,
+    task_id: str,
+) -> dict[str, Any]:
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/v1/datasets/{dataset_id}/refresh/{task_id}/cancel".format(
+            dataset_id=quote(str(dataset_id), safe=""),
+            task_id=quote(str(task_id), safe=""),
+        ),
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> HTTPValidationError | WorkerTaskResponse | None:
+    if response.status_code == 200:
+        response_200 = WorkerTaskResponse.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 422:
+        response_422 = HTTPValidationError.from_dict(response.json())
+
+        return response_422
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[HTTPValidationError | WorkerTaskResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    dataset_id: str,
+    task_id: str,
+    *,
+    client: AuthenticatedClient | Client,
+) -> Response[HTTPValidationError | WorkerTaskResponse]:
+    """Cancel Refresh Task
+
+     Cancel a pending or running refresh task for a dataset. Requires member role.
+
+    Args:
+        dataset_id (str):
+        task_id (str):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[HTTPValidationError | WorkerTaskResponse]
+    """
+
+    kwargs = _get_kwargs(
+        dataset_id=dataset_id,
+        task_id=task_id,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    dataset_id: str,
+    task_id: str,
+    *,
+    client: AuthenticatedClient | Client,
+) -> HTTPValidationError | WorkerTaskResponse | None:
+    """Cancel Refresh Task
+
+     Cancel a pending or running refresh task for a dataset. Requires member role.
+
+    Args:
+        dataset_id (str):
+        task_id (str):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        HTTPValidationError | WorkerTaskResponse
+    """
+
+    return sync_detailed(
+        dataset_id=dataset_id,
+        task_id=task_id,
+        client=client,
+    ).parsed
+
+
+async def asyncio_detailed(
+    dataset_id: str,
+    task_id: str,
+    *,
+    client: AuthenticatedClient | Client,
+) -> Response[HTTPValidationError | WorkerTaskResponse]:
+    """Cancel Refresh Task
+
+     Cancel a pending or running refresh task for a dataset. Requires member role.
+
+    Args:
+        dataset_id (str):
+        task_id (str):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[HTTPValidationError | WorkerTaskResponse]
+    """
+
+    kwargs = _get_kwargs(
+        dataset_id=dataset_id,
+        task_id=task_id,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    dataset_id: str,
+    task_id: str,
+    *,
+    client: AuthenticatedClient | Client,
+) -> HTTPValidationError | WorkerTaskResponse | None:
+    """Cancel Refresh Task
+
+     Cancel a pending or running refresh task for a dataset. Requires member role.
+
+    Args:
+        dataset_id (str):
+        task_id (str):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        HTTPValidationError | WorkerTaskResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            dataset_id=dataset_id,
+            task_id=task_id,
+            client=client,
+        )
+    ).parsed

{convexity_api_client-0.4.3.dev152 → convexity_api_client-0.5.0}/convexity_api_client/models/__init__.py

@@ -139,6 +139,7 @@ from .date_range_value import DateRangeValue
 from .delete_connection_v1_connections_connection_id_delete_response_delete_connection_v1_connections_connection_id_delete import (
     DeleteConnectionV1ConnectionsConnectionIdDeleteResponseDeleteConnectionV1ConnectionsConnectionIdDelete,
 )
+from .delete_dataset_payload import DeleteDatasetPayload
 from .delete_file_request import DeleteFileRequest
 from .delete_file_response import DeleteFileResponse
 from .delete_run_response import DeleteRunResponse
@@ -223,6 +224,8 @@ from .public_dashboard_response import PublicDashboardResponse
 from .public_dashboard_response_filters_type_0_item import PublicDashboardResponseFiltersType0Item
 from .read_file_request import ReadFileRequest
 from .read_file_response import ReadFileResponse
+from .refresh_dataset_payload import RefreshDatasetPayload
+from .refresh_dataset_payload_parameters_type_0 import RefreshDatasetPayloadParametersType0
 from .refresh_mode import RefreshMode
 from .refresh_task import RefreshTask
 from .refresh_task_list_response import RefreshTaskListResponse
@@ -251,16 +254,15 @@ from .series_style_line_style_type_0 import SeriesStyleLineStyleType0
 from .session_info import SessionInfo
 from .share_token_exchange_request import ShareTokenExchangeRequest
 from .spreadsheet_preview import SpreadsheetPreview
-from .sql_dataset import SqlDataset
 from .sql_dataset_config import SqlDatasetConfig
 from .sql_dataset_config_parameters_type_0 import SqlDatasetConfigParametersType0
-from .sql_dataset_rows_item import SqlDatasetRowsItem
 from .sql_query_request import SqlQueryRequest
 from .sql_query_response import SqlQueryResponse
 from .sql_query_response_rows_item import SqlQueryResponseRowsItem
 from .start_run_response import StartRunResponse
 from .static_dataset_config import StaticDatasetConfig
 from .static_dataset_config_data_item import StaticDatasetConfigDataItem
+from .sync_catalog_to_s3_payload import SyncCatalogToS3Payload
 from .table_info import TableInfo
 from .table_list_response import TableListResponse
 from .task_connection import TaskConnection
@@ -297,6 +299,7 @@ from .update_share_request import UpdateShareRequest
 from .update_task_request import UpdateTaskRequest
 from .update_task_request_parameter_schema_type_0 import UpdateTaskRequestParameterSchemaType0
 from .update_webhook_connection_request import UpdateWebhookConnectionRequest
+from .vacuum_ducklake_payload import VacuumDucklakePayload
 from .validation_error import ValidationError
 from .verify_result import VerifyResult
 from .visual_encoding_config import VisualEncodingConfig
@@ -453,6 +456,7 @@ __all__ = (
     "DatasetType",
     "DateRangeValue",
     "DeleteConnectionV1ConnectionsConnectionIdDeleteResponseDeleteConnectionV1ConnectionsConnectionIdDelete",
+    "DeleteDatasetPayload",
     "DeleteFileRequest",
     "DeleteFileResponse",
     "DeleteRunResponse",
@@ -537,6 +541,8 @@ __all__ = (
     "PublicDashboardResponseFiltersType0Item",
     "ReadFileRequest",
     "ReadFileResponse",
+    "RefreshDatasetPayload",
+    "RefreshDatasetPayloadParametersType0",
     "RefreshMode",
     "RefreshTask",
     "RefreshTaskListResponse",
@@ -565,16 +571,15 @@ __all__ = (
     "SessionInfo",
     "ShareTokenExchangeRequest",
     "SpreadsheetPreview",
-    "SqlDataset",
     "SqlDatasetConfig",
     "SqlDatasetConfigParametersType0",
-    "SqlDatasetRowsItem",
     "SqlQueryRequest",
     "SqlQueryResponse",
     "SqlQueryResponseRowsItem",
     "StartRunResponse",
     "StaticDatasetConfig",
     "StaticDatasetConfigDataItem",
+    "SyncCatalogToS3Payload",
     "TableInfo",
     "TableListResponse",
     "TaskConnection",
@@ -611,6 +616,7 @@ __all__ = (
     "UpdateTaskRequest",
     "UpdateTaskRequestParameterSchemaType0",
     "UpdateWebhookConnectionRequest",
+    "VacuumDucklakePayload",
     "ValidationError",
     "VerifyResult",
     "VisualEncodingConfig",

{convexity_api_client-0.4.3.dev152 → convexity_api_client-0.5.0}/convexity_api_client/models/dashboard_chart_widget.py

@@ -26,6 +26,7 @@ class DashboardChartWidget:
         show_title (bool | None | Unset): Whether to display the widget title Default: True.
         show_description (bool | None | Unset): Whether to display the widget description Default: True.
         type_ (Literal['chart'] | Unset):  Default: 'chart'.
+        dataset_id (None | str | Unset): Dataset ID for the chart
         chart_config (ChartConfig | None | Unset): Chart configuration
     """
 
@@ -35,6 +36,7 @@ class DashboardChartWidget:
     show_title: bool | None | Unset = True
     show_description: bool | None | Unset = True
     type_: Literal["chart"] | Unset = "chart"
+    dataset_id: None | str | Unset = UNSET
     chart_config: ChartConfig | None | Unset = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -69,6 +71,12 @@ class DashboardChartWidget:
 
         type_ = self.type_
 
+        dataset_id: None | str | Unset
+        if isinstance(self.dataset_id, Unset):
+            dataset_id = UNSET
+        else:
+            dataset_id = self.dataset_id
+
         chart_config: dict[str, Any] | None | Unset
         if isinstance(self.chart_config, Unset):
             chart_config = UNSET
@@ -94,6 +102,8 @@ class DashboardChartWidget:
             field_dict["showDescription"] = show_description
         if type_ is not UNSET:
             field_dict["type"] = type_
+        if dataset_id is not UNSET:
+            field_dict["datasetId"] = dataset_id
         if chart_config is not UNSET:
             field_dict["chartConfig"] = chart_config
 
@@ -146,6 +156,15 @@ class DashboardChartWidget:
         if type_ != "chart" and not isinstance(type_, Unset):
             raise ValueError(f"type must match const 'chart', got '{type_}'")
 
+        def _parse_dataset_id(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        dataset_id = _parse_dataset_id(d.pop("datasetId", UNSET))
+
         def _parse_chart_config(data: object) -> ChartConfig | None | Unset:
             if data is None:
                 return data
@@ -170,6 +189,7 @@ class DashboardChartWidget:
             show_title=show_title,
             show_description=show_description,
             type_=type_,
+            dataset_id=dataset_id,
             chart_config=chart_config,
         )
 

{convexity_api_client-0.4.3.dev152 → convexity_api_client-0.5.0}/convexity_api_client/models/dashboard_sql_chart_widget.py

@@ -10,7 +10,6 @@ from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
     from ..models.chart_config import ChartConfig
-    from ..models.sql_dataset import SqlDataset
 
 
 T = TypeVar("T", bound="DashboardSqlChartWidget")
@@ -18,7 +17,7 @@ T = TypeVar("T", bound="DashboardSqlChartWidget")
 
 @_attrs_define
 class DashboardSqlChartWidget:
-    """SQL Chart widget definition.
+    """SQL Chart widget definition. Executes SQL queries against the DuckDB datalake.
 
     Attributes:
         id (str): Unique widget identifier
@@ -27,13 +26,9 @@ class DashboardSqlChartWidget:
         show_title (bool | None | Unset): Whether to display the widget title Default: True.
         show_description (bool | None | Unset): Whether to display the widget description Default: True.
         type_ (Literal['sqlchart'] | Unset):  Default: 'sqlchart'.
-        connection_id (None | str | Unset): Database connection ID
-        query (None | str | Unset): SQL query
+        query (None | str | Unset): SQL query to execute against the datalake
         row_limit (int | None | Unset): Row limit for query
-        sql_dataset (None | SqlDataset | Unset): SQL query result data
         chart_config (ChartConfig | None | Unset): Chart configuration
-        last_error (None | str | Unset): Last error message
-        last_run_at (None | str | Unset): Last run timestamp
     """
 
     id: str
@@ -42,18 +37,13 @@ class DashboardSqlChartWidget:
     show_title: bool | None | Unset = True
     show_description: bool | None | Unset = True
     type_: Literal["sqlchart"] | Unset = "sqlchart"
-    connection_id: None | str | Unset = UNSET
     query: None | str | Unset = UNSET
     row_limit: int | None | Unset = UNSET
-    sql_dataset: None | SqlDataset | Unset = UNSET
     chart_config: ChartConfig | None | Unset = UNSET
-    last_error: None | str | Unset = UNSET
-    last_run_at: None | str | Unset = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         from ..models.chart_config import ChartConfig
-        from ..models.sql_dataset import SqlDataset
 
         id = self.id
 
@@ -83,12 +73,6 @@ class DashboardSqlChartWidget:
 
         type_ = self.type_
 
-        connection_id: None | str | Unset
-        if isinstance(self.connection_id, Unset):
-            connection_id = UNSET
-        else:
-            connection_id = self.connection_id
-
         query: None | str | Unset
         if isinstance(self.query, Unset):
             query = UNSET
@@ -101,14 +85,6 @@ class DashboardSqlChartWidget:
         else:
             row_limit = self.row_limit
 
-        sql_dataset: dict[str, Any] | None | Unset
-        if isinstance(self.sql_dataset, Unset):
-            sql_dataset = UNSET
-        elif isinstance(self.sql_dataset, SqlDataset):
-            sql_dataset = self.sql_dataset.to_dict()
-        else:
-            sql_dataset = self.sql_dataset
-
         chart_config: dict[str, Any] | None | Unset
         if isinstance(self.chart_config, Unset):
             chart_config = UNSET
@@ -117,18 +93,6 @@ class DashboardSqlChartWidget:
         else:
             chart_config = self.chart_config
 
-        last_error: None | str | Unset
-        if isinstance(self.last_error, Unset):
-            last_error = UNSET
-        else:
-            last_error = self.last_error
-
-        last_run_at: None | str | Unset
-        if isinstance(self.last_run_at, Unset):
-            last_run_at = UNSET
-        else:
-            last_run_at = self.last_run_at
-
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -146,27 +110,18 @@ class DashboardSqlChartWidget:
             field_dict["showDescription"] = show_description
         if type_ is not UNSET:
             field_dict["type"] = type_
-        if connection_id is not UNSET:
-            field_dict["connectionId"] = connection_id
         if query is not UNSET:
             field_dict["query"] = query
         if row_limit is not UNSET:
             field_dict["rowLimit"] = row_limit
-        if sql_dataset is not UNSET:
-            field_dict["sqlDataset"] = sql_dataset
         if chart_config is not UNSET:
             field_dict["chartConfig"] = chart_config
-        if last_error is not UNSET:
-            field_dict["lastError"] = last_error
-        if last_run_at is not UNSET:
-            field_dict["lastRunAt"] = last_run_at
 
         return field_dict
 
     @classmethod
     def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
         from ..models.chart_config import ChartConfig
-        from ..models.sql_dataset import SqlDataset
 
         d = dict(src_dict)
         id = d.pop("id")
@@ -211,15 +166,6 @@ class DashboardSqlChartWidget:
         if type_ != "sqlchart" and not isinstance(type_, Unset):
             raise ValueError(f"type must match const 'sqlchart', got '{type_}'")
 
-        def _parse_connection_id(data: object) -> None | str | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(None | str | Unset, data)
-
-        connection_id = _parse_connection_id(d.pop("connectionId", UNSET))
-
         def _parse_query(data: object) -> None | str | Unset:
             if data is None:
                 return data
@@ -238,23 +184,6 @@ class DashboardSqlChartWidget:
 
         row_limit = _parse_row_limit(d.pop("rowLimit", UNSET))
 
-        def _parse_sql_dataset(data: object) -> None | SqlDataset | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                sql_dataset_type_0 = SqlDataset.from_dict(data)
-
-                return sql_dataset_type_0
-            except (TypeError, ValueError, AttributeError, KeyError):
-                pass
-            return cast(None | SqlDataset | Unset, data)
-
-        sql_dataset = _parse_sql_dataset(d.pop("sqlDataset", UNSET))
-
         def _parse_chart_config(data: object) -> ChartConfig | None | Unset:
             if data is None:
                 return data
@@ -272,24 +201,6 @@ class DashboardSqlChartWidget:
 
         chart_config = _parse_chart_config(d.pop("chartConfig", UNSET))
 
-        def _parse_last_error(data: object) -> None | str | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(None | str | Unset, data)
-
-        last_error = _parse_last_error(d.pop("lastError", UNSET))
-
-        def _parse_last_run_at(data: object) -> None | str | Unset:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(None | str | Unset, data)
-
-        last_run_at = _parse_last_run_at(d.pop("lastRunAt", UNSET))
-
         dashboard_sql_chart_widget = cls(
             id=id,
             title=title,
@@ -297,13 +208,9 @@ class DashboardSqlChartWidget:
             show_title=show_title,
             show_description=show_description,
             type_=type_,
-            connection_id=connection_id,
             query=query,
             row_limit=row_limit,
-            sql_dataset=sql_dataset,
             chart_config=chart_config,
-            last_error=last_error,
-            last_run_at=last_run_at,
         )
 
         dashboard_sql_chart_widget.additional_properties = d

convexity_api_client-0.5.0/convexity_api_client/models/delete_dataset_payload.py

@@ -0,0 +1,86 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, Literal, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="DeleteDatasetPayload")
+
+
+@_attrs_define
+class DeleteDatasetPayload:
+    """Payload for deleting a dataset's DuckLake table.
+
+    Drops the DuckLake table associated with the dataset.
+    Parquet data files remain on S3 until VACUUM is run.
+
+        Attributes:
+            dataset_id (str): ID of the dataset whose DuckLake table to delete
+            project_id (str): Project the dataset belongs to
+            type_ (Literal['delete_dataset'] | Unset):  Default: 'delete_dataset'.
+    """
+
+    dataset_id: str
+    project_id: str
+    type_: Literal["delete_dataset"] | Unset = "delete_dataset"
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        dataset_id = self.dataset_id
+
+        project_id = self.project_id
+
+        type_ = self.type_
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "datasetId": dataset_id,
+                "projectId": project_id,
+            }
+        )
+        if type_ is not UNSET:
+            field_dict["type"] = type_
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        dataset_id = d.pop("datasetId")
+
+        project_id = d.pop("projectId")
+
+        type_ = cast(Literal["delete_dataset"] | Unset, d.pop("type", UNSET))
+        if type_ != "delete_dataset" and not isinstance(type_, Unset):
+            raise ValueError(f"type must match const 'delete_dataset', got '{type_}'")
+
+        delete_dataset_payload = cls(
+            dataset_id=dataset_id,
+            project_id=project_id,
+            type_=type_,
+        )
+
+        delete_dataset_payload.additional_properties = d
+        return delete_dataset_payload
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties