windmill-api 1.528.0__py3-none-any.whl → 1.530.0__py3-none-any.whl

windmill_api/api/job/run_flow_preview_and_wait_result.py

@@ -0,0 +1,107 @@
+from http import HTTPStatus
+from typing import Any, Dict, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.run_flow_preview_and_wait_result_json_body import RunFlowPreviewAndWaitResultJsonBody
+from ...types import Response
+
+
+def _get_kwargs(
+    workspace: str,
+    *,
+    json_body: RunFlowPreviewAndWaitResultJsonBody,
+) -> Dict[str, Any]:
+    pass
+
+    json_json_body = json_body.to_dict()
+
+    return {
+        "method": "post",
+        "url": "/w/{workspace}/jobs/run_wait_result/preview_flow".format(
+            workspace=workspace,
+        ),
+        "json": json_json_body,
+    }
+
+
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Any]:
+    if response.status_code == HTTPStatus.OK:
+        return None
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    workspace: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+    json_body: RunFlowPreviewAndWaitResultJsonBody,
+) -> Response[Any]:
+    """run flow preview and wait for result
+
+    Args:
+        workspace (str):
+        json_body (RunFlowPreviewAndWaitResultJsonBody):
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        workspace=workspace,
+        json_body=json_body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    workspace: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+    json_body: RunFlowPreviewAndWaitResultJsonBody,
+) -> Response[Any]:
+    """run flow preview and wait for result
+
+    Args:
+        workspace (str):
+        json_body (RunFlowPreviewAndWaitResultJsonBody):
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        workspace=workspace,
+        json_body=json_body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)

windmill_api/api/job/run_script_preview_and_wait_result.py

@@ -0,0 +1,107 @@
+from http import HTTPStatus
+from typing import Any, Dict, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.run_script_preview_and_wait_result_json_body import RunScriptPreviewAndWaitResultJsonBody
+from ...types import Response
+
+
+def _get_kwargs(
+    workspace: str,
+    *,
+    json_body: RunScriptPreviewAndWaitResultJsonBody,
+) -> Dict[str, Any]:
+    pass
+
+    json_json_body = json_body.to_dict()
+
+    return {
+        "method": "post",
+        "url": "/w/{workspace}/jobs/run_wait_result/preview".format(
+            workspace=workspace,
+        ),
+        "json": json_json_body,
+    }
+
+
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Any]:
+    if response.status_code == HTTPStatus.OK:
+        return None
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    workspace: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+    json_body: RunScriptPreviewAndWaitResultJsonBody,
+) -> Response[Any]:
+    """run script preview and wait for result
+
+    Args:
+        workspace (str):
+        json_body (RunScriptPreviewAndWaitResultJsonBody):
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        workspace=workspace,
+        json_body=json_body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    workspace: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+    json_body: RunScriptPreviewAndWaitResultJsonBody,
+) -> Response[Any]:
+    """run script preview and wait for result
+
+    Args:
+        workspace (str):
+        json_body (RunScriptPreviewAndWaitResultJsonBody):
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        workspace=workspace,
+        json_body=json_body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)

windmill_api/api/worker/get_counts_of_running_jobs_per_tag.py

@@ -0,0 +1,126 @@
+from http import HTTPStatus
+from typing import Any, Dict, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.get_counts_of_running_jobs_per_tag_response_200 import GetCountsOfRunningJobsPerTagResponse200
+from ...types import Response
+
+
+def _get_kwargs() -> Dict[str, Any]:
+    pass
+
+    return {
+        "method": "get",
+        "url": "/workers/queue_running_counts",
+    }
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[GetCountsOfRunningJobsPerTagResponse200]:
+    if response.status_code == HTTPStatus.OK:
+        response_200 = GetCountsOfRunningJobsPerTagResponse200.from_dict(response.json())
+
+        return response_200
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[GetCountsOfRunningJobsPerTagResponse200]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: Union[AuthenticatedClient, Client],
+) -> Response[GetCountsOfRunningJobsPerTagResponse200]:
+    """get counts of currently running jobs per tag
+
+    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[GetCountsOfRunningJobsPerTagResponse200]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: Union[AuthenticatedClient, Client],
+) -> Optional[GetCountsOfRunningJobsPerTagResponse200]:
+    """get counts of currently running jobs per tag
+
+    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:
+        GetCountsOfRunningJobsPerTagResponse200
+    """
+
+    return sync_detailed(
+        client=client,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: Union[AuthenticatedClient, Client],
+) -> Response[GetCountsOfRunningJobsPerTagResponse200]:
+    """get counts of currently running jobs per tag
+
+    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[GetCountsOfRunningJobsPerTagResponse200]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: Union[AuthenticatedClient, Client],
+) -> Optional[GetCountsOfRunningJobsPerTagResponse200]:
+    """get counts of currently running jobs per tag
+
+    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:
+        GetCountsOfRunningJobsPerTagResponse200
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+        )
+    ).parsed

windmill_api/models/get_counts_of_running_jobs_per_tag_response_200.py

@@ -0,0 +1,44 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="GetCountsOfRunningJobsPerTagResponse200")
+
+
+@_attrs_define
+class GetCountsOfRunningJobsPerTagResponse200:
+    """ """
+
+    additional_properties: Dict[str, int] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        get_counts_of_running_jobs_per_tag_response_200 = cls()
+
+        get_counts_of_running_jobs_per_tag_response_200.additional_properties = d
+        return get_counts_of_running_jobs_per_tag_response_200
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> int:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: int) -> 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

windmill_api/models/preview.py

@@ -19,9 +19,9 @@ class Preview:
     """
     Attributes:
         args (PreviewArgs): The arguments to pass to the script or flow
-        content (Union[Unset, str]):
-        path (Union[Unset, str]):
-        script_hash (Union[Unset, str]):
+        content (Union[Unset, str]): The code to run
+        path (Union[Unset, str]): The path to the script
+        script_hash (Union[Unset, str]): The hash of the script
         language (Union[Unset, PreviewLanguage]):
         tag (Union[Unset, str]):
         kind (Union[Unset, PreviewKind]):

windmill_api/models/run_flow_preview_and_wait_result_json_body.py

@@ -0,0 +1,114 @@
+from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.run_flow_preview_and_wait_result_json_body_args import RunFlowPreviewAndWaitResultJsonBodyArgs
+    from ..models.run_flow_preview_and_wait_result_json_body_restarted_from import (
+        RunFlowPreviewAndWaitResultJsonBodyRestartedFrom,
+    )
+    from ..models.run_flow_preview_and_wait_result_json_body_value import RunFlowPreviewAndWaitResultJsonBodyValue
+
+
+T = TypeVar("T", bound="RunFlowPreviewAndWaitResultJsonBody")
+
+
+@_attrs_define
+class RunFlowPreviewAndWaitResultJsonBody:
+    """
+    Attributes:
+        value (RunFlowPreviewAndWaitResultJsonBodyValue):
+        args (RunFlowPreviewAndWaitResultJsonBodyArgs): The arguments to pass to the script or flow
+        path (Union[Unset, str]):
+        tag (Union[Unset, str]):
+        restarted_from (Union[Unset, RunFlowPreviewAndWaitResultJsonBodyRestartedFrom]):
+    """
+
+    value: "RunFlowPreviewAndWaitResultJsonBodyValue"
+    args: "RunFlowPreviewAndWaitResultJsonBodyArgs"
+    path: Union[Unset, str] = UNSET
+    tag: Union[Unset, str] = UNSET
+    restarted_from: Union[Unset, "RunFlowPreviewAndWaitResultJsonBodyRestartedFrom"] = UNSET
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        value = self.value.to_dict()
+
+        args = self.args.to_dict()
+
+        path = self.path
+        tag = self.tag
+        restarted_from: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.restarted_from, Unset):
+            restarted_from = self.restarted_from.to_dict()
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "value": value,
+                "args": args,
+            }
+        )
+        if path is not UNSET:
+            field_dict["path"] = path
+        if tag is not UNSET:
+            field_dict["tag"] = tag
+        if restarted_from is not UNSET:
+            field_dict["restarted_from"] = restarted_from
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.run_flow_preview_and_wait_result_json_body_args import RunFlowPreviewAndWaitResultJsonBodyArgs
+        from ..models.run_flow_preview_and_wait_result_json_body_restarted_from import (
+            RunFlowPreviewAndWaitResultJsonBodyRestartedFrom,
+        )
+        from ..models.run_flow_preview_and_wait_result_json_body_value import RunFlowPreviewAndWaitResultJsonBodyValue
+
+        d = src_dict.copy()
+        value = RunFlowPreviewAndWaitResultJsonBodyValue.from_dict(d.pop("value"))
+
+        args = RunFlowPreviewAndWaitResultJsonBodyArgs.from_dict(d.pop("args"))
+
+        path = d.pop("path", UNSET)
+
+        tag = d.pop("tag", UNSET)
+
+        _restarted_from = d.pop("restarted_from", UNSET)
+        restarted_from: Union[Unset, RunFlowPreviewAndWaitResultJsonBodyRestartedFrom]
+        if isinstance(_restarted_from, Unset):
+            restarted_from = UNSET
+        else:
+            restarted_from = RunFlowPreviewAndWaitResultJsonBodyRestartedFrom.from_dict(_restarted_from)
+
+        run_flow_preview_and_wait_result_json_body = cls(
+            value=value,
+            args=args,
+            path=path,
+            tag=tag,
+            restarted_from=restarted_from,
+        )
+
+        run_flow_preview_and_wait_result_json_body.additional_properties = d
+        return run_flow_preview_and_wait_result_json_body
+
+    @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

windmill_api/models/run_flow_preview_and_wait_result_json_body_args.py

@@ -0,0 +1,44 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RunFlowPreviewAndWaitResultJsonBodyArgs")
+
+
+@_attrs_define
+class RunFlowPreviewAndWaitResultJsonBodyArgs:
+    """The arguments to pass to the script or flow"""
+
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        run_flow_preview_and_wait_result_json_body_args = cls()
+
+        run_flow_preview_and_wait_result_json_body_args.additional_properties = d
+        return run_flow_preview_and_wait_result_json_body_args
+
+    @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

windmill_api/models/run_flow_preview_and_wait_result_json_body_restarted_from.py

@@ -0,0 +1,74 @@
+from typing import Any, Dict, List, Type, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="RunFlowPreviewAndWaitResultJsonBodyRestartedFrom")
+
+
+@_attrs_define
+class RunFlowPreviewAndWaitResultJsonBodyRestartedFrom:
+    """
+    Attributes:
+        flow_job_id (Union[Unset, str]):
+        step_id (Union[Unset, str]):
+        branch_or_iteration_n (Union[Unset, int]):
+    """
+
+    flow_job_id: Union[Unset, str] = UNSET
+    step_id: Union[Unset, str] = UNSET
+    branch_or_iteration_n: Union[Unset, int] = UNSET
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        flow_job_id = self.flow_job_id
+        step_id = self.step_id
+        branch_or_iteration_n = self.branch_or_iteration_n
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if flow_job_id is not UNSET:
+            field_dict["flow_job_id"] = flow_job_id
+        if step_id is not UNSET:
+            field_dict["step_id"] = step_id
+        if branch_or_iteration_n is not UNSET:
+            field_dict["branch_or_iteration_n"] = branch_or_iteration_n
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        flow_job_id = d.pop("flow_job_id", UNSET)
+
+        step_id = d.pop("step_id", UNSET)
+
+        branch_or_iteration_n = d.pop("branch_or_iteration_n", UNSET)
+
+        run_flow_preview_and_wait_result_json_body_restarted_from = cls(
+            flow_job_id=flow_job_id,
+            step_id=step_id,
+            branch_or_iteration_n=branch_or_iteration_n,
+        )
+
+        run_flow_preview_and_wait_result_json_body_restarted_from.additional_properties = d
+        return run_flow_preview_and_wait_result_json_body_restarted_from
+
+    @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