mithril-client 0.1.0a1__cp314-cp314-macosx_11_0_arm64.whl

mithril/api/bindings/models/auction_model.py

@@ -0,0 +1,146 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, Self, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="AuctionModel")
+
+
+@_attrs_define
+class AuctionModel:
+    """Attributes:
+    fid (str):
+    instance_type (str):
+    region (str):
+    capacity (int):
+    last_instance_price (str):
+    adjustable_memory (bool):
+    lowest_allocated_price (None | str | Unset):
+    default_image_version (None | str | Unset):
+    """
+
+    fid: str
+    instance_type: str
+    region: str
+    capacity: int
+    last_instance_price: str
+    adjustable_memory: bool
+    lowest_allocated_price: None | str | Unset = UNSET
+    default_image_version: None | str | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        fid = self.fid
+
+        instance_type = self.instance_type
+
+        region = self.region
+
+        capacity = self.capacity
+
+        last_instance_price = self.last_instance_price
+
+        adjustable_memory = self.adjustable_memory
+
+        lowest_allocated_price: None | str | Unset
+        if isinstance(self.lowest_allocated_price, Unset):
+            lowest_allocated_price = UNSET
+        else:
+            lowest_allocated_price = self.lowest_allocated_price
+
+        default_image_version: None | str | Unset
+        if isinstance(self.default_image_version, Unset):
+            default_image_version = UNSET
+        else:
+            default_image_version = self.default_image_version
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "fid": fid,
+                "instance_type": instance_type,
+                "region": region,
+                "capacity": capacity,
+                "last_instance_price": last_instance_price,
+                "adjustable_memory": adjustable_memory,
+            }
+        )
+        if lowest_allocated_price is not UNSET:
+            field_dict["lowest_allocated_price"] = lowest_allocated_price
+        if default_image_version is not UNSET:
+            field_dict["default_image_version"] = default_image_version
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
+        d = dict(src_dict)
+        fid = d.pop("fid")
+
+        instance_type = d.pop("instance_type")
+
+        region = d.pop("region")
+
+        capacity = d.pop("capacity")
+
+        last_instance_price = d.pop("last_instance_price")
+
+        adjustable_memory = d.pop("adjustable_memory")
+
+        def _parse_lowest_allocated_price(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        lowest_allocated_price = _parse_lowest_allocated_price(
+            d.pop("lowest_allocated_price", UNSET)
+        )
+
+        def _parse_default_image_version(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        default_image_version = _parse_default_image_version(
+            d.pop("default_image_version", UNSET)
+        )
+
+        auction_model = cls(
+            fid=fid,
+            instance_type=instance_type,
+            region=region,
+            capacity=capacity,
+            last_instance_price=last_instance_price,
+            adjustable_memory=adjustable_memory,
+            lowest_allocated_price=lowest_allocated_price,
+            default_image_version=default_image_version,
+        )
+
+        auction_model.additional_properties = d
+        return auction_model
+
+    @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

mithril/api/bindings/models/availability_slot_model.py

@@ -0,0 +1,76 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, Self, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="AvailabilitySlotModel")
+
+
+@_attrs_define
+class AvailabilitySlotModel:
+    """Attributes:
+    start_time (str):
+    end_time (str):
+    quantity (int):
+    """
+
+    start_time: str
+    end_time: str
+    quantity: int
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        start_time = self.start_time
+
+        end_time = self.end_time
+
+        quantity = self.quantity
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "start_time": start_time,
+                "end_time": end_time,
+                "quantity": quantity,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
+        d = dict(src_dict)
+        start_time = d.pop("start_time")
+
+        end_time = d.pop("end_time")
+
+        quantity = d.pop("quantity")
+
+        availability_slot_model = cls(
+            start_time=start_time,
+            end_time=end_time,
+            quantity=quantity,
+        )
+
+        availability_slot_model.additional_properties = d
+        return availability_slot_model
+
+    @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

mithril/api/bindings/models/bid_history_event_model.py

@@ -0,0 +1,157 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, Self, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.bid_history_event_model_event_type import BidHistoryEventModelEventType
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.persistent_disk_change import PersistentDiskChange
+
+
+T = TypeVar("T", bound="BidHistoryEventModel")
+
+
+@_attrs_define
+class BidHistoryEventModel:
+    """A single event in the history of a spot bid.
+
+    Attributes:
+        timestamp (str): When the event occurred
+        event_type (BidHistoryEventModelEventType):
+        limit_price (None | str | Unset):
+        user_id (None | str | Unset):
+        memory_gb (int | None | Unset):
+        persistent_disk_change (PersistentDiskChange | Unset):
+    """
+
+    timestamp: str
+    event_type: BidHistoryEventModelEventType
+    limit_price: None | str | Unset = UNSET
+    user_id: None | str | Unset = UNSET
+    memory_gb: int | None | Unset = UNSET
+    persistent_disk_change: PersistentDiskChange | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        timestamp = self.timestamp
+
+        event_type = self.event_type.value
+
+        limit_price: None | str | Unset
+        if isinstance(self.limit_price, Unset):
+            limit_price = UNSET
+        else:
+            limit_price = self.limit_price
+
+        user_id: None | str | Unset
+        if isinstance(self.user_id, Unset):
+            user_id = UNSET
+        else:
+            user_id = self.user_id
+
+        memory_gb: int | None | Unset
+        if isinstance(self.memory_gb, Unset):
+            memory_gb = UNSET
+        else:
+            memory_gb = self.memory_gb
+
+        persistent_disk_change: dict[str, Any] | Unset = UNSET
+        if not isinstance(self.persistent_disk_change, Unset):
+            persistent_disk_change = self.persistent_disk_change.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "timestamp": timestamp,
+                "event_type": event_type,
+            }
+        )
+        if limit_price is not UNSET:
+            field_dict["limit_price"] = limit_price
+        if user_id is not UNSET:
+            field_dict["user_id"] = user_id
+        if memory_gb is not UNSET:
+            field_dict["memory_gb"] = memory_gb
+        if persistent_disk_change is not UNSET:
+            field_dict["persistent_disk_change"] = persistent_disk_change
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
+        from ..models.persistent_disk_change import PersistentDiskChange
+
+        d = dict(src_dict)
+        timestamp = d.pop("timestamp")
+
+        event_type = BidHistoryEventModelEventType(d.pop("event_type"))
+
+        def _parse_limit_price(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        limit_price = _parse_limit_price(d.pop("limit_price", UNSET))
+
+        def _parse_user_id(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        user_id = _parse_user_id(d.pop("user_id", UNSET))
+
+        def _parse_memory_gb(data: object) -> int | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(int | None | Unset, data)
+
+        memory_gb = _parse_memory_gb(d.pop("memory_gb", UNSET))
+
+        _persistent_disk_change = d.pop("persistent_disk_change", UNSET)
+        persistent_disk_change: PersistentDiskChange | Unset
+        if isinstance(_persistent_disk_change, Unset):
+            persistent_disk_change = UNSET
+        else:
+            persistent_disk_change = PersistentDiskChange.from_dict(
+                _persistent_disk_change
+            )
+
+        bid_history_event_model = cls(
+            timestamp=timestamp,
+            event_type=event_type,
+            limit_price=limit_price,
+            user_id=user_id,
+            memory_gb=memory_gb,
+            persistent_disk_change=persistent_disk_change,
+        )
+
+        bid_history_event_model.additional_properties = d
+        return bid_history_event_model
+
+    @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

mithril/api/bindings/models/bid_history_event_model_event_type.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+
+from enum import Enum
+
+
+class BidHistoryEventModelEventType(str, Enum):
+    ALLOCATED = "Allocated"
+    LIMIT_PRICE_CHANGED = "Limit Price Changed"
+    MEMORY_ADJUSTED = "Memory Adjusted"
+    PAUSED = "Paused"
+    PLACED = "Placed"
+    PREEMPTED = "Preempted"
+    RESUMED = "Resumed"
+    TERMINATED = "Terminated"
+    VOLUMES_ATTACHED = "Volumes Attached"
+    VOLUMES_REMOVED = "Volumes Removed"
+
+    def __str__(self) -> str:
+        return str(self.value)

mithril/api/bindings/models/bid_history_response.py

@@ -0,0 +1,84 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, Self, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.bid_history_event_model import BidHistoryEventModel
+
+
+T = TypeVar("T", bound="BidHistoryResponse")
+
+
+@_attrs_define
+class BidHistoryResponse:
+    """Response model for bid history.
+
+    Attributes:
+        bid_fid (str):
+        events (list[BidHistoryEventModel]): List of historical events for this bid, ordered by timestamp
+    """
+
+    bid_fid: str
+    events: list[BidHistoryEventModel]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        bid_fid = self.bid_fid
+
+        events = []
+        for events_item_data in self.events:
+            events_item = events_item_data.to_dict()
+            events.append(events_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "bid_fid": bid_fid,
+                "events": events,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
+        from ..models.bid_history_event_model import BidHistoryEventModel
+
+        d = dict(src_dict)
+        bid_fid = d.pop("bid_fid")
+
+        events = []
+        _events = d.pop("events")
+        for events_item_data in _events:
+            events_item = BidHistoryEventModel.from_dict(events_item_data)
+
+            events.append(events_item)
+
+        bid_history_response = cls(
+            bid_fid=bid_fid,
+            events=events,
+        )
+
+        bid_history_response.additional_properties = d
+        return bid_history_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

mithril/api/bindings/models/bid_model.py

@@ -0,0 +1,191 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, Self, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.bid_model_status import BidModelStatus
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.launch_specification_model import LaunchSpecificationModel
+
+
+T = TypeVar("T", bound="BidModel")
+
+
+@_attrs_define
+class BidModel:
+    """Attributes:
+    fid (str):
+    name (str):
+    project (str):
+    created_by (str):
+    created_at (str):
+    limit_price (str):
+    instance_quantity (int):
+    instance_type (str):
+    instances (list[str]):
+    launch_specification (LaunchSpecificationModel):
+    status (BidModelStatus):
+    deactivated_at (None | str | Unset):
+    region (None | str | Unset):
+    """
+
+    fid: str
+    name: str
+    project: str
+    created_by: str
+    created_at: str
+    limit_price: str
+    instance_quantity: int
+    instance_type: str
+    instances: list[str]
+    launch_specification: LaunchSpecificationModel
+    status: BidModelStatus
+    deactivated_at: None | str | Unset = UNSET
+    region: None | str | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        fid = self.fid
+
+        name = self.name
+
+        project = self.project
+
+        created_by = self.created_by
+
+        created_at = self.created_at
+
+        limit_price = self.limit_price
+
+        instance_quantity = self.instance_quantity
+
+        instance_type = self.instance_type
+
+        instances = self.instances
+
+        launch_specification = self.launch_specification.to_dict()
+
+        status = self.status.value
+
+        deactivated_at: None | str | Unset
+        if isinstance(self.deactivated_at, Unset):
+            deactivated_at = UNSET
+        else:
+            deactivated_at = self.deactivated_at
+
+        region: None | str | Unset
+        if isinstance(self.region, Unset):
+            region = UNSET
+        else:
+            region = self.region
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "fid": fid,
+                "name": name,
+                "project": project,
+                "created_by": created_by,
+                "created_at": created_at,
+                "limit_price": limit_price,
+                "instance_quantity": instance_quantity,
+                "instance_type": instance_type,
+                "instances": instances,
+                "launch_specification": launch_specification,
+                "status": status,
+            }
+        )
+        if deactivated_at is not UNSET:
+            field_dict["deactivated_at"] = deactivated_at
+        if region is not UNSET:
+            field_dict["region"] = region
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
+        from ..models.launch_specification_model import LaunchSpecificationModel
+
+        d = dict(src_dict)
+        fid = d.pop("fid")
+
+        name = d.pop("name")
+
+        project = d.pop("project")
+
+        created_by = d.pop("created_by")
+
+        created_at = d.pop("created_at")
+
+        limit_price = d.pop("limit_price")
+
+        instance_quantity = d.pop("instance_quantity")
+
+        instance_type = d.pop("instance_type")
+
+        instances = cast(list[str], d.pop("instances"))
+
+        launch_specification = LaunchSpecificationModel.from_dict(
+            d.pop("launch_specification")
+        )
+
+        status = BidModelStatus(d.pop("status"))
+
+        def _parse_deactivated_at(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        deactivated_at = _parse_deactivated_at(d.pop("deactivated_at", UNSET))
+
+        def _parse_region(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        region = _parse_region(d.pop("region", UNSET))
+
+        bid_model = cls(
+            fid=fid,
+            name=name,
+            project=project,
+            created_by=created_by,
+            created_at=created_at,
+            limit_price=limit_price,
+            instance_quantity=instance_quantity,
+            instance_type=instance_type,
+            instances=instances,
+            launch_specification=launch_specification,
+            status=status,
+            deactivated_at=deactivated_at,
+            region=region,
+        )
+
+        bid_model.additional_properties = d
+        return bid_model
+
+    @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

mithril/api/bindings/models/bid_model_status.py

@@ -0,0 +1,14 @@
+from __future__ import annotations
+
+from enum import Enum
+
+
+class BidModelStatus(str, Enum):
+    ALLOCATED = "Allocated"
+    OPEN = "Open"
+    PAUSED = "Paused"
+    PREEMPTING = "Preempting"
+    TERMINATED = "Terminated"
+
+    def __str__(self) -> str:
+        return str(self.value)