streamtime-sdk 1.1.17__py3-none-any.whl

streamtime_sdk/models/cluster_bench_mark.py

@@ -0,0 +1,309 @@
+from __future__ import annotations
+
+import datetime
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.cluster_bench_mark_status_enum import ClusterBenchMarkStatusEnum
+from ..models.configuration_state_7a4_enum import ConfigurationState7A4Enum
+from ..models.desired_lifecycle_0b2_enum import DesiredLifecycle0B2Enum
+from ..models.reconciliation_state_fb_0_enum import ReconciliationStateFb0Enum
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ClusterBenchMark")
+
+
+@_attrs_define
+class ClusterBenchMark:
+    """Base serializer for all ADRF model serializers.
+
+    Attributes:
+        id (UUID):
+        organization (str): An alphanumeric string (< 24 characters) containing only lowercase characters, "-", and "_",
+            starting with a letter or number.
+        kafka (str): An alphanumeric string (< 24 characters) containing only lowercase characters, "-", and "_",
+            starting with a letter or number.
+        aggregate_throughput (str):
+        producer_latency (str):
+        identifier (str): Server-generated from this row's own id.
+        mode (str):
+        status (ClusterBenchMarkStatusEnum): * `Incomplete` - Incomplete
+            * `Pending` - Pending
+            * `Running` - Running
+            * `Completed` - Completed
+            * `Failed` - Failed
+            * `Cancelled` - Cancelled
+        configuration_state (ConfigurationState7A4Enum): * `incomplete` - Incomplete
+            * `valid` - Valid
+            * `invalid` - Invalid
+        generation (int): Incremented whenever deployable desired state changes. Reconcile dispatches this generation's
+            snapshot.
+        observed_generation (int): Latest generation successfully reconciled. 0 means never reconciled.
+        desired_lifecycle (DesiredLifecycle0B2Enum): * `present` - Present
+            * `absent` - Absent
+        reconciliation_state (ReconciliationStateFb0Enum): * `never_reconciled` - Never reconciled
+            * `pending` - Pending
+            * `queued` - Queued
+            * `running` - Running
+            * `succeeded` - Succeeded
+            * `failed` - Failed
+        workflow_id (None | str):
+        created_at (datetime.datetime):
+        updated_at (datetime.datetime):
+        topic (str | Unset):
+        scenario_name (str | Unset):
+        placement_config (Any | Unset): BenchStress placement config (tolerations/matchExpressions + optional
+            kubeconfig).
+        advanced_config (Any | Unset): BenchStress advanced config (mode-specific: simple or advanced).
+        started_at (datetime.datetime | None | Unset):
+        completed_at (datetime.datetime | None | Unset):
+    """
+
+    id: UUID
+    organization: str
+    kafka: str
+    aggregate_throughput: str
+    producer_latency: str
+    identifier: str
+    mode: str
+    status: ClusterBenchMarkStatusEnum
+    configuration_state: ConfigurationState7A4Enum
+    generation: int
+    observed_generation: int
+    desired_lifecycle: DesiredLifecycle0B2Enum
+    reconciliation_state: ReconciliationStateFb0Enum
+    workflow_id: None | str
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    topic: str | Unset = UNSET
+    scenario_name: str | Unset = UNSET
+    placement_config: Any | Unset = UNSET
+    advanced_config: Any | Unset = UNSET
+    started_at: datetime.datetime | None | Unset = UNSET
+    completed_at: datetime.datetime | None | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = str(self.id)
+
+        organization = self.organization
+
+        kafka = self.kafka
+
+        aggregate_throughput = self.aggregate_throughput
+
+        producer_latency = self.producer_latency
+
+        identifier = self.identifier
+
+        mode = self.mode
+
+        status = self.status.value
+
+        configuration_state = self.configuration_state.value
+
+        generation = self.generation
+
+        observed_generation = self.observed_generation
+
+        desired_lifecycle = self.desired_lifecycle.value
+
+        reconciliation_state = self.reconciliation_state.value
+
+        workflow_id: None | str
+        workflow_id = self.workflow_id
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        topic = self.topic
+
+        scenario_name = self.scenario_name
+
+        placement_config = self.placement_config
+
+        advanced_config = self.advanced_config
+
+        started_at: None | str | Unset
+        if isinstance(self.started_at, Unset):
+            started_at = UNSET
+        elif isinstance(self.started_at, datetime.datetime):
+            started_at = self.started_at.isoformat()
+        else:
+            started_at = self.started_at
+
+        completed_at: None | str | Unset
+        if isinstance(self.completed_at, Unset):
+            completed_at = UNSET
+        elif isinstance(self.completed_at, datetime.datetime):
+            completed_at = self.completed_at.isoformat()
+        else:
+            completed_at = self.completed_at
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "organization": organization,
+                "kafka": kafka,
+                "aggregate_throughput": aggregate_throughput,
+                "producer_latency": producer_latency,
+                "identifier": identifier,
+                "mode": mode,
+                "status": status,
+                "configuration_state": configuration_state,
+                "generation": generation,
+                "observed_generation": observed_generation,
+                "desired_lifecycle": desired_lifecycle,
+                "reconciliation_state": reconciliation_state,
+                "workflow_id": workflow_id,
+                "created_at": created_at,
+                "updated_at": updated_at,
+            }
+        )
+        if topic is not UNSET:
+            field_dict["topic"] = topic
+        if scenario_name is not UNSET:
+            field_dict["scenario_name"] = scenario_name
+        if placement_config is not UNSET:
+            field_dict["placement_config"] = placement_config
+        if advanced_config is not UNSET:
+            field_dict["advanced_config"] = advanced_config
+        if started_at is not UNSET:
+            field_dict["started_at"] = started_at
+        if completed_at is not UNSET:
+            field_dict["completed_at"] = completed_at
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        id = UUID(d.pop("id"))
+
+        organization = d.pop("organization")
+
+        kafka = d.pop("kafka")
+
+        aggregate_throughput = d.pop("aggregate_throughput")
+
+        producer_latency = d.pop("producer_latency")
+
+        identifier = d.pop("identifier")
+
+        mode = d.pop("mode")
+
+        status = ClusterBenchMarkStatusEnum(d.pop("status"))
+
+        configuration_state = ConfigurationState7A4Enum(d.pop("configuration_state"))
+
+        generation = d.pop("generation")
+
+        observed_generation = d.pop("observed_generation")
+
+        desired_lifecycle = DesiredLifecycle0B2Enum(d.pop("desired_lifecycle"))
+
+        reconciliation_state = ReconciliationStateFb0Enum(d.pop("reconciliation_state"))
+
+        def _parse_workflow_id(data: object) -> None | str:
+            if data is None:
+                return data
+            return cast(None | str, data)
+
+        workflow_id = _parse_workflow_id(d.pop("workflow_id"))
+
+        created_at = datetime.datetime.fromisoformat(d.pop("created_at"))
+
+        updated_at = datetime.datetime.fromisoformat(d.pop("updated_at"))
+
+        topic = d.pop("topic", UNSET)
+
+        scenario_name = d.pop("scenario_name", UNSET)
+
+        placement_config = d.pop("placement_config", UNSET)
+
+        advanced_config = d.pop("advanced_config", UNSET)
+
+        def _parse_started_at(data: object) -> datetime.datetime | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                started_at_type_0 = datetime.datetime.fromisoformat(data)
+
+                return started_at_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(datetime.datetime | None | Unset, data)
+
+        started_at = _parse_started_at(d.pop("started_at", UNSET))
+
+        def _parse_completed_at(data: object) -> datetime.datetime | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                completed_at_type_0 = datetime.datetime.fromisoformat(data)
+
+                return completed_at_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(datetime.datetime | None | Unset, data)
+
+        completed_at = _parse_completed_at(d.pop("completed_at", UNSET))
+
+        cluster_bench_mark = cls(
+            id=id,
+            organization=organization,
+            kafka=kafka,
+            aggregate_throughput=aggregate_throughput,
+            producer_latency=producer_latency,
+            identifier=identifier,
+            mode=mode,
+            status=status,
+            configuration_state=configuration_state,
+            generation=generation,
+            observed_generation=observed_generation,
+            desired_lifecycle=desired_lifecycle,
+            reconciliation_state=reconciliation_state,
+            workflow_id=workflow_id,
+            created_at=created_at,
+            updated_at=updated_at,
+            topic=topic,
+            scenario_name=scenario_name,
+            placement_config=placement_config,
+            advanced_config=advanced_config,
+            started_at=started_at,
+            completed_at=completed_at,
+        )
+
+        cluster_bench_mark.additional_properties = d
+        return cluster_bench_mark
+
+    @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

streamtime_sdk/models/cluster_bench_mark_status_enum.py

@@ -0,0 +1,13 @@
+from enum import StrEnum
+
+
+class ClusterBenchMarkStatusEnum(StrEnum):
+    CANCELLED = "Cancelled"
+    COMPLETED = "Completed"
+    FAILED = "Failed"
+    INCOMPLETE = "Incomplete"
+    PENDING = "Pending"
+    RUNNING = "Running"
+
+    def __str__(self) -> str:
+        return str(self.value)

streamtime_sdk/models/cluster_benchmark_result.py

@@ -0,0 +1,110 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.cluster_benchmark_result_type_enum import ClusterBenchmarkResultTypeEnum
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ClusterBenchmarkResult")
+
+
+@_attrs_define
+class ClusterBenchmarkResult:
+    """Base serializer for all ADRF model serializers.
+
+    Attributes:
+        id (UUID):
+        type_ (ClusterBenchmarkResultTypeEnum): * `Producer` - Producer
+            * `Consumer` - Consumer
+            * `Server` - Server
+        name (str):
+        cluster_benchmark (UUID):
+        result (Any | Unset):
+        config (Any | Unset):
+    """
+
+    id: UUID
+    type_: ClusterBenchmarkResultTypeEnum
+    name: str
+    cluster_benchmark: UUID
+    result: Any | Unset = UNSET
+    config: Any | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = str(self.id)
+
+        type_ = self.type_.value
+
+        name = self.name
+
+        cluster_benchmark = str(self.cluster_benchmark)
+
+        result = self.result
+
+        config = self.config
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "type": type_,
+                "name": name,
+                "cluster_benchmark": cluster_benchmark,
+            }
+        )
+        if result is not UNSET:
+            field_dict["result"] = result
+        if config is not UNSET:
+            field_dict["config"] = config
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        id = UUID(d.pop("id"))
+
+        type_ = ClusterBenchmarkResultTypeEnum(d.pop("type"))
+
+        name = d.pop("name")
+
+        cluster_benchmark = UUID(d.pop("cluster_benchmark"))
+
+        result = d.pop("result", UNSET)
+
+        config = d.pop("config", UNSET)
+
+        cluster_benchmark_result = cls(
+            id=id,
+            type_=type_,
+            name=name,
+            cluster_benchmark=cluster_benchmark,
+            result=result,
+            config=config,
+        )
+
+        cluster_benchmark_result.additional_properties = d
+        return cluster_benchmark_result
+
+    @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

streamtime_sdk/models/cluster_benchmark_result_type_enum.py

@@ -0,0 +1,10 @@
+from enum import StrEnum
+
+
+class ClusterBenchmarkResultTypeEnum(StrEnum):
+    CONSUMER = "Consumer"
+    PRODUCER = "Producer"
+    SERVER = "Server"
+
+    def __str__(self) -> str:
+        return str(self.value)

streamtime_sdk/models/cluster_history.py

@@ -0,0 +1,139 @@
+from __future__ import annotations
+
+import datetime
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.cluster_type_enum import ClusterTypeEnum
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ClusterHistory")
+
+
+@_attrs_define
+class ClusterHistory:
+    """Base serializer for all ADRF model serializers.
+
+    Attributes:
+        id (UUID):
+        organization (str): An alphanumeric string (< 24 characters) containing only lowercase characters, "-", and "_",
+            starting with a letter or number.
+        cluster_type (ClusterTypeEnum): * `Fleet` - Fleet
+            * `Kafka` - Kafka
+        provider_sku (str):
+        cluster_id (str):
+        created (datetime.datetime):
+        deleted (datetime.datetime | None | Unset):
+    """
+
+    id: UUID
+    organization: str
+    cluster_type: ClusterTypeEnum
+    provider_sku: str
+    cluster_id: str
+    created: datetime.datetime
+    deleted: datetime.datetime | None | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = str(self.id)
+
+        organization = self.organization
+
+        cluster_type = self.cluster_type.value
+
+        provider_sku = self.provider_sku
+
+        cluster_id = self.cluster_id
+
+        created = self.created.isoformat()
+
+        deleted: None | str | Unset
+        if isinstance(self.deleted, Unset):
+            deleted = UNSET
+        elif isinstance(self.deleted, datetime.datetime):
+            deleted = self.deleted.isoformat()
+        else:
+            deleted = self.deleted
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "organization": organization,
+                "cluster_type": cluster_type,
+                "provider_sku": provider_sku,
+                "cluster_id": cluster_id,
+                "created": created,
+            }
+        )
+        if deleted is not UNSET:
+            field_dict["deleted"] = deleted
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        id = UUID(d.pop("id"))
+
+        organization = d.pop("organization")
+
+        cluster_type = ClusterTypeEnum(d.pop("cluster_type"))
+
+        provider_sku = d.pop("provider_sku")
+
+        cluster_id = d.pop("cluster_id")
+
+        created = datetime.datetime.fromisoformat(d.pop("created"))
+
+        def _parse_deleted(data: object) -> datetime.datetime | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                deleted_type_0 = datetime.datetime.fromisoformat(data)
+
+                return deleted_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(datetime.datetime | None | Unset, data)
+
+        deleted = _parse_deleted(d.pop("deleted", UNSET))
+
+        cluster_history = cls(
+            id=id,
+            organization=organization,
+            cluster_type=cluster_type,
+            provider_sku=provider_sku,
+            cluster_id=cluster_id,
+            created=created,
+            deleted=deleted,
+        )
+
+        cluster_history.additional_properties = d
+        return cluster_history
+
+    @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