waldur-api-client 7.8.3__py3-none-any.whl → 7.8.5__py3-none-any.whl

waldur_api_client/models/offering_create.py

@@ -19,8 +19,10 @@ if TYPE_CHECKING:
     from ..models.merged_secret_options import MergedSecretOptions
     from ..models.nested_endpoint import NestedEndpoint
     from ..models.nested_offering_file import NestedOfferingFile
+    from ..models.nested_partition import NestedPartition
     from ..models.nested_role import NestedRole
     from ..models.nested_screenshot import NestedScreenshot
+    from ..models.nested_software_catalog import NestedSoftwareCatalog
     from ..models.offering_component import OfferingComponent
     from ..models.offering_create_service_attributes import OfferingCreateServiceAttributes
     from ..models.offering_options import OfferingOptions
@@ -40,6 +42,8 @@ class OfferingCreate:
         created (datetime.datetime):
         name (str):
         endpoints (list['NestedEndpoint']):
+        software_catalogs (list['NestedSoftwareCatalog']):
+        partitions (list['NestedPartition']):
         roles (list['NestedRole']):
         customer_uuid (Union[None, UUID]):
         customer_name (Union[None, str]):
@@ -105,6 +109,8 @@ class OfferingCreate:
     created: datetime.datetime
     name: str
     endpoints: list["NestedEndpoint"]
+    software_catalogs: list["NestedSoftwareCatalog"]
+    partitions: list["NestedPartition"]
     roles: list["NestedRole"]
     customer_uuid: Union[None, UUID]
     customer_name: Union[None, str]
@@ -179,6 +185,16 @@ class OfferingCreate:
             endpoints_item = endpoints_item_data.to_dict()
             endpoints.append(endpoints_item)
 
+        software_catalogs = []
+        for software_catalogs_item_data in self.software_catalogs:
+            software_catalogs_item = software_catalogs_item_data.to_dict()
+            software_catalogs.append(software_catalogs_item)
+
+        partitions = []
+        for partitions_item_data in self.partitions:
+            partitions_item = partitions_item_data.to_dict()
+            partitions.append(partitions_item)
+
         roles = []
         for roles_item_data in self.roles:
             roles_item = roles_item_data.to_dict()
@@ -396,6 +412,8 @@ class OfferingCreate:
                 "created": created,
                 "name": name,
                 "endpoints": endpoints,
+                "software_catalogs": software_catalogs,
+                "partitions": partitions,
                 "roles": roles,
                 "customer_uuid": customer_uuid,
                 "customer_name": customer_name,
@@ -491,8 +509,10 @@ class OfferingCreate:
         from ..models.merged_secret_options import MergedSecretOptions
         from ..models.nested_endpoint import NestedEndpoint
         from ..models.nested_offering_file import NestedOfferingFile
+        from ..models.nested_partition import NestedPartition
         from ..models.nested_role import NestedRole
         from ..models.nested_screenshot import NestedScreenshot
+        from ..models.nested_software_catalog import NestedSoftwareCatalog
         from ..models.offering_component import OfferingComponent
         from ..models.offering_create_service_attributes import OfferingCreateServiceAttributes
         from ..models.offering_options import OfferingOptions
@@ -515,6 +535,20 @@ class OfferingCreate:
 
             endpoints.append(endpoints_item)
 
+        software_catalogs = []
+        _software_catalogs = d.pop("software_catalogs")
+        for software_catalogs_item_data in _software_catalogs:
+            software_catalogs_item = NestedSoftwareCatalog.from_dict(software_catalogs_item_data)
+
+            software_catalogs.append(software_catalogs_item)
+
+        partitions = []
+        _partitions = d.pop("partitions")
+        for partitions_item_data in _partitions:
+            partitions_item = NestedPartition.from_dict(partitions_item_data)
+
+            partitions.append(partitions_item)
+
         roles = []
         _roles = d.pop("roles")
         for roles_item_data in _roles:
@@ -864,6 +898,8 @@ class OfferingCreate:
             created=created,
             name=name,
             endpoints=endpoints,
+            software_catalogs=software_catalogs,
+            partitions=partitions,
             roles=roles,
             customer_uuid=customer_uuid,
             customer_name=customer_name,

waldur_api_client/models/offering_partition.py

@@ -0,0 +1,440 @@
+import datetime
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="OfferingPartition")
+
+
+@_attrs_define
+class OfferingPartition:
+    """
+    Attributes:
+        uuid (UUID):
+        created (datetime.datetime):
+        modified (datetime.datetime):
+        offering (UUID):
+        offering_name (str):
+        partition_name (str): Name of the SLURM partition
+        cpu_bind (Union[None, Unset, int]): Default task binding policy (SLURM cpu_bind)
+        def_cpu_per_gpu (Union[None, Unset, int]): Default CPUs allocated per GPU
+        max_cpus_per_node (Union[None, Unset, int]): Maximum allocated CPUs per node
+        max_cpus_per_socket (Union[None, Unset, int]): Maximum allocated CPUs per socket
+        def_mem_per_cpu (Union[None, Unset, int]): Default memory per CPU in MB
+        def_mem_per_gpu (Union[None, Unset, int]): Default memory per GPU in MB
+        def_mem_per_node (Union[None, Unset, int]): Default memory per node in MB
+        max_mem_per_cpu (Union[None, Unset, int]): Maximum memory per CPU in MB
+        max_mem_per_node (Union[None, Unset, int]): Maximum memory per node in MB
+        default_time (Union[None, Unset, int]): Default time limit in minutes
+        max_time (Union[None, Unset, int]): Maximum time limit in minutes
+        grace_time (Union[None, Unset, int]): Preemption grace time in seconds
+        max_nodes (Union[None, Unset, int]): Maximum nodes per job
+        min_nodes (Union[None, Unset, int]): Minimum nodes per job
+        exclusive_topo (Union[Unset, bool]): Exclusive topology access required
+        exclusive_user (Union[Unset, bool]): Exclusive user access required
+        priority_tier (Union[None, Unset, int]): Priority tier for scheduling and preemption
+        qos (Union[Unset, str]): Quality of Service (QOS) name
+        req_resv (Union[Unset, bool]): Require reservation for job allocation
+    """
+
+    uuid: UUID
+    created: datetime.datetime
+    modified: datetime.datetime
+    offering: UUID
+    offering_name: str
+    partition_name: str
+    cpu_bind: Union[None, Unset, int] = UNSET
+    def_cpu_per_gpu: Union[None, Unset, int] = UNSET
+    max_cpus_per_node: Union[None, Unset, int] = UNSET
+    max_cpus_per_socket: Union[None, Unset, int] = UNSET
+    def_mem_per_cpu: Union[None, Unset, int] = UNSET
+    def_mem_per_gpu: Union[None, Unset, int] = UNSET
+    def_mem_per_node: Union[None, Unset, int] = UNSET
+    max_mem_per_cpu: Union[None, Unset, int] = UNSET
+    max_mem_per_node: Union[None, Unset, int] = UNSET
+    default_time: Union[None, Unset, int] = UNSET
+    max_time: Union[None, Unset, int] = UNSET
+    grace_time: Union[None, Unset, int] = UNSET
+    max_nodes: Union[None, Unset, int] = UNSET
+    min_nodes: Union[None, Unset, int] = UNSET
+    exclusive_topo: Union[Unset, bool] = UNSET
+    exclusive_user: Union[Unset, bool] = UNSET
+    priority_tier: Union[None, Unset, int] = UNSET
+    qos: Union[Unset, str] = UNSET
+    req_resv: Union[Unset, bool] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        uuid = str(self.uuid)
+
+        created = self.created.isoformat()
+
+        modified = self.modified.isoformat()
+
+        offering = str(self.offering)
+
+        offering_name = self.offering_name
+
+        partition_name = self.partition_name
+
+        cpu_bind: Union[None, Unset, int]
+        if isinstance(self.cpu_bind, Unset):
+            cpu_bind = UNSET
+        else:
+            cpu_bind = self.cpu_bind
+
+        def_cpu_per_gpu: Union[None, Unset, int]
+        if isinstance(self.def_cpu_per_gpu, Unset):
+            def_cpu_per_gpu = UNSET
+        else:
+            def_cpu_per_gpu = self.def_cpu_per_gpu
+
+        max_cpus_per_node: Union[None, Unset, int]
+        if isinstance(self.max_cpus_per_node, Unset):
+            max_cpus_per_node = UNSET
+        else:
+            max_cpus_per_node = self.max_cpus_per_node
+
+        max_cpus_per_socket: Union[None, Unset, int]
+        if isinstance(self.max_cpus_per_socket, Unset):
+            max_cpus_per_socket = UNSET
+        else:
+            max_cpus_per_socket = self.max_cpus_per_socket
+
+        def_mem_per_cpu: Union[None, Unset, int]
+        if isinstance(self.def_mem_per_cpu, Unset):
+            def_mem_per_cpu = UNSET
+        else:
+            def_mem_per_cpu = self.def_mem_per_cpu
+
+        def_mem_per_gpu: Union[None, Unset, int]
+        if isinstance(self.def_mem_per_gpu, Unset):
+            def_mem_per_gpu = UNSET
+        else:
+            def_mem_per_gpu = self.def_mem_per_gpu
+
+        def_mem_per_node: Union[None, Unset, int]
+        if isinstance(self.def_mem_per_node, Unset):
+            def_mem_per_node = UNSET
+        else:
+            def_mem_per_node = self.def_mem_per_node
+
+        max_mem_per_cpu: Union[None, Unset, int]
+        if isinstance(self.max_mem_per_cpu, Unset):
+            max_mem_per_cpu = UNSET
+        else:
+            max_mem_per_cpu = self.max_mem_per_cpu
+
+        max_mem_per_node: Union[None, Unset, int]
+        if isinstance(self.max_mem_per_node, Unset):
+            max_mem_per_node = UNSET
+        else:
+            max_mem_per_node = self.max_mem_per_node
+
+        default_time: Union[None, Unset, int]
+        if isinstance(self.default_time, Unset):
+            default_time = UNSET
+        else:
+            default_time = self.default_time
+
+        max_time: Union[None, Unset, int]
+        if isinstance(self.max_time, Unset):
+            max_time = UNSET
+        else:
+            max_time = self.max_time
+
+        grace_time: Union[None, Unset, int]
+        if isinstance(self.grace_time, Unset):
+            grace_time = UNSET
+        else:
+            grace_time = self.grace_time
+
+        max_nodes: Union[None, Unset, int]
+        if isinstance(self.max_nodes, Unset):
+            max_nodes = UNSET
+        else:
+            max_nodes = self.max_nodes
+
+        min_nodes: Union[None, Unset, int]
+        if isinstance(self.min_nodes, Unset):
+            min_nodes = UNSET
+        else:
+            min_nodes = self.min_nodes
+
+        exclusive_topo = self.exclusive_topo
+
+        exclusive_user = self.exclusive_user
+
+        priority_tier: Union[None, Unset, int]
+        if isinstance(self.priority_tier, Unset):
+            priority_tier = UNSET
+        else:
+            priority_tier = self.priority_tier
+
+        qos = self.qos
+
+        req_resv = self.req_resv
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "uuid": uuid,
+                "created": created,
+                "modified": modified,
+                "offering": offering,
+                "offering_name": offering_name,
+                "partition_name": partition_name,
+            }
+        )
+        if cpu_bind is not UNSET:
+            field_dict["cpu_bind"] = cpu_bind
+        if def_cpu_per_gpu is not UNSET:
+            field_dict["def_cpu_per_gpu"] = def_cpu_per_gpu
+        if max_cpus_per_node is not UNSET:
+            field_dict["max_cpus_per_node"] = max_cpus_per_node
+        if max_cpus_per_socket is not UNSET:
+            field_dict["max_cpus_per_socket"] = max_cpus_per_socket
+        if def_mem_per_cpu is not UNSET:
+            field_dict["def_mem_per_cpu"] = def_mem_per_cpu
+        if def_mem_per_gpu is not UNSET:
+            field_dict["def_mem_per_gpu"] = def_mem_per_gpu
+        if def_mem_per_node is not UNSET:
+            field_dict["def_mem_per_node"] = def_mem_per_node
+        if max_mem_per_cpu is not UNSET:
+            field_dict["max_mem_per_cpu"] = max_mem_per_cpu
+        if max_mem_per_node is not UNSET:
+            field_dict["max_mem_per_node"] = max_mem_per_node
+        if default_time is not UNSET:
+            field_dict["default_time"] = default_time
+        if max_time is not UNSET:
+            field_dict["max_time"] = max_time
+        if grace_time is not UNSET:
+            field_dict["grace_time"] = grace_time
+        if max_nodes is not UNSET:
+            field_dict["max_nodes"] = max_nodes
+        if min_nodes is not UNSET:
+            field_dict["min_nodes"] = min_nodes
+        if exclusive_topo is not UNSET:
+            field_dict["exclusive_topo"] = exclusive_topo
+        if exclusive_user is not UNSET:
+            field_dict["exclusive_user"] = exclusive_user
+        if priority_tier is not UNSET:
+            field_dict["priority_tier"] = priority_tier
+        if qos is not UNSET:
+            field_dict["qos"] = qos
+        if req_resv is not UNSET:
+            field_dict["req_resv"] = req_resv
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        uuid = UUID(d.pop("uuid"))
+
+        created = isoparse(d.pop("created"))
+
+        modified = isoparse(d.pop("modified"))
+
+        offering = UUID(d.pop("offering"))
+
+        offering_name = d.pop("offering_name")
+
+        partition_name = d.pop("partition_name")
+
+        def _parse_cpu_bind(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        cpu_bind = _parse_cpu_bind(d.pop("cpu_bind", UNSET))
+
+        def _parse_def_cpu_per_gpu(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        def_cpu_per_gpu = _parse_def_cpu_per_gpu(d.pop("def_cpu_per_gpu", UNSET))
+
+        def _parse_max_cpus_per_node(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        max_cpus_per_node = _parse_max_cpus_per_node(d.pop("max_cpus_per_node", UNSET))
+
+        def _parse_max_cpus_per_socket(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        max_cpus_per_socket = _parse_max_cpus_per_socket(d.pop("max_cpus_per_socket", UNSET))
+
+        def _parse_def_mem_per_cpu(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        def_mem_per_cpu = _parse_def_mem_per_cpu(d.pop("def_mem_per_cpu", UNSET))
+
+        def _parse_def_mem_per_gpu(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        def_mem_per_gpu = _parse_def_mem_per_gpu(d.pop("def_mem_per_gpu", UNSET))
+
+        def _parse_def_mem_per_node(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        def_mem_per_node = _parse_def_mem_per_node(d.pop("def_mem_per_node", UNSET))
+
+        def _parse_max_mem_per_cpu(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        max_mem_per_cpu = _parse_max_mem_per_cpu(d.pop("max_mem_per_cpu", UNSET))
+
+        def _parse_max_mem_per_node(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        max_mem_per_node = _parse_max_mem_per_node(d.pop("max_mem_per_node", UNSET))
+
+        def _parse_default_time(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        default_time = _parse_default_time(d.pop("default_time", UNSET))
+
+        def _parse_max_time(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        max_time = _parse_max_time(d.pop("max_time", UNSET))
+
+        def _parse_grace_time(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        grace_time = _parse_grace_time(d.pop("grace_time", UNSET))
+
+        def _parse_max_nodes(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        max_nodes = _parse_max_nodes(d.pop("max_nodes", UNSET))
+
+        def _parse_min_nodes(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        min_nodes = _parse_min_nodes(d.pop("min_nodes", UNSET))
+
+        exclusive_topo = d.pop("exclusive_topo", UNSET)
+
+        exclusive_user = d.pop("exclusive_user", UNSET)
+
+        def _parse_priority_tier(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        priority_tier = _parse_priority_tier(d.pop("priority_tier", UNSET))
+
+        qos = d.pop("qos", UNSET)
+
+        req_resv = d.pop("req_resv", UNSET)
+
+        offering_partition = cls(
+            uuid=uuid,
+            created=created,
+            modified=modified,
+            offering=offering,
+            offering_name=offering_name,
+            partition_name=partition_name,
+            cpu_bind=cpu_bind,
+            def_cpu_per_gpu=def_cpu_per_gpu,
+            max_cpus_per_node=max_cpus_per_node,
+            max_cpus_per_socket=max_cpus_per_socket,
+            def_mem_per_cpu=def_mem_per_cpu,
+            def_mem_per_gpu=def_mem_per_gpu,
+            def_mem_per_node=def_mem_per_node,
+            max_mem_per_cpu=max_mem_per_cpu,
+            max_mem_per_node=max_mem_per_node,
+            default_time=default_time,
+            max_time=max_time,
+            grace_time=grace_time,
+            max_nodes=max_nodes,
+            min_nodes=min_nodes,
+            exclusive_topo=exclusive_topo,
+            exclusive_user=exclusive_user,
+            priority_tier=priority_tier,
+            qos=qos,
+            req_resv=req_resv,
+        )
+
+        offering_partition.additional_properties = d
+        return offering_partition
+
+    @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