# coding: utf-8

"""
    node-server

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

    The version of the OpenAPI document: 0.0.237
    Generated by: https://openapi-generator.tech
"""

from datetime import date, datetime  # noqa: F401
import decimal  # noqa: F401
import functools  # noqa: F401
import io  # noqa: F401
import re  # noqa: F401
import typing  # noqa: F401
import typing_extensions  # noqa: F401
import uuid  # noqa: F401

import frozendict  # noqa: F401

from tensorleap_openapi_client import schemas  # noqa: F401


class HeatmapChartsParams(
    schemas.DictSchema
):
    """NOTE: This class is auto generated by OpenAPI Generator.
    Ref: https://openapi-generator.tech

    Do not edit the class manually.
    """


    class MetaOapg:
        required = {
            "yAxisSizeInterval",
            "yDataDistributionType",
            "xField",
            "yField",
            "xDataDistributionType",
            "xAxisSizeInterval",
            "colorAggregationMethod",
            "modelExtIds",
            "colorField",
        }
        
        class properties:
            xField = schemas.StrSchema
            xAxisSizeInterval = schemas.Float64Schema
        
            @staticmethod
            def xDataDistributionType() -> typing.Type['DataDistributionType']:
                return DataDistributionType
            yField = schemas.StrSchema
            yAxisSizeInterval = schemas.Float64Schema
        
            @staticmethod
            def yDataDistributionType() -> typing.Type['DataDistributionType']:
                return DataDistributionType
            colorField = schemas.StrSchema
        
            @staticmethod
            def colorAggregationMethod() -> typing.Type['AggregationMethod']:
                return AggregationMethod
            
            
            class modelExtIds(
                schemas.ListSchema
            ):
            
            
                class MetaOapg:
                    items = schemas.StrSchema
            
                def __new__(
                    cls,
                    arg: typing.Union[typing.Tuple[typing.Union[MetaOapg.items, str, ]], typing.List[typing.Union[MetaOapg.items, str, ]]],
                    _configuration: typing.Optional[schemas.Configuration] = None,
                ) -> 'modelExtIds':
                    return super().__new__(
                        cls,
                        arg,
                        _configuration=_configuration,
                    )
            
                def __getitem__(self, i: int) -> MetaOapg.items:
                    return super().__getitem__(i)
        
            @staticmethod
            def xOrderParams() -> typing.Type['OrderType']:
                return OrderType
        
            @staticmethod
            def yOrderParams() -> typing.Type['OrderType']:
                return OrderType
            verticalSplit = schemas.StrSchema
            horizontalSplit = schemas.StrSchema
            
            
            class filters(
                schemas.ListSchema
            ):
            
            
                class MetaOapg:
                    
                    @staticmethod
                    def items() -> typing.Type['ESFilter']:
                        return ESFilter
            
                def __new__(
                    cls,
                    arg: typing.Union[typing.Tuple['ESFilter'], typing.List['ESFilter']],
                    _configuration: typing.Optional[schemas.Configuration] = None,
                ) -> 'filters':
                    return super().__new__(
                        cls,
                        arg,
                        _configuration=_configuration,
                    )
            
                def __getitem__(self, i: int) -> 'ESFilter':
                    return super().__getitem__(i)
            lastEpochOnly = schemas.BoolSchema
            __annotations__ = {
                "xField": xField,
                "xAxisSizeInterval": xAxisSizeInterval,
                "xDataDistributionType": xDataDistributionType,
                "yField": yField,
                "yAxisSizeInterval": yAxisSizeInterval,
                "yDataDistributionType": yDataDistributionType,
                "colorField": colorField,
                "colorAggregationMethod": colorAggregationMethod,
                "modelExtIds": modelExtIds,
                "xOrderParams": xOrderParams,
                "yOrderParams": yOrderParams,
                "verticalSplit": verticalSplit,
                "horizontalSplit": horizontalSplit,
                "filters": filters,
                "lastEpochOnly": lastEpochOnly,
            }
        additional_properties = schemas.NotAnyTypeSchema
    
    yAxisSizeInterval: MetaOapg.properties.yAxisSizeInterval
    yDataDistributionType: 'DataDistributionType'
    xField: MetaOapg.properties.xField
    yField: MetaOapg.properties.yField
    xDataDistributionType: 'DataDistributionType'
    xAxisSizeInterval: MetaOapg.properties.xAxisSizeInterval
    colorAggregationMethod: 'AggregationMethod'
    modelExtIds: MetaOapg.properties.modelExtIds
    colorField: MetaOapg.properties.colorField
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["yAxisSizeInterval"]) -> MetaOapg.properties.yAxisSizeInterval: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["yDataDistributionType"]) -> 'DataDistributionType': ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["xField"]) -> MetaOapg.properties.xField: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["yField"]) -> MetaOapg.properties.yField: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["xDataDistributionType"]) -> 'DataDistributionType': ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["xAxisSizeInterval"]) -> MetaOapg.properties.xAxisSizeInterval: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["colorAggregationMethod"]) -> 'AggregationMethod': ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["modelExtIds"]) -> MetaOapg.properties.modelExtIds: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["colorField"]) -> MetaOapg.properties.colorField: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["xOrderParams"]) -> 'OrderType': ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["yOrderParams"]) -> 'OrderType': ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["verticalSplit"]) -> MetaOapg.properties.verticalSplit: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["horizontalSplit"]) -> MetaOapg.properties.horizontalSplit: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["filters"]) -> MetaOapg.properties.filters: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["lastEpochOnly"]) -> MetaOapg.properties.lastEpochOnly: ...
    
    def __getitem__(self, name: typing.Union[typing_extensions.Literal["yAxisSizeInterval"], typing_extensions.Literal["yDataDistributionType"], typing_extensions.Literal["xField"], typing_extensions.Literal["yField"], typing_extensions.Literal["xDataDistributionType"], typing_extensions.Literal["xAxisSizeInterval"], typing_extensions.Literal["colorAggregationMethod"], typing_extensions.Literal["modelExtIds"], typing_extensions.Literal["colorField"], typing_extensions.Literal["xOrderParams"], typing_extensions.Literal["yOrderParams"], typing_extensions.Literal["verticalSplit"], typing_extensions.Literal["horizontalSplit"], typing_extensions.Literal["filters"], typing_extensions.Literal["lastEpochOnly"], ]):
        # dict_instance[name] accessor
        return super().__getitem__(name)
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["yAxisSizeInterval"]) -> MetaOapg.properties.yAxisSizeInterval: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["yDataDistributionType"]) -> 'DataDistributionType': ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["xField"]) -> MetaOapg.properties.xField: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["yField"]) -> MetaOapg.properties.yField: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["xDataDistributionType"]) -> 'DataDistributionType': ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["xAxisSizeInterval"]) -> MetaOapg.properties.xAxisSizeInterval: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["colorAggregationMethod"]) -> 'AggregationMethod': ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["modelExtIds"]) -> MetaOapg.properties.modelExtIds: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["colorField"]) -> MetaOapg.properties.colorField: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["xOrderParams"]) -> typing.Union['OrderType', schemas.Unset]: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["yOrderParams"]) -> typing.Union['OrderType', schemas.Unset]: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["verticalSplit"]) -> typing.Union[MetaOapg.properties.verticalSplit, schemas.Unset]: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["horizontalSplit"]) -> typing.Union[MetaOapg.properties.horizontalSplit, schemas.Unset]: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["filters"]) -> typing.Union[MetaOapg.properties.filters, schemas.Unset]: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["lastEpochOnly"]) -> typing.Union[MetaOapg.properties.lastEpochOnly, schemas.Unset]: ...
    
    def get_item_oapg(self, name: typing.Union[typing_extensions.Literal["yAxisSizeInterval"], typing_extensions.Literal["yDataDistributionType"], typing_extensions.Literal["xField"], typing_extensions.Literal["yField"], typing_extensions.Literal["xDataDistributionType"], typing_extensions.Literal["xAxisSizeInterval"], typing_extensions.Literal["colorAggregationMethod"], typing_extensions.Literal["modelExtIds"], typing_extensions.Literal["colorField"], typing_extensions.Literal["xOrderParams"], typing_extensions.Literal["yOrderParams"], typing_extensions.Literal["verticalSplit"], typing_extensions.Literal["horizontalSplit"], typing_extensions.Literal["filters"], typing_extensions.Literal["lastEpochOnly"], ]):
        return super().get_item_oapg(name)

    def __new__(
        cls,
        *args: typing.Union[dict, frozendict.frozendict, ],
        yAxisSizeInterval: typing.Union[MetaOapg.properties.yAxisSizeInterval, decimal.Decimal, int, float, ],
        yDataDistributionType: 'DataDistributionType',
        xField: typing.Union[MetaOapg.properties.xField, str, ],
        yField: typing.Union[MetaOapg.properties.yField, str, ],
        xDataDistributionType: 'DataDistributionType',
        xAxisSizeInterval: typing.Union[MetaOapg.properties.xAxisSizeInterval, decimal.Decimal, int, float, ],
        colorAggregationMethod: 'AggregationMethod',
        modelExtIds: typing.Union[MetaOapg.properties.modelExtIds, list, tuple, ],
        colorField: typing.Union[MetaOapg.properties.colorField, str, ],
        xOrderParams: typing.Union['OrderType', schemas.Unset] = schemas.unset,
        yOrderParams: typing.Union['OrderType', schemas.Unset] = schemas.unset,
        verticalSplit: typing.Union[MetaOapg.properties.verticalSplit, str, schemas.Unset] = schemas.unset,
        horizontalSplit: typing.Union[MetaOapg.properties.horizontalSplit, str, schemas.Unset] = schemas.unset,
        filters: typing.Union[MetaOapg.properties.filters, list, tuple, schemas.Unset] = schemas.unset,
        lastEpochOnly: typing.Union[MetaOapg.properties.lastEpochOnly, bool, schemas.Unset] = schemas.unset,
        _configuration: typing.Optional[schemas.Configuration] = None,
    ) -> 'HeatmapChartsParams':
        return super().__new__(
            cls,
            *args,
            yAxisSizeInterval=yAxisSizeInterval,
            yDataDistributionType=yDataDistributionType,
            xField=xField,
            yField=yField,
            xDataDistributionType=xDataDistributionType,
            xAxisSizeInterval=xAxisSizeInterval,
            colorAggregationMethod=colorAggregationMethod,
            modelExtIds=modelExtIds,
            colorField=colorField,
            xOrderParams=xOrderParams,
            yOrderParams=yOrderParams,
            verticalSplit=verticalSplit,
            horizontalSplit=horizontalSplit,
            filters=filters,
            lastEpochOnly=lastEpochOnly,
            _configuration=_configuration,
        )

from tensorleap_openapi_client.model.aggregation_method import AggregationMethod
from tensorleap_openapi_client.model.data_distribution_type import DataDistributionType
from tensorleap_openapi_client.model.es_filter import ESFilter
from tensorleap_openapi_client.model.order_type import OrderType
