# coding: utf-8

"""
    Kubernetes

    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)

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

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


from __future__ import annotations
import pprint
import re  # noqa: F401
import json

from collections.abc import Mapping as _Mapping
from pydantic import AliasChoices, BaseModel, ConfigDict, Field, StrictInt, StrictStr
from typing import Any, ClassVar, Dict, List, Optional, cast as _cast
from kubernetes.aio.client.models.v1alpha3_disruption_mode import V1alpha3DisruptionMode
from kubernetes.aio.client.models.v1alpha3_pod_group_resource_claim import V1alpha3PodGroupResourceClaim
from kubernetes.aio.client.models.v1alpha3_pod_group_scheduling_constraints import V1alpha3PodGroupSchedulingConstraints
from kubernetes.aio.client.models.v1alpha3_pod_group_scheduling_policy import V1alpha3PodGroupSchedulingPolicy
from kubernetes.aio.client.models.v1alpha3_workload_reference import V1alpha3WorkloadReference
from typing import Optional, Set
from typing_extensions import Self
from pydantic_core import to_jsonable_python


_OPENAPI_GENERATOR_TO_DICT = "_openapi_generator_to_dict"


def _get_openapi_to_dict(value: Any) -> Any:
    # Reciprocal function references preserve inherited generated methods
    # without reserving model member names. Checking both directions also
    # rejects overrides whose decorators copy function attributes.
    to_dict = getattr(value, "to_dict", None)
    type_to_dict = getattr(type(value), "to_dict", None)
    if (
        not callable(to_dict)
        or getattr(to_dict, "__func__", None) is not type_to_dict
    ):
        return None

    openapi_to_dict = getattr(
        type_to_dict, _OPENAPI_GENERATOR_TO_DICT, None
    )
    if (
        not callable(openapi_to_dict)
        or getattr(
            openapi_to_dict,
            _OPENAPI_GENERATOR_TO_DICT,
            None,
        ) is not type_to_dict
    ):
        return None
    return openapi_to_dict


def _to_legacy_item(value: Any, serialize: bool) -> Any:
    to_dict = getattr(value, "to_dict", None)
    if _get_openapi_to_dict(value) is not None and callable(to_dict):
        return to_dict(serialize=serialize)
    if callable(to_dict):
        return to_dict()
    return value


def _to_legacy_value(value: Any, serialize: bool) -> Any:
    # python-legacy converted only immediate list elements or dictionary values:
    # https://github.com/OpenAPITools/openapi-generator/blob/c84b949df1a9ec04ba75989cb49b45c940c2f694/modules/openapi-generator/src/main/resources/python-legacy/model.mustache#L203-L231
    if isinstance(value, list):
        return [_to_legacy_item(item, serialize) for item in value]
    if isinstance(value, dict):
        return {
            key: _to_legacy_item(item, serialize)
            for key, item in value.items()
        }
    return _to_legacy_item(value, serialize)


def _to_openapi_value(value: Any) -> Any:
    if isinstance(value, list):
        return [_to_openapi_value(item) for item in value]
    if isinstance(value, dict):
        return {key: _to_openapi_value(item) for key, item in value.items()}

    to_openapi_dict = _get_openapi_to_dict(value)
    if to_openapi_dict is not None:
        return to_openapi_dict(value)

    to_dict = getattr(value, "to_dict", None)
    if callable(to_dict):
        return to_dict()
    return value


class V1alpha3PodGroupSpec(BaseModel):
    """
    PodGroupSpec defines the desired state of a PodGroup.

    Attributes:
      openapi_types (dict): The key is attribute name
                            and the value is attribute type.
      attribute_map (dict): The key is attribute name
                            and the value is json key in definition.
    """ # noqa: E501
    disruption_mode: Optional[V1alpha3DisruptionMode] = Field(default=None, validation_alias=AliasChoices("disruptionMode", "disruption_mode"), serialization_alias="disruptionMode")
    parent_composite_pod_group_name: Optional[StrictStr] = Field(default=None, validation_alias=AliasChoices("parentCompositePodGroupName", "parent_composite_pod_group_name"), serialization_alias="parentCompositePodGroupName", description="parentCompositePodGroupName contains the name of the parent composite pod group within the same namespace as this pod group. If it's nil, then this pod group is a root of a workload's hierarchy. This field is used only when the CompositePodGroup feature gate is enabled. This field is immutable.")
    preemption_policy: Optional[StrictStr] = Field(default=None, validation_alias=AliasChoices("preemptionPolicy", "preemption_policy"), serialization_alias="preemptionPolicy", description="preemptionPolicy is the Policy for preempting pods/podgroups with lower priority. One of Never, PreemptLowerPriority. Defaults to PreemptLowerPriority if unset. When Priority Admission Controller is enabled, it populates this field from PriorityClassName, and defaults to PreemptLowerPriority if value is unset in PriorityClass. This field is immutable. This field is available only when the PodGroupPreemptionPolicy feature gate is enabled.")
    priority: Optional[StrictInt] = Field(default=None, description="priority is the value of priority of this pod group. Various system components use this field to find the priority of the pod group. When Priority Admission Controller is enabled, it prevents users from setting this field. The admission controller populates this field from PriorityClassName. The higher the value, the higher the priority. This field is immutable.")
    priority_class_name: Optional[StrictStr] = Field(default=None, validation_alias=AliasChoices("priorityClassName", "priority_class_name"), serialization_alias="priorityClassName", description="priorityClassName defines the priority that should be considered when scheduling this pod group. Controllers are expected to fill this field by copying it from a PodGroupTemplate. Otherwise, it is validated and resolved similarly to the PriorityClassName on PodGroupTemplate (i.e. if no priority class is specified, admission control can set this to the global default priority class if it exists. Otherwise, the pod group's priority will be zero). This field is immutable.")
    resource_claims: Optional[List[V1alpha3PodGroupResourceClaim]] = Field(default=None, validation_alias=AliasChoices("resourceClaims", "resource_claims"), serialization_alias="resourceClaims", description="resourceClaims defines which ResourceClaims may be shared among Pods in the group. Pods consume the devices allocated to a PodGroup's claim by defining a claim in its own Spec.ResourceClaims that matches the PodGroup's claim exactly. The claim must have the same name and refer to the same ResourceClaim or ResourceClaimTemplate.  This is a beta-level field and requires that the DRAWorkloadResourceClaims feature gate is enabled.  This field is immutable.")
    scheduling_constraints: Optional[V1alpha3PodGroupSchedulingConstraints] = Field(default=None, validation_alias=AliasChoices("schedulingConstraints", "scheduling_constraints"), serialization_alias="schedulingConstraints")
    scheduling_policy: V1alpha3PodGroupSchedulingPolicy = Field(validation_alias=AliasChoices("schedulingPolicy", "scheduling_policy"), serialization_alias="schedulingPolicy")
    workload_ref: Optional[V1alpha3WorkloadReference] = Field(default=None, validation_alias=AliasChoices("workloadRef", "workload_ref"), serialization_alias="workloadRef")
    openapi_types: ClassVar[Dict[str, str]] = {
        "disruption_mode": "V1alpha3DisruptionMode",
        "parent_composite_pod_group_name": "str",
        "preemption_policy": "str",
        "priority": "int",
        "priority_class_name": "str",
        "resource_claims": "List[V1alpha3PodGroupResourceClaim]",
        "scheduling_constraints": "V1alpha3PodGroupSchedulingConstraints",
        "scheduling_policy": "V1alpha3PodGroupSchedulingPolicy",
        "workload_ref": "V1alpha3WorkloadReference"
    }

    attribute_map: ClassVar[Dict[str, str]] = {
        "disruption_mode": "disruptionMode",
        "parent_composite_pod_group_name": "parentCompositePodGroupName",
        "preemption_policy": "preemptionPolicy",
        "priority": "priority",
        "priority_class_name": "priorityClassName",
        "resource_claims": "resourceClaims",
        "scheduling_constraints": "schedulingConstraints",
        "scheduling_policy": "schedulingPolicy",
        "workload_ref": "workloadRef"
    }
    __properties: ClassVar[List[str]] = ["disruptionMode", "parentCompositePodGroupName", "preemptionPolicy", "priority", "priorityClassName", "resourceClaims", "schedulingConstraints", "schedulingPolicy", "workloadRef"]

    @classmethod
    def __preprocess_input_names(
        cls,
        obj: Any,
        remove_hidden_storage_names: bool = True,
    ) -> Any:
        if not isinstance(obj, _Mapping):
            return obj
        obj = dict(obj)
        if "disruptionMode" not in obj and "disruption_mode" in obj:
            obj["disruptionMode"] = obj["disruption_mode"]
        obj.pop("disruption_mode", None)
        if "parentCompositePodGroupName" not in obj and "parent_composite_pod_group_name" in obj:
            obj["parentCompositePodGroupName"] = obj["parent_composite_pod_group_name"]
        obj.pop("parent_composite_pod_group_name", None)
        if "preemptionPolicy" not in obj and "preemption_policy" in obj:
            obj["preemptionPolicy"] = obj["preemption_policy"]
        obj.pop("preemption_policy", None)
        if "priorityClassName" not in obj and "priority_class_name" in obj:
            obj["priorityClassName"] = obj["priority_class_name"]
        obj.pop("priority_class_name", None)
        if "resourceClaims" not in obj and "resource_claims" in obj:
            obj["resourceClaims"] = obj["resource_claims"]
        obj.pop("resource_claims", None)
        if "schedulingConstraints" not in obj and "scheduling_constraints" in obj:
            obj["schedulingConstraints"] = obj["scheduling_constraints"]
        obj.pop("scheduling_constraints", None)
        if "schedulingPolicy" not in obj and "scheduling_policy" in obj:
            obj["schedulingPolicy"] = obj["scheduling_policy"]
        obj.pop("scheduling_policy", None)
        if "workloadRef" not in obj and "workload_ref" in obj:
            obj["workloadRef"] = obj["workload_ref"]
        obj.pop("workload_ref", None)
        return obj

    model_config = ConfigDict(
        validate_by_name=True,
        validate_by_alias=True,
        validate_assignment=True,
        extra="forbid",
        protected_namespaces=(),
    )


    def to_str(self) -> str:
        """Returns the string representation of the model"""
        return pprint.pformat(self.to_dict())

    def __repr__(self) -> str:
        """For print and pprint"""
        return self.to_str()

    def __eq__(self, other: object) -> bool:
        """Returns true if both objects are equal"""
        if not isinstance(other, V1alpha3PodGroupSpec):
            return False

        return self.to_dict() == other.to_dict()

    def __ne__(self, other: object) -> bool:
        """Returns true if both objects are not equal"""
        if not isinstance(other, V1alpha3PodGroupSpec):
            return True

        return not (self == other)


    def to_json(self) -> str:
        """Returns the JSON representation of the model using alias"""
        to_openapi_dict = _get_openapi_to_dict(self)
        if to_openapi_dict is not None:
            return json.dumps(to_jsonable_python(to_openapi_dict(self)))
        return json.dumps(to_jsonable_python(self.to_dict()))

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

    def to_dict(self, serialize: bool = False) -> Dict[str, Any]:
        """Return all declared model fields using public or wire names."""
        return {
            ("disruptionMode" if serialize else "disruption_mode"): _to_legacy_value(getattr(self, "disruption_mode", None), serialize),
            ("parentCompositePodGroupName" if serialize else "parent_composite_pod_group_name"): _to_legacy_value(getattr(self, "parent_composite_pod_group_name", None), serialize),
            ("preemptionPolicy" if serialize else "preemption_policy"): _to_legacy_value(getattr(self, "preemption_policy", None), serialize),
            ("priority" if serialize else "priority"): _to_legacy_value(getattr(self, "priority", None), serialize),
            ("priorityClassName" if serialize else "priority_class_name"): _to_legacy_value(getattr(self, "priority_class_name", None), serialize),
            ("resourceClaims" if serialize else "resource_claims"): _to_legacy_value(getattr(self, "resource_claims", None), serialize),
            ("schedulingConstraints" if serialize else "scheduling_constraints"): _to_legacy_value(getattr(self, "scheduling_constraints", None), serialize),
            ("schedulingPolicy" if serialize else "scheduling_policy"): _to_legacy_value(getattr(self, "scheduling_policy", None), serialize),
            ("workloadRef" if serialize else "workload_ref"): _to_legacy_value(getattr(self, "workload_ref", None), serialize),
        }

    def __openapi_generator_modern_projection(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 disruption_mode
        if self.disruption_mode:
            _dict['disruptionMode'] = _to_openapi_value(self.disruption_mode)
        # override the default output from pydantic by calling `to_dict()` of each item in resource_claims (list)
        _items = []
        if self.resource_claims:
            for _item_resource_claims in self.resource_claims:
                _items.append(_to_openapi_value(_item_resource_claims) if _item_resource_claims is not None else None)
            _dict['resourceClaims'] = _items
        # override the default output from pydantic by calling `to_dict()` of scheduling_constraints
        if self.scheduling_constraints:
            _dict['schedulingConstraints'] = _to_openapi_value(self.scheduling_constraints)
        # override the default output from pydantic by calling `to_dict()` of scheduling_policy
        if self.scheduling_policy:
            _dict['schedulingPolicy'] = _to_openapi_value(self.scheduling_policy)
        # override the default output from pydantic by calling `to_dict()` of workload_ref
        if self.workload_ref:
            _dict['workloadRef'] = _to_openapi_value(self.workload_ref)
        return _dict

    setattr(
        to_dict,
        _OPENAPI_GENERATOR_TO_DICT,
        __openapi_generator_modern_projection,
    )
    setattr(
        __openapi_generator_modern_projection,
        _OPENAPI_GENERATOR_TO_DICT,
        to_dict,
    )
    del __openapi_generator_modern_projection

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

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

        obj = _cast(
            Dict[str, Any],
            cls.__preprocess_input_names(
                obj,
                remove_hidden_storage_names=True,
            ),
        )

        _obj = cls.model_validate({
            "disruptionMode": V1alpha3DisruptionMode.from_dict(obj["disruptionMode"]) if obj.get("disruptionMode") is not None else None,
            "parentCompositePodGroupName": obj.get("parentCompositePodGroupName"),
            "preemptionPolicy": obj.get("preemptionPolicy"),
            "priority": obj.get("priority"),
            "priorityClassName": obj.get("priorityClassName"),
            "resourceClaims": [V1alpha3PodGroupResourceClaim.from_dict(_item) for _item in obj["resourceClaims"]] if obj.get("resourceClaims") is not None else None,
            "schedulingConstraints": V1alpha3PodGroupSchedulingConstraints.from_dict(obj["schedulingConstraints"]) if obj.get("schedulingConstraints") is not None else None,
            "schedulingPolicy": V1alpha3PodGroupSchedulingPolicy.from_dict(obj["schedulingPolicy"]) if obj.get("schedulingPolicy") is not None else None,
            "workloadRef": V1alpha3WorkloadReference.from_dict(obj["workloadRef"]) if obj.get("workloadRef") is not None else None
        })
        return _obj