thrustlab 0.1.0__py3-none-any.whl

thrustlab/_models/simulation_resource.py

@@ -0,0 +1,283 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, Literal, TypeVar, cast
+
+from attrs import define as _attrs_define
+
+if TYPE_CHECKING:
+    from thrustlab._models.rotor_group_resource import RotorGroupResource
+    from thrustlab._models.simulation_inputs import SimulationInputs
+    from thrustlab._models.simulation_resource_error_type_0 import SimulationResourceErrorType0
+    from thrustlab._models.simulation_resource_plot_config_json_type_0_item import SimulationResourcePlotConfigJsonType0Item
+    from thrustlab._models.simulation_resource_result_type_0 import SimulationResourceResultType0
+
+
+T = TypeVar("T", bound="SimulationResource")
+
+
+@_attrs_define
+class SimulationResource:
+    """
+    Attributes:
+        analysis_type (str):
+        battery_component_id (str):
+        completed_at (None | str):
+        created_at (None | str):
+        credits_cost (int):
+        error (None | SimulationResourceErrorType0):
+        id (str):
+        inputs (SimulationInputs):
+        is_starred (bool):
+        name (None | str):
+        object_ (Literal['simulation']):
+        plot_config_json (list[SimulationResourcePlotConfigJsonType0Item] | None):
+        progress (float | None):
+        project_id (str):
+        result (None | SimulationResourceResultType0):
+        rotor_groups (list[RotorGroupResource]):
+        status (str):
+    """
+
+    analysis_type: str
+    battery_component_id: str
+    completed_at: None | str
+    created_at: None | str
+    credits_cost: int
+    error: None | SimulationResourceErrorType0
+    id: str
+    inputs: SimulationInputs
+    is_starred: bool
+    name: None | str
+    object_: Literal["simulation"]
+    plot_config_json: list[SimulationResourcePlotConfigJsonType0Item] | None
+    progress: float | None
+    project_id: str
+    result: None | SimulationResourceResultType0
+    rotor_groups: list[RotorGroupResource]
+    status: str
+
+    def to_dict(self) -> dict[str, Any]:
+        from thrustlab._models.simulation_resource_error_type_0 import SimulationResourceErrorType0
+        from thrustlab._models.simulation_resource_result_type_0 import SimulationResourceResultType0
+
+        analysis_type = self.analysis_type
+
+        battery_component_id = self.battery_component_id
+
+        completed_at: None | str
+        completed_at = self.completed_at
+
+        created_at: None | str
+        created_at = self.created_at
+
+        credits_cost = self.credits_cost
+
+        error: dict[str, Any] | None
+        if isinstance(self.error, SimulationResourceErrorType0):
+            error = self.error.to_dict()
+        else:
+            error = self.error
+
+        id = self.id
+
+        inputs = self.inputs.to_dict()
+
+        is_starred = self.is_starred
+
+        name: None | str
+        name = self.name
+
+        object_ = self.object_
+
+        plot_config_json: list[dict[str, Any]] | None
+        if isinstance(self.plot_config_json, list):
+            plot_config_json = []
+            for plot_config_json_type_0_item_data in self.plot_config_json:
+                plot_config_json_type_0_item = plot_config_json_type_0_item_data.to_dict()
+                plot_config_json.append(plot_config_json_type_0_item)
+
+        else:
+            plot_config_json = self.plot_config_json
+
+        progress: float | None
+        progress = self.progress
+
+        project_id = self.project_id
+
+        result: dict[str, Any] | None
+        if isinstance(self.result, SimulationResourceResultType0):
+            result = self.result.to_dict()
+        else:
+            result = self.result
+
+        rotor_groups = []
+        for rotor_groups_item_data in self.rotor_groups:
+            rotor_groups_item = rotor_groups_item_data.to_dict()
+            rotor_groups.append(rotor_groups_item)
+
+        status = self.status
+
+        field_dict: dict[str, Any] = {}
+
+        field_dict.update(
+            {
+                "analysis_type": analysis_type,
+                "battery_component_id": battery_component_id,
+                "completed_at": completed_at,
+                "created_at": created_at,
+                "credits_cost": credits_cost,
+                "error": error,
+                "id": id,
+                "inputs": inputs,
+                "is_starred": is_starred,
+                "name": name,
+                "object": object_,
+                "plot_config_json": plot_config_json,
+                "progress": progress,
+                "project_id": project_id,
+                "result": result,
+                "rotor_groups": rotor_groups,
+                "status": status,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from thrustlab._models.rotor_group_resource import RotorGroupResource
+        from thrustlab._models.simulation_inputs import SimulationInputs
+        from thrustlab._models.simulation_resource_error_type_0 import SimulationResourceErrorType0
+        from thrustlab._models.simulation_resource_plot_config_json_type_0_item import SimulationResourcePlotConfigJsonType0Item
+        from thrustlab._models.simulation_resource_result_type_0 import SimulationResourceResultType0
+
+        d = dict(src_dict)
+        analysis_type = d.pop("analysis_type")
+
+        battery_component_id = d.pop("battery_component_id")
+
+        def _parse_completed_at(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        completed_at = _parse_completed_at(d.pop("completed_at"))
+
+        def _parse_created_at(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        created_at = _parse_created_at(d.pop("created_at"))
+
+        credits_cost = d.pop("credits_cost")
+
+        def _parse_error(data: object) -> None | SimulationResourceErrorType0:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                error_type_0 = SimulationResourceErrorType0.from_dict(data)
+
+                return error_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | SimulationResourceErrorType0, data)
+
+        error = _parse_error(d.pop("error"))
+
+        id = d.pop("id")
+
+        inputs = SimulationInputs.from_dict(d.pop("inputs"))
+
+        is_starred = d.pop("is_starred")
+
+        def _parse_name(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        name = _parse_name(d.pop("name"))
+
+        object_ = cast(Literal["simulation"], d.pop("object"))
+        if object_ != "simulation":
+            raise ValueError(f"object must match const 'simulation', got '{object_}'")
+
+        def _parse_plot_config_json(data: object) -> list[SimulationResourcePlotConfigJsonType0Item] | None:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                plot_config_json_type_0 = []
+                _plot_config_json_type_0 = data
+                for plot_config_json_type_0_item_data in _plot_config_json_type_0:
+                    plot_config_json_type_0_item = SimulationResourcePlotConfigJsonType0Item.from_dict(
+                        plot_config_json_type_0_item_data
+                    )
+
+                    plot_config_json_type_0.append(plot_config_json_type_0_item)
+
+                return plot_config_json_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(list[SimulationResourcePlotConfigJsonType0Item] | None, data)
+
+        plot_config_json = _parse_plot_config_json(d.pop("plot_config_json"))
+
+        def _parse_progress(data: object) -> float | None:
+            if data is None:
+                return data
+            return cast(float | None, data)
+
+        progress = _parse_progress(d.pop("progress"))
+
+        project_id = d.pop("project_id")
+
+        def _parse_result(data: object) -> None | SimulationResourceResultType0:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                result_type_0 = SimulationResourceResultType0.from_dict(data)
+
+                return result_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | SimulationResourceResultType0, data)
+
+        result = _parse_result(d.pop("result"))
+
+        rotor_groups = []
+        _rotor_groups = d.pop("rotor_groups")
+        for rotor_groups_item_data in _rotor_groups:
+            rotor_groups_item = RotorGroupResource.from_dict(rotor_groups_item_data)
+
+            rotor_groups.append(rotor_groups_item)
+
+        status = d.pop("status")
+
+        simulation_resource = cls(
+            analysis_type=analysis_type,
+            battery_component_id=battery_component_id,
+            completed_at=completed_at,
+            created_at=created_at,
+            credits_cost=credits_cost,
+            error=error,
+            id=id,
+            inputs=inputs,
+            is_starred=is_starred,
+            name=name,
+            object_=object_,
+            plot_config_json=plot_config_json,
+            progress=progress,
+            project_id=project_id,
+            result=result,
+            rotor_groups=rotor_groups,
+            status=status,
+        )
+
+        return simulation_resource

thrustlab/_models/simulation_resource_error_type_0.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="SimulationResourceErrorType0")
+
+
+@_attrs_define
+class SimulationResourceErrorType0:
+    """ """
+
+    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)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        simulation_resource_error_type_0 = cls()
+
+        simulation_resource_error_type_0.additional_properties = d
+        return simulation_resource_error_type_0
+
+    @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

thrustlab/_models/simulation_resource_plot_config_json_type_0_item.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="SimulationResourcePlotConfigJsonType0Item")
+
+
+@_attrs_define
+class SimulationResourcePlotConfigJsonType0Item:
+    """ """
+
+    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)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        simulation_resource_plot_config_json_type_0_item = cls()
+
+        simulation_resource_plot_config_json_type_0_item.additional_properties = d
+        return simulation_resource_plot_config_json_type_0_item
+
+    @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

thrustlab/_models/simulation_resource_result_type_0.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="SimulationResourceResultType0")
+
+
+@_attrs_define
+class SimulationResourceResultType0:
+    """ """
+
+    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)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        simulation_resource_result_type_0 = cls()
+
+        simulation_resource_result_type_0.additional_properties = d
+        return simulation_resource_result_type_0
+
+    @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

thrustlab/_models/starred_component_create_body.py

@@ -0,0 +1,80 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="StarredComponentCreateBody")
+
+
+@_attrs_define
+class StarredComponentCreateBody:
+    """Accepts either prefixed (``proj_<ksuid>`` / ``comp_<ksuid>``) or raw
+    UUID strings — resolution happens in the handler so the same body shape
+    works during the reference-app transition.
+
+        Attributes:
+            component_id (str):
+            component_type (str):
+            project_id (str):
+    """
+
+    component_id: str
+    component_type: str
+    project_id: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        component_id = self.component_id
+
+        component_type = self.component_type
+
+        project_id = self.project_id
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "component_id": component_id,
+                "component_type": component_type,
+                "project_id": project_id,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        component_id = d.pop("component_id")
+
+        component_type = d.pop("component_type")
+
+        project_id = d.pop("project_id")
+
+        starred_component_create_body = cls(
+            component_id=component_id,
+            component_type=component_type,
+            project_id=project_id,
+        )
+
+        starred_component_create_body.additional_properties = d
+        return starred_component_create_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

thrustlab/_models/starred_component_list_response.py

@@ -0,0 +1,108 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from thrustlab._models.types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from thrustlab._models.starred_component_resource import StarredComponentResource
+
+
+T = TypeVar("T", bound="StarredComponentListResponse")
+
+
+@_attrs_define
+class StarredComponentListResponse:
+    """
+    Attributes:
+        data (list[StarredComponentResource]):
+        has_more (bool):
+        next_cursor (None | str):
+        object_ (str | Unset):  Default: 'list'.
+    """
+
+    data: list[StarredComponentResource]
+    has_more: bool
+    next_cursor: None | str
+    object_: str | Unset = "list"
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        data = []
+        for data_item_data in self.data:
+            data_item = data_item_data.to_dict()
+            data.append(data_item)
+
+        has_more = self.has_more
+
+        next_cursor: None | str
+        next_cursor = self.next_cursor
+
+        object_ = self.object_
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "data": data,
+                "has_more": has_more,
+                "next_cursor": next_cursor,
+            }
+        )
+        if object_ is not UNSET:
+            field_dict["object"] = object_
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from thrustlab._models.starred_component_resource import StarredComponentResource
+
+        d = dict(src_dict)
+        data = []
+        _data = d.pop("data")
+        for data_item_data in _data:
+            data_item = StarredComponentResource.from_dict(data_item_data)
+
+            data.append(data_item)
+
+        has_more = d.pop("has_more")
+
+        def _parse_next_cursor(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        next_cursor = _parse_next_cursor(d.pop("next_cursor"))
+
+        object_ = d.pop("object", UNSET)
+
+        starred_component_list_response = cls(
+            data=data,
+            has_more=has_more,
+            next_cursor=next_cursor,
+            object_=object_,
+        )
+
+        starred_component_list_response.additional_properties = d
+        return starred_component_list_response
+
+    @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