thrustlab 0.1.0__py3-none-any.whl

thrustlab/_models/simulation_create_body.py

@@ -0,0 +1,128 @@
+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 thrustlab._models.types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from thrustlab._models.rotor_group_in import RotorGroupIn
+
+
+T = TypeVar("T", bound="SimulationCreateBody")
+
+
+@_attrs_define
+class SimulationCreateBody:
+    """
+    Attributes:
+        airspeed_m_s (float):
+        battery_component_id (str):
+        density_kg_m3 (float):
+        project_id (str):
+        rotor_groups (list[RotorGroupIn]):
+        battery_charge_pct (float | Unset):  Default: 100.0.
+        name (None | str | Unset):
+        solver (str | Unset):  Default: 'bem'.
+    """
+
+    airspeed_m_s: float
+    battery_component_id: str
+    density_kg_m3: float
+    project_id: str
+    rotor_groups: list[RotorGroupIn]
+    battery_charge_pct: float | Unset = 100.0
+    name: None | str | Unset = UNSET
+    solver: str | Unset = "bem"
+
+    def to_dict(self) -> dict[str, Any]:
+        airspeed_m_s = self.airspeed_m_s
+
+        battery_component_id = self.battery_component_id
+
+        density_kg_m3 = self.density_kg_m3
+
+        project_id = self.project_id
+
+        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)
+
+        battery_charge_pct = self.battery_charge_pct
+
+        name: None | str | Unset
+        if isinstance(self.name, Unset):
+            name = UNSET
+        else:
+            name = self.name
+
+        solver = self.solver
+
+        field_dict: dict[str, Any] = {}
+
+        field_dict.update(
+            {
+                "airspeed_m_s": airspeed_m_s,
+                "battery_component_id": battery_component_id,
+                "density_kg_m3": density_kg_m3,
+                "project_id": project_id,
+                "rotor_groups": rotor_groups,
+            }
+        )
+        if battery_charge_pct is not UNSET:
+            field_dict["battery_charge_pct"] = battery_charge_pct
+        if name is not UNSET:
+            field_dict["name"] = name
+        if solver is not UNSET:
+            field_dict["solver"] = solver
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from thrustlab._models.rotor_group_in import RotorGroupIn
+
+        d = dict(src_dict)
+        airspeed_m_s = d.pop("airspeed_m_s")
+
+        battery_component_id = d.pop("battery_component_id")
+
+        density_kg_m3 = d.pop("density_kg_m3")
+
+        project_id = d.pop("project_id")
+
+        rotor_groups = []
+        _rotor_groups = d.pop("rotor_groups")
+        for rotor_groups_item_data in _rotor_groups:
+            rotor_groups_item = RotorGroupIn.from_dict(rotor_groups_item_data)
+
+            rotor_groups.append(rotor_groups_item)
+
+        battery_charge_pct = d.pop("battery_charge_pct", UNSET)
+
+        def _parse_name(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        name = _parse_name(d.pop("name", UNSET))
+
+        solver = d.pop("solver", UNSET)
+
+        simulation_create_body = cls(
+            airspeed_m_s=airspeed_m_s,
+            battery_component_id=battery_component_id,
+            density_kg_m3=density_kg_m3,
+            project_id=project_id,
+            rotor_groups=rotor_groups,
+            battery_charge_pct=battery_charge_pct,
+            name=name,
+            solver=solver,
+        )
+
+        return simulation_create_body

thrustlab/_models/simulation_inputs.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+
+T = TypeVar("T", bound="SimulationInputs")
+
+
+@_attrs_define
+class SimulationInputs:
+    """
+    Attributes:
+        airspeed_m_s (float):
+        battery_charge_pct (float):
+        density_kg_m3 (float):
+        solver (str):
+    """
+
+    airspeed_m_s: float
+    battery_charge_pct: float
+    density_kg_m3: float
+    solver: str
+
+    def to_dict(self) -> dict[str, Any]:
+        airspeed_m_s = self.airspeed_m_s
+
+        battery_charge_pct = self.battery_charge_pct
+
+        density_kg_m3 = self.density_kg_m3
+
+        solver = self.solver
+
+        field_dict: dict[str, Any] = {}
+
+        field_dict.update(
+            {
+                "airspeed_m_s": airspeed_m_s,
+                "battery_charge_pct": battery_charge_pct,
+                "density_kg_m3": density_kg_m3,
+                "solver": solver,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        airspeed_m_s = d.pop("airspeed_m_s")
+
+        battery_charge_pct = d.pop("battery_charge_pct")
+
+        density_kg_m3 = d.pop("density_kg_m3")
+
+        solver = d.pop("solver")
+
+        simulation_inputs = cls(
+            airspeed_m_s=airspeed_m_s,
+            battery_charge_pct=battery_charge_pct,
+            density_kg_m3=density_kg_m3,
+            solver=solver,
+        )
+
+        return simulation_inputs

thrustlab/_models/simulation_list_response.py

@@ -0,0 +1,88 @@
+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.simulation_resource import SimulationResource
+
+
+T = TypeVar("T", bound="SimulationListResponse")
+
+
+@_attrs_define
+class SimulationListResponse:
+    """
+    Attributes:
+        data (list[SimulationResource]):
+        has_more (bool):
+        next_cursor (None | str):
+        object_ (Literal['list']):
+    """
+
+    data: list[SimulationResource]
+    has_more: bool
+    next_cursor: None | str
+    object_: Literal["list"]
+
+    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(
+            {
+                "data": data,
+                "has_more": has_more,
+                "next_cursor": next_cursor,
+                "object": object_,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from thrustlab._models.simulation_resource import SimulationResource
+
+        d = dict(src_dict)
+        data = []
+        _data = d.pop("data")
+        for data_item_data in _data:
+            data_item = SimulationResource.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_ = cast(Literal["list"], d.pop("object"))
+        if object_ != "list":
+            raise ValueError(f"object must match const 'list', got '{object_}'")
+
+        simulation_list_response = cls(
+            data=data,
+            has_more=has_more,
+            next_cursor=next_cursor,
+            object_=object_,
+        )
+
+        return simulation_list_response

thrustlab/_models/simulation_patch.py

@@ -0,0 +1,121 @@
+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 thrustlab._models.types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from thrustlab._models.simulation_patch_plot_config_json_type_0_item import SimulationPatchPlotConfigJsonType0Item
+
+
+T = TypeVar("T", bound="SimulationPatch")
+
+
+@_attrs_define
+class SimulationPatch:
+    """
+    Attributes:
+        is_starred (bool | None | Unset):
+        name (None | str | Unset):
+        plot_config_json (list[SimulationPatchPlotConfigJsonType0Item] | None | Unset):
+    """
+
+    is_starred: bool | None | Unset = UNSET
+    name: None | str | Unset = UNSET
+    plot_config_json: list[SimulationPatchPlotConfigJsonType0Item] | None | Unset = UNSET
+
+    def to_dict(self) -> dict[str, Any]:
+        is_starred: bool | None | Unset
+        if isinstance(self.is_starred, Unset):
+            is_starred = UNSET
+        else:
+            is_starred = self.is_starred
+
+        name: None | str | Unset
+        if isinstance(self.name, Unset):
+            name = UNSET
+        else:
+            name = self.name
+
+        plot_config_json: list[dict[str, Any]] | None | Unset
+        if isinstance(self.plot_config_json, Unset):
+            plot_config_json = UNSET
+        elif 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
+
+        field_dict: dict[str, Any] = {}
+
+        field_dict.update({})
+        if is_starred is not UNSET:
+            field_dict["is_starred"] = is_starred
+        if name is not UNSET:
+            field_dict["name"] = name
+        if plot_config_json is not UNSET:
+            field_dict["plot_config_json"] = plot_config_json
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from thrustlab._models.simulation_patch_plot_config_json_type_0_item import SimulationPatchPlotConfigJsonType0Item
+
+        d = dict(src_dict)
+
+        def _parse_is_starred(data: object) -> bool | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(bool | None | Unset, data)
+
+        is_starred = _parse_is_starred(d.pop("is_starred", UNSET))
+
+        def _parse_name(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        name = _parse_name(d.pop("name", UNSET))
+
+        def _parse_plot_config_json(data: object) -> list[SimulationPatchPlotConfigJsonType0Item] | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                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 = SimulationPatchPlotConfigJsonType0Item.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[SimulationPatchPlotConfigJsonType0Item] | None | Unset, data)
+
+        plot_config_json = _parse_plot_config_json(d.pop("plot_config_json", UNSET))
+
+        simulation_patch = cls(
+            is_starred=is_starred,
+            name=name,
+            plot_config_json=plot_config_json,
+        )
+
+        return simulation_patch

thrustlab/_models/simulation_patch_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="SimulationPatchPlotConfigJsonType0Item")
+
+
+@_attrs_define
+class SimulationPatchPlotConfigJsonType0Item:
+    """ """
+
+    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_patch_plot_config_json_type_0_item = cls()
+
+        simulation_patch_plot_config_json_type_0_item.additional_properties = d
+        return simulation_patch_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