airflow_client/client/models/response_get_xcom_entry.py

# coding: utf-8

"""
    Airflow API

    Airflow API. All endpoints located under ``/api/v2`` can be used safely, are stable and backward compatible. Endpoints located under ``/ui`` are dedicated to the UI and are subject to breaking change depending on the need of the frontend. Users should not rely on those but use the public ones instead.

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

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


from __future__ import annotations
from inspect import getfullargspec
import json
import pprint
import re  # noqa: F401
from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
from typing import Optional
from airflow_client.client.models.x_com_response_native import XComResponseNative
from airflow_client.client.models.x_com_response_string import XComResponseString
from typing import Union, Any, List, Set, TYPE_CHECKING, Optional, Dict
from typing_extensions import Literal, Self
from pydantic import Field

RESPONSEGETXCOMENTRY_ANY_OF_SCHEMAS = ["XComResponseNative", "XComResponseString"]

class ResponseGetXcomEntry(BaseModel):
    """
    ResponseGetXcomEntry
    """

    # data type: XComResponseNative
    anyof_schema_1_validator: Optional[XComResponseNative] = None
    # data type: XComResponseString
    anyof_schema_2_validator: Optional[XComResponseString] = None
    if TYPE_CHECKING:
        actual_instance: Optional[Union[XComResponseNative, XComResponseString]] = None
    else:
        actual_instance: Any = None
    any_of_schemas: Set[str] = { "XComResponseNative", "XComResponseString" }

    model_config = {
        "validate_assignment": True,
        "protected_namespaces": (),
    }

    def __init__(self, *args, **kwargs) -> None:
        if args:
            if len(args) > 1:
                raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`")
            if kwargs:
                raise ValueError("If a position argument is used, keyword arguments cannot be used.")
            super().__init__(actual_instance=args[0])
        else:
            super().__init__(**kwargs)

    @field_validator('actual_instance')
    def actual_instance_must_validate_anyof(cls, v):
        instance = ResponseGetXcomEntry.model_construct()
        error_messages = []
        # validate data type: XComResponseNative
        if not isinstance(v, XComResponseNative):
            error_messages.append(f"Error! Input type `{type(v)}` is not `XComResponseNative`")
        else:
            return v

        # validate data type: XComResponseString
        if not isinstance(v, XComResponseString):
            error_messages.append(f"Error! Input type `{type(v)}` is not `XComResponseString`")
        else:
            return v

        if error_messages:
            # no match
            raise ValueError("No match found when setting the actual_instance in ResponseGetXcomEntry with anyOf schemas: XComResponseNative, XComResponseString. Details: " + ", ".join(error_messages))
        else:
            return v

    @classmethod
    def from_dict(cls, obj: Dict[str, Any]) -> Self:
        return cls.from_json(json.dumps(obj))

    @classmethod
    def from_json(cls, json_str: str) -> Self:
        """Returns the object represented by the json string"""
        instance = cls.model_construct()
        error_messages = []
        # anyof_schema_1_validator: Optional[XComResponseNative] = None
        try:
            instance.actual_instance = XComResponseNative.from_json(json_str)
            return instance
        except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
        # anyof_schema_2_validator: Optional[XComResponseString] = None
        try:
            instance.actual_instance = XComResponseString.from_json(json_str)
            return instance
        except (ValidationError, ValueError) as e:
             error_messages.append(str(e))

        if error_messages:
            # no match
            raise ValueError("No match found when deserializing the JSON string into ResponseGetXcomEntry with anyOf schemas: XComResponseNative, XComResponseString. Details: " + ", ".join(error_messages))
        else:
            return instance

    def to_json(self) -> str:
        """Returns the JSON representation of the actual instance"""
        if self.actual_instance is None:
            return "null"

        if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json):
            return self.actual_instance.to_json()
        else:
            return json.dumps(self.actual_instance)

    def to_dict(self) -> Optional[Union[Dict[str, Any], XComResponseNative, XComResponseString]]:
        """Returns the dict representation of the actual instance"""
        if self.actual_instance is None:
            return None

        if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict):
            return self.actual_instance.to_dict()
        else:
            return self.actual_instance

    def to_str(self) -> str:
        """Returns the string representation of the actual instance"""
        return pprint.pformat(self.model_dump())