thrustlab 0.1.0__py3-none-any.whl

thrustlab/_models/sweep_config_in.py

@@ -0,0 +1,172 @@
+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.sweep_param_range_in import SweepParamRangeIn
+
+
+T = TypeVar("T", bound="SweepConfigIn")
+
+
+@_attrs_define
+class SweepConfigIn:
+    """
+    Attributes:
+        rotor_sweep_mask (list[bool]):
+        airspeed (None | SweepParamRangeIn | Unset):
+        battery_charge (None | SweepParamRangeIn | Unset):
+        density (None | SweepParamRangeIn | Unset):
+        throttle (None | SweepParamRangeIn | Unset):
+    """
+
+    rotor_sweep_mask: list[bool]
+    airspeed: None | SweepParamRangeIn | Unset = UNSET
+    battery_charge: None | SweepParamRangeIn | Unset = UNSET
+    density: None | SweepParamRangeIn | Unset = UNSET
+    throttle: None | SweepParamRangeIn | Unset = UNSET
+
+    def to_dict(self) -> dict[str, Any]:
+        from thrustlab._models.sweep_param_range_in import SweepParamRangeIn
+
+        rotor_sweep_mask = self.rotor_sweep_mask
+
+        airspeed: dict[str, Any] | None | Unset
+        if isinstance(self.airspeed, Unset):
+            airspeed = UNSET
+        elif isinstance(self.airspeed, SweepParamRangeIn):
+            airspeed = self.airspeed.to_dict()
+        else:
+            airspeed = self.airspeed
+
+        battery_charge: dict[str, Any] | None | Unset
+        if isinstance(self.battery_charge, Unset):
+            battery_charge = UNSET
+        elif isinstance(self.battery_charge, SweepParamRangeIn):
+            battery_charge = self.battery_charge.to_dict()
+        else:
+            battery_charge = self.battery_charge
+
+        density: dict[str, Any] | None | Unset
+        if isinstance(self.density, Unset):
+            density = UNSET
+        elif isinstance(self.density, SweepParamRangeIn):
+            density = self.density.to_dict()
+        else:
+            density = self.density
+
+        throttle: dict[str, Any] | None | Unset
+        if isinstance(self.throttle, Unset):
+            throttle = UNSET
+        elif isinstance(self.throttle, SweepParamRangeIn):
+            throttle = self.throttle.to_dict()
+        else:
+            throttle = self.throttle
+
+        field_dict: dict[str, Any] = {}
+
+        field_dict.update(
+            {
+                "rotor_sweep_mask": rotor_sweep_mask,
+            }
+        )
+        if airspeed is not UNSET:
+            field_dict["airspeed"] = airspeed
+        if battery_charge is not UNSET:
+            field_dict["battery_charge"] = battery_charge
+        if density is not UNSET:
+            field_dict["density"] = density
+        if throttle is not UNSET:
+            field_dict["throttle"] = throttle
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from thrustlab._models.sweep_param_range_in import SweepParamRangeIn
+
+        d = dict(src_dict)
+        rotor_sweep_mask = cast(list[bool], d.pop("rotor_sweep_mask"))
+
+        def _parse_airspeed(data: object) -> None | SweepParamRangeIn | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                airspeed_type_0 = SweepParamRangeIn.from_dict(data)
+
+                return airspeed_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | SweepParamRangeIn | Unset, data)
+
+        airspeed = _parse_airspeed(d.pop("airspeed", UNSET))
+
+        def _parse_battery_charge(data: object) -> None | SweepParamRangeIn | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                battery_charge_type_0 = SweepParamRangeIn.from_dict(data)
+
+                return battery_charge_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | SweepParamRangeIn | Unset, data)
+
+        battery_charge = _parse_battery_charge(d.pop("battery_charge", UNSET))
+
+        def _parse_density(data: object) -> None | SweepParamRangeIn | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                density_type_0 = SweepParamRangeIn.from_dict(data)
+
+                return density_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | SweepParamRangeIn | Unset, data)
+
+        density = _parse_density(d.pop("density", UNSET))
+
+        def _parse_throttle(data: object) -> None | SweepParamRangeIn | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                throttle_type_0 = SweepParamRangeIn.from_dict(data)
+
+                return throttle_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | SweepParamRangeIn | Unset, data)
+
+        throttle = _parse_throttle(d.pop("throttle", UNSET))
+
+        sweep_config_in = cls(
+            rotor_sweep_mask=rotor_sweep_mask,
+            airspeed=airspeed,
+            battery_charge=battery_charge,
+            density=density,
+            throttle=throttle,
+        )
+
+        return sweep_config_in

thrustlab/_models/sweep_create_body.py

@@ -0,0 +1,138 @@
+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.sweep_config_in import SweepConfigIn
+    from thrustlab._models.sweep_rotor_group_in import SweepRotorGroupIn
+
+
+T = TypeVar("T", bound="SweepCreateBody")
+
+
+@_attrs_define
+class SweepCreateBody:
+    """
+    Attributes:
+        airspeed_m_s (float):
+        battery_component_id (str):
+        density_kg_m3 (float):
+        project_id (str):
+        rotor_groups (list[SweepRotorGroupIn]):
+        sweep (SweepConfigIn):
+        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[SweepRotorGroupIn]
+    sweep: SweepConfigIn
+    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)
+
+        sweep = self.sweep.to_dict()
+
+        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,
+                "sweep": sweep,
+            }
+        )
+        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.sweep_config_in import SweepConfigIn
+        from thrustlab._models.sweep_rotor_group_in import SweepRotorGroupIn
+
+        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 = SweepRotorGroupIn.from_dict(rotor_groups_item_data)
+
+            rotor_groups.append(rotor_groups_item)
+
+        sweep = SweepConfigIn.from_dict(d.pop("sweep"))
+
+        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)
+
+        sweep_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,
+            sweep=sweep,
+            battery_charge_pct=battery_charge_pct,
+            name=name,
+            solver=solver,
+        )
+
+        return sweep_create_body

thrustlab/_models/sweep_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="SweepInputs")
+
+
+@_attrs_define
+class SweepInputs:
+    """
+    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")
+
+        sweep_inputs = cls(
+            airspeed_m_s=airspeed_m_s,
+            battery_charge_pct=battery_charge_pct,
+            density_kg_m3=density_kg_m3,
+            solver=solver,
+        )
+
+        return sweep_inputs

thrustlab/_models/sweep_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.sweep_resource import SweepResource
+
+
+T = TypeVar("T", bound="SweepListResponse")
+
+
+@_attrs_define
+class SweepListResponse:
+    """
+    Attributes:
+        data (list[SweepResource]):
+        has_more (bool):
+        next_cursor (None | str):
+        object_ (Literal['list']):
+    """
+
+    data: list[SweepResource]
+    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.sweep_resource import SweepResource
+
+        d = dict(src_dict)
+        data = []
+        _data = d.pop("data")
+        for data_item_data in _data:
+            data_item = SweepResource.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_}'")
+
+        sweep_list_response = cls(
+            data=data,
+            has_more=has_more,
+            next_cursor=next_cursor,
+            object_=object_,
+        )
+
+        return sweep_list_response

thrustlab/_models/sweep_param_range_in.py

@@ -0,0 +1,136 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+
+from thrustlab._models.sweep_param_range_in_mode import SweepParamRangeInMode
+from thrustlab._models.types import UNSET, Unset
+
+T = TypeVar("T", bound="SweepParamRangeIn")
+
+
+@_attrs_define
+class SweepParamRangeIn:
+    """
+    Attributes:
+        mode (SweepParamRangeInMode):
+        start (float | None | Unset):
+        steps (int | None | Unset):
+        stop (float | None | Unset):
+        values (list[float] | None | Unset):
+    """
+
+    mode: SweepParamRangeInMode
+    start: float | None | Unset = UNSET
+    steps: int | None | Unset = UNSET
+    stop: float | None | Unset = UNSET
+    values: list[float] | None | Unset = UNSET
+
+    def to_dict(self) -> dict[str, Any]:
+        mode = self.mode.value
+
+        start: float | None | Unset
+        if isinstance(self.start, Unset):
+            start = UNSET
+        else:
+            start = self.start
+
+        steps: int | None | Unset
+        if isinstance(self.steps, Unset):
+            steps = UNSET
+        else:
+            steps = self.steps
+
+        stop: float | None | Unset
+        if isinstance(self.stop, Unset):
+            stop = UNSET
+        else:
+            stop = self.stop
+
+        values: list[float] | None | Unset
+        if isinstance(self.values, Unset):
+            values = UNSET
+        elif isinstance(self.values, list):
+            values = self.values
+
+        else:
+            values = self.values
+
+        field_dict: dict[str, Any] = {}
+
+        field_dict.update(
+            {
+                "mode": mode,
+            }
+        )
+        if start is not UNSET:
+            field_dict["start"] = start
+        if steps is not UNSET:
+            field_dict["steps"] = steps
+        if stop is not UNSET:
+            field_dict["stop"] = stop
+        if values is not UNSET:
+            field_dict["values"] = values
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        mode = SweepParamRangeInMode(d.pop("mode"))
+
+        def _parse_start(data: object) -> float | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(float | None | Unset, data)
+
+        start = _parse_start(d.pop("start", UNSET))
+
+        def _parse_steps(data: object) -> int | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(int | None | Unset, data)
+
+        steps = _parse_steps(d.pop("steps", UNSET))
+
+        def _parse_stop(data: object) -> float | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(float | None | Unset, data)
+
+        stop = _parse_stop(d.pop("stop", UNSET))
+
+        def _parse_values(data: object) -> list[float] | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                values_type_0 = cast(list[float], data)
+
+                return values_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(list[float] | None | Unset, data)
+
+        values = _parse_values(d.pop("values", UNSET))
+
+        sweep_param_range_in = cls(
+            mode=mode,
+            start=start,
+            steps=steps,
+            stop=stop,
+            values=values,
+        )
+
+        return sweep_param_range_in

thrustlab/_models/sweep_param_range_in_mode.py

@@ -0,0 +1,9 @@
+from enum import Enum
+
+
+class SweepParamRangeInMode(str, Enum):
+    CUSTOM = "custom"
+    RANGE = "range"
+
+    def __str__(self) -> str:
+        return str(self.value)