# 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
from typing import Any, ClassVar, Dict, List, Optional, cast as _cast
from kubernetes.client.models.v1alpha3_workload_pod_group_disruption_mode import V1alpha3WorkloadPodGroupDisruptionMode
from kubernetes.client.models.v1alpha3_workload_pod_group_resource_claim import V1alpha3WorkloadPodGroupResourceClaim
from kubernetes.client.models.v1alpha3_workload_pod_group_scheduling_constraints import V1alpha3WorkloadPodGroupSchedulingConstraints
from kubernetes.client.models.v1alpha3_workload_pod_group_scheduling_policy import V1alpha3WorkloadPodGroupSchedulingPolicy
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 V1JobSchedulingConfiguration(BaseModel):
    """
    JobSchedulingConfiguration composes the reusable workload-aware scheduling building blocks.

    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[V1alpha3WorkloadPodGroupDisruptionMode] = Field(default=None, validation_alias=AliasChoices("disruptionMode", "disruption_mode"), serialization_alias="disruptionMode")
    resource_claims: Optional[List[V1alpha3WorkloadPodGroupResourceClaim]] = Field(default=None, validation_alias=AliasChoices("resourceClaims", "resource_claims"), serialization_alias="resourceClaims", description="ResourceClaims defines which ResourceClaims may be shared among Pods in the Job. 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. At most 4 claims may be set, matching the limit on the resulting PodGroup. This list is immutable after creation: entries may neither be added, removed, nor modified.")
    scheduling_constraints: Optional[V1alpha3WorkloadPodGroupSchedulingConstraints] = Field(default=None, validation_alias=AliasChoices("schedulingConstraints", "scheduling_constraints"), serialization_alias="schedulingConstraints")
    scheduling_policy: Optional[V1alpha3WorkloadPodGroupSchedulingPolicy] = Field(default=None, validation_alias=AliasChoices("schedulingPolicy", "scheduling_policy"), serialization_alias="schedulingPolicy")
    openapi_types: ClassVar[Dict[str, str]] = {
        "disruption_mode": "V1alpha3WorkloadPodGroupDisruptionMode",
        "resource_claims": "List[V1alpha3WorkloadPodGroupResourceClaim]",
        "scheduling_constraints": "V1alpha3WorkloadPodGroupSchedulingConstraints",
        "scheduling_policy": "V1alpha3WorkloadPodGroupSchedulingPolicy"
    }

    attribute_map: ClassVar[Dict[str, str]] = {
        "disruption_mode": "disruptionMode",
        "resource_claims": "resourceClaims",
        "scheduling_constraints": "schedulingConstraints",
        "scheduling_policy": "schedulingPolicy"
    }
    __properties: ClassVar[List[str]] = ["disruptionMode", "resourceClaims", "schedulingConstraints", "schedulingPolicy"]

    @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 "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)
        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, V1JobSchedulingConfiguration):
            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, V1JobSchedulingConfiguration):
            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 V1JobSchedulingConfiguration 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),
            ("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),
        }

    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)
        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 V1JobSchedulingConfiguration 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": V1alpha3WorkloadPodGroupDisruptionMode.from_dict(obj["disruptionMode"]) if obj.get("disruptionMode") is not None else None,
            "resourceClaims": [V1alpha3WorkloadPodGroupResourceClaim.from_dict(_item) for _item in obj["resourceClaims"]] if obj.get("resourceClaims") is not None else None,
            "schedulingConstraints": V1alpha3WorkloadPodGroupSchedulingConstraints.from_dict(obj["schedulingConstraints"]) if obj.get("schedulingConstraints") is not None else None,
            "schedulingPolicy": V1alpha3WorkloadPodGroupSchedulingPolicy.from_dict(obj["schedulingPolicy"]) if obj.get("schedulingPolicy") is not None else None
        })
        return _obj