thrustlab 0.1.0__py3-none-any.whl

thrustlab/_models/sweep_resource.py

@@ -0,0 +1,327 @@
+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.sweep_inputs import SweepInputs
+    from thrustlab._models.sweep_resource_error_type_0 import SweepResourceErrorType0
+    from thrustlab._models.sweep_resource_plot_config_json_type_0_item import SweepResourcePlotConfigJsonType0Item
+    from thrustlab._models.sweep_resource_summary_type_0 import SweepResourceSummaryType0
+    from thrustlab._models.sweep_resource_sweep_config_type_0 import SweepResourceSweepConfigType0
+
+
+T = TypeVar("T", bound="SweepResource")
+
+
+@_attrs_define
+class SweepResource:
+    """
+    Attributes:
+        analysis_type (str):
+        battery_component_id (str):
+        completed_at (None | str):
+        completed_points (int):
+        created_at (None | str):
+        credits_cost (int):
+        error (None | SweepResourceErrorType0):
+        id (str):
+        inputs (SweepInputs):
+        is_starred (bool):
+        name (None | str):
+        object_ (Literal['sweep']):
+        plot_config_json (list[SweepResourcePlotConfigJsonType0Item] | None):
+        progress (float | None):
+        project_id (str):
+        rotor_groups (list[RotorGroupResource]):
+        status (str):
+        summary (None | SweepResourceSummaryType0):
+        sweep_config (None | SweepResourceSweepConfigType0):
+        total_points (int):
+    """
+
+    analysis_type: str
+    battery_component_id: str
+    completed_at: None | str
+    completed_points: int
+    created_at: None | str
+    credits_cost: int
+    error: None | SweepResourceErrorType0
+    id: str
+    inputs: SweepInputs
+    is_starred: bool
+    name: None | str
+    object_: Literal["sweep"]
+    plot_config_json: list[SweepResourcePlotConfigJsonType0Item] | None
+    progress: float | None
+    project_id: str
+    rotor_groups: list[RotorGroupResource]
+    status: str
+    summary: None | SweepResourceSummaryType0
+    sweep_config: None | SweepResourceSweepConfigType0
+    total_points: int
+
+    def to_dict(self) -> dict[str, Any]:
+        from thrustlab._models.sweep_resource_error_type_0 import SweepResourceErrorType0
+        from thrustlab._models.sweep_resource_summary_type_0 import SweepResourceSummaryType0
+        from thrustlab._models.sweep_resource_sweep_config_type_0 import SweepResourceSweepConfigType0
+
+        analysis_type = self.analysis_type
+
+        battery_component_id = self.battery_component_id
+
+        completed_at: None | str
+        completed_at = self.completed_at
+
+        completed_points = self.completed_points
+
+        created_at: None | str
+        created_at = self.created_at
+
+        credits_cost = self.credits_cost
+
+        error: dict[str, Any] | None
+        if isinstance(self.error, SweepResourceErrorType0):
+            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
+
+        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
+
+        summary: dict[str, Any] | None
+        if isinstance(self.summary, SweepResourceSummaryType0):
+            summary = self.summary.to_dict()
+        else:
+            summary = self.summary
+
+        sweep_config: dict[str, Any] | None
+        if isinstance(self.sweep_config, SweepResourceSweepConfigType0):
+            sweep_config = self.sweep_config.to_dict()
+        else:
+            sweep_config = self.sweep_config
+
+        total_points = self.total_points
+
+        field_dict: dict[str, Any] = {}
+
+        field_dict.update(
+            {
+                "analysis_type": analysis_type,
+                "battery_component_id": battery_component_id,
+                "completed_at": completed_at,
+                "completed_points": completed_points,
+                "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,
+                "rotor_groups": rotor_groups,
+                "status": status,
+                "summary": summary,
+                "sweep_config": sweep_config,
+                "total_points": total_points,
+            }
+        )
+
+        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.sweep_inputs import SweepInputs
+        from thrustlab._models.sweep_resource_error_type_0 import SweepResourceErrorType0
+        from thrustlab._models.sweep_resource_plot_config_json_type_0_item import SweepResourcePlotConfigJsonType0Item
+        from thrustlab._models.sweep_resource_summary_type_0 import SweepResourceSummaryType0
+        from thrustlab._models.sweep_resource_sweep_config_type_0 import SweepResourceSweepConfigType0
+
+        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"))
+
+        completed_points = d.pop("completed_points")
+
+        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 | SweepResourceErrorType0:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                error_type_0 = SweepResourceErrorType0.from_dict(data)
+
+                return error_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | SweepResourceErrorType0, data)
+
+        error = _parse_error(d.pop("error"))
+
+        id = d.pop("id")
+
+        inputs = SweepInputs.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["sweep"], d.pop("object"))
+        if object_ != "sweep":
+            raise ValueError(f"object must match const 'sweep', got '{object_}'")
+
+        def _parse_plot_config_json(data: object) -> list[SweepResourcePlotConfigJsonType0Item] | 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 = SweepResourcePlotConfigJsonType0Item.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[SweepResourcePlotConfigJsonType0Item] | 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")
+
+        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")
+
+        def _parse_summary(data: object) -> None | SweepResourceSummaryType0:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                summary_type_0 = SweepResourceSummaryType0.from_dict(data)
+
+                return summary_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | SweepResourceSummaryType0, data)
+
+        summary = _parse_summary(d.pop("summary"))
+
+        def _parse_sweep_config(data: object) -> None | SweepResourceSweepConfigType0:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                sweep_config_type_0 = SweepResourceSweepConfigType0.from_dict(data)
+
+                return sweep_config_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | SweepResourceSweepConfigType0, data)
+
+        sweep_config = _parse_sweep_config(d.pop("sweep_config"))
+
+        total_points = d.pop("total_points")
+
+        sweep_resource = cls(
+            analysis_type=analysis_type,
+            battery_component_id=battery_component_id,
+            completed_at=completed_at,
+            completed_points=completed_points,
+            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,
+            rotor_groups=rotor_groups,
+            status=status,
+            summary=summary,
+            sweep_config=sweep_config,
+            total_points=total_points,
+        )
+
+        return sweep_resource

thrustlab/_models/sweep_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="SweepResourceErrorType0")
+
+
+@_attrs_define
+class SweepResourceErrorType0:
+    """ """
+
+    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)
+        sweep_resource_error_type_0 = cls()
+
+        sweep_resource_error_type_0.additional_properties = d
+        return sweep_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/sweep_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="SweepResourcePlotConfigJsonType0Item")
+
+
+@_attrs_define
+class SweepResourcePlotConfigJsonType0Item:
+    """ """
+
+    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)
+        sweep_resource_plot_config_json_type_0_item = cls()
+
+        sweep_resource_plot_config_json_type_0_item.additional_properties = d
+        return sweep_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/sweep_resource_summary_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="SweepResourceSummaryType0")
+
+
+@_attrs_define
+class SweepResourceSummaryType0:
+    """ """
+
+    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)
+        sweep_resource_summary_type_0 = cls()
+
+        sweep_resource_summary_type_0.additional_properties = d
+        return sweep_resource_summary_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/sweep_resource_sweep_config_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="SweepResourceSweepConfigType0")
+
+
+@_attrs_define
+class SweepResourceSweepConfigType0:
+    """ """
+
+    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)
+        sweep_resource_sweep_config_type_0 = cls()
+
+        sweep_resource_sweep_config_type_0.additional_properties = d
+        return sweep_resource_sweep_config_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/sweep_rotor_group_in.py

@@ -0,0 +1,95 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+
+from thrustlab._models.types import UNSET, Unset
+
+T = TypeVar("T", bound="SweepRotorGroupIn")
+
+
+@_attrs_define
+class SweepRotorGroupIn:
+    """
+    Attributes:
+        label (str):
+        motor_component_id (str):
+        propeller_component_id (str):
+        throttle_pct (float):
+        count (int | Unset):  Default: 1.
+        esc_motor_wire_resistance_mohm (float | Unset):  Default: 0.0.
+        esc_resistance_mohm (float | Unset):  Default: 0.0.
+    """
+
+    label: str
+    motor_component_id: str
+    propeller_component_id: str
+    throttle_pct: float
+    count: int | Unset = 1
+    esc_motor_wire_resistance_mohm: float | Unset = 0.0
+    esc_resistance_mohm: float | Unset = 0.0
+
+    def to_dict(self) -> dict[str, Any]:
+        label = self.label
+
+        motor_component_id = self.motor_component_id
+
+        propeller_component_id = self.propeller_component_id
+
+        throttle_pct = self.throttle_pct
+
+        count = self.count
+
+        esc_motor_wire_resistance_mohm = self.esc_motor_wire_resistance_mohm
+
+        esc_resistance_mohm = self.esc_resistance_mohm
+
+        field_dict: dict[str, Any] = {}
+
+        field_dict.update(
+            {
+                "label": label,
+                "motor_component_id": motor_component_id,
+                "propeller_component_id": propeller_component_id,
+                "throttle_pct": throttle_pct,
+            }
+        )
+        if count is not UNSET:
+            field_dict["count"] = count
+        if esc_motor_wire_resistance_mohm is not UNSET:
+            field_dict["esc_motor_wire_resistance_mohm"] = esc_motor_wire_resistance_mohm
+        if esc_resistance_mohm is not UNSET:
+            field_dict["esc_resistance_mohm"] = esc_resistance_mohm
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        label = d.pop("label")
+
+        motor_component_id = d.pop("motor_component_id")
+
+        propeller_component_id = d.pop("propeller_component_id")
+
+        throttle_pct = d.pop("throttle_pct")
+
+        count = d.pop("count", UNSET)
+
+        esc_motor_wire_resistance_mohm = d.pop("esc_motor_wire_resistance_mohm", UNSET)
+
+        esc_resistance_mohm = d.pop("esc_resistance_mohm", UNSET)
+
+        sweep_rotor_group_in = cls(
+            label=label,
+            motor_component_id=motor_component_id,
+            propeller_component_id=propeller_component_id,
+            throttle_pct=throttle_pct,
+            count=count,
+            esc_motor_wire_resistance_mohm=esc_motor_wire_resistance_mohm,
+            esc_resistance_mohm=esc_resistance_mohm,
+        )
+
+        return sweep_rotor_group_in

thrustlab/_models/types.py

@@ -0,0 +1,55 @@
+"""Shared types for generated model properties.
+
+This file is derived from the openapi-python-client types.py.jinja template and
+is required by the generated _models/ package.
+"""
+
+from collections.abc import Mapping, MutableMapping
+from http import HTTPStatus
+from typing import BinaryIO, Generic, IO, Literal, TypeVar
+
+from attrs import define
+
+
+class Unset:
+    def __bool__(self) -> Literal[False]:
+        return False
+
+
+UNSET: Unset = Unset()
+
+FileContent = IO[bytes] | bytes | str
+FileTypes = (
+    tuple[str | None, FileContent, str | None]
+    | tuple[str | None, FileContent, str | None, Mapping[str, str]]
+)
+RequestFiles = list[tuple[str, FileTypes]]
+
+
+@define
+class File:
+    """Contains information for file uploads."""
+
+    payload: BinaryIO
+    file_name: str | None = None
+    mime_type: str | None = None
+
+    def to_tuple(self) -> FileTypes:
+        """Return a tuple representation that httpx will accept for multipart/form-data."""
+        return self.file_name, self.payload, self.mime_type
+
+
+T = TypeVar("T")
+
+
+@define
+class Response(Generic[T]):
+    """A response from an endpoint."""
+
+    status_code: HTTPStatus
+    content: bytes
+    headers: MutableMapping[str, str]
+    parsed: T | None
+
+
+__all__ = ["UNSET", "File", "FileTypes", "RequestFiles", "Response", "Unset"]