cluster_status.py

# coding: utf-8

"""
    STACKIT Kubernetes Engine API

    The SKE API provides endpoints to create, update or delete clusters within STACKIT projects and to trigger further cluster management tasks.

    The version of the OpenAPI document: 2.0
    Generated by OpenAPI Generator (https://openapi-generator.tech)

    Do not edit the class manually.
"""  # noqa: E501

from __future__ import annotations

import json
import pprint
import re  # noqa: F401
from datetime import datetime
from typing import Any, ClassVar, Dict, List, Optional, Set

from pydantic import (
    BaseModel,
    ConfigDict,
    Field,
    StrictBool,
    StrictStr,
    field_validator,
)
from typing_extensions import Self

from stackit.ske.models.cluster_error import ClusterError
from stackit.ske.models.cluster_status_state import ClusterStatusState
from stackit.ske.models.credentials_rotation_state import CredentialsRotationState
from stackit.ske.models.expiratoaion_status import ExpiratoaionStatus
from stackit.ske.models.runtime_error import RuntimeError


class ClusterStatus(BaseModel):
    """
    ClusterStatus
    """  # noqa: E501

    aggregated: Optional[ClusterStatusState] = None
    creation_time: Optional[datetime] = Field(
        default=None, description="Format: `2024-02-15T11:06:29Z`", alias="creationTime"
    )
    credentials_rotation: Optional[CredentialsRotationState] = Field(default=None, alias="credentialsRotation")
    egress_address_ranges: Optional[List[StrictStr]] = Field(
        default=None,
        description="The outgoing network ranges (in CIDR notation) of traffic originating from workload on the cluster.",
        alias="egressAddressRanges",
    )
    error: Optional[RuntimeError] = None
    errors: Optional[List[ClusterError]] = None
    expiration: Optional[ExpiratoaionStatus] = None
    hibernated: Optional[StrictBool] = None
    identity: Optional[StrictStr] = None
    pod_address_ranges: Optional[List[StrictStr]] = Field(
        default=None,
        description="The network ranges (in CIDR notation) used by pods of the cluster.",
        alias="podAddressRanges",
    )
    __properties: ClassVar[List[str]] = [
        "aggregated",
        "creationTime",
        "credentialsRotation",
        "egressAddressRanges",
        "error",
        "errors",
        "expiration",
        "hibernated",
        "identity",
        "podAddressRanges",
    ]

    @field_validator("creation_time", mode="before")
    def creation_time_change_year_zero_to_one(cls, value):
        """Workaround which prevents year 0 issue"""
        if isinstance(value, str):
            # Check for year "0000" at the beginning of the string
            # This assumes common date formats like YYYY-MM-DDTHH:MM:SS+00:00 or YYYY-MM-DDTHH:MM:SSZ
            if value.startswith("0000-01-01T") and re.match(
                r"^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(\+\d{2}:\d{2}|Z)$", value
            ):
                # Workaround: Replace "0000" with "0001"
                return "0001" + value[4:]  # Take "0001" and append the rest of the string
        return value

    model_config = ConfigDict(
        populate_by_name=True,
        validate_assignment=True,
        protected_namespaces=(),
    )

    def to_str(self) -> str:
        """Returns the string representation of the model using alias"""
        return pprint.pformat(self.model_dump(by_alias=True))

    def to_json(self) -> str:
        """Returns the JSON representation of the model using alias"""
        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
        return json.dumps(self.to_dict())

    @classmethod
    def from_json(cls, json_str: str) -> Optional[Self]:
        """Create an instance of ClusterStatus from a JSON string"""
        return cls.from_dict(json.loads(json_str))

    def to_dict(self) -> Dict[str, Any]:
        """Return the dictionary representation of the model using alias.

        This has the following differences from calling pydantic's
        `self.model_dump(by_alias=True)`:

        * `None` is only added to the output dict for nullable fields that
          were set at model initialization. Other fields with value `None`
          are ignored.
        """
        excluded_fields: Set[str] = set([])

        _dict = self.model_dump(
            by_alias=True,
            exclude=excluded_fields,
            exclude_none=True,
        )
        # override the default output from pydantic by calling `to_dict()` of credentials_rotation
        if self.credentials_rotation:
            _dict["credentialsRotation"] = self.credentials_rotation.to_dict()
        # override the default output from pydantic by calling `to_dict()` of error
        if self.error:
            _dict["error"] = self.error.to_dict()
        # override the default output from pydantic by calling `to_dict()` of each item in errors (list)
        _items = []
        if self.errors:
            for _item_errors in self.errors:
                if _item_errors:
                    _items.append(_item_errors.to_dict())
            _dict["errors"] = _items
        # override the default output from pydantic by calling `to_dict()` of expiration
        if self.expiration:
            _dict["expiration"] = self.expiration.to_dict()
        return _dict

    @classmethod
    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
        """Create an instance of ClusterStatus from a dict"""
        if obj is None:
            return None

        if not isinstance(obj, dict):
            return cls.model_validate(obj)

        _obj = cls.model_validate(
            {
                "aggregated": obj.get("aggregated"),
                "creationTime": obj.get("creationTime"),
                "credentialsRotation": (
                    CredentialsRotationState.from_dict(obj["credentialsRotation"])
                    if obj.get("credentialsRotation") is not None
                    else None
                ),
                "egressAddressRanges": obj.get("egressAddressRanges"),
                "error": RuntimeError.from_dict(obj["error"]) if obj.get("error") is not None else None,
                "errors": (
                    [ClusterError.from_dict(_item) for _item in obj["errors"]]
                    if obj.get("errors") is not None
                    else None
                ),
                "expiration": (
                    ExpiratoaionStatus.from_dict(obj["expiration"]) if obj.get("expiration") is not None else None
                ),
                "hibernated": obj.get("hibernated"),
                "identity": obj.get("identity"),
                "podAddressRanges": obj.get("podAddressRanges"),
            }
        )
        return _obj