from dataclasses import dataclass, field
from typing import List, Optional

from xsdata.models.datatype import XmlDate

from ..defaults import DEFAULT_TIME_ZONE
from ..validations import validate_list
from .payload_message import PayloadMessage, PayloadMessageResponse


@dataclass(kw_only=True)
class DsoPortfolioQueryConnection:
    """
    A Connection that is part of the congestion point.

    :ivar entity_address: EntityAddress of the Connection.
    :ivar agr_domain: The internet domain of the AGR that represents the
        prosumer connected on this Connection, if applicable.
    """
    class Meta:
        name = "DSOPortfolioQueryConnection"

    entity_address: str = field(
        metadata={
            "name": "EntityAddress",
            "type": "Attribute",
            "required": True,
            "pattern": r"(ea1\.[0-9]{4}-[0-9]{2}\..{1,244}:.{1,244}|ean\.[0-9]{12,34})",
        }
    )
    agr_domain: Optional[str] = field(
        default=None,
        metadata={
            "name": "AGR-Domain",
            "type": "Attribute",
            "pattern": r"([a-z0-9]+(-[a-z0-9]+)*\.)+[a-z]{2,}",
        }
    )


@dataclass(kw_only=True)
class DsoPortfolioQueryCongestionPoint:
    """
    :ivar connection:
    :ivar entity_address: EntityAddress of the Connection.
    """
    class Meta:
        name = "DSOPortfolioQueryCongestionPoint"

    connections: List[DsoPortfolioQueryConnection] = field(
        default_factory=list,
        metadata={
            "name": "Connection",
            "type": "Element",
            "min_occurs": 1,
        }
    )
    entity_address: str = field(
        metadata={
            "name": "EntityAddress",
            "type": "Attribute",
            "required": True,
            "pattern": r"(ea1\.[0-9]{4}-[0-9]{2}\..{1,244}:.{1,244}|ean\.[0-9]{12,34})",
        }
    )

    def __post_init__(self):
        validate_list('connections', self.connections, DsoPortfolioQueryConnection, 1)


@dataclass(kw_only=True)
class DsoPortfolioQueryResponse(PayloadMessageResponse):
    """
    :ivar congestion_point:
    :ivar time_zone: Time zone ID (as per the IANA time zone database,
        http://www.iana.org/time-zones, for example: Europe/Amsterdam)
        indicating the UTC offset that applies to the Period referenced
        in this message. Although the time zone is a market-wide fixed
        value, making this assumption explicit in each message is
        important for validation purposes, allowing implementations to
        reject messages with an errant UTC offset.
    :ivar period: The Period for which the AGR requests the portfolio
        information.
    """
    class Meta:
        name = "DSOPortfolioQueryResponse"

    dso_portfolio_query_message_id: str = field(
        metadata={
            "name": "DSOPortfolioQueryMessageID",
            "type": "Attribute",
            "required": True,
            "pattern": r"[0-9A-Fa-f]{8}-[0-9A-Fa-f]{4}-[0-9A-Fa-f]{4}-[0-9A-Fa-f]{4}-[0-9A-Fa-f]{12}",
        }
    )

    congestion_point: Optional[DsoPortfolioQueryCongestionPoint] = field(
        default=None,
        metadata={
            "name": "CongestionPoint",
            "type": "Element",
        }
    )
    time_zone: str = field(
        default=DEFAULT_TIME_ZONE,
        metadata={
            "name": "TimeZone",
            "type": "Attribute",
            "required": True,
            "pattern": r"(Africa|America|Australia|Europe|Pacific)/[a-zA-Z0-9_/]{3,}",
        }
    )
    period: XmlDate = field(
        metadata={
            "name": "Period",
            "type": "Attribute",
            "format": "%Y-%m-%d",
            "required": True,
        }
    )


@dataclass(kw_only=True)
class DsoPortfolioQuery(PayloadMessage):
    """
    :ivar time_zone: Time zone ID (as per the IANA time zone database,
        http://www.iana.org/time-zones, for example: Europe/Amsterdam)
        indicating the UTC offset that applies to the Period referenced
        in this message. Although the time zone is a market-wide fixed
        value, making this assumption explicit in each message is
        important for validation purposes, allowing implementations to
        reject messages with an errant UTC offset.
    :ivar period: The Period for which the AGR requests the portfolio
        information.
    :ivar entity_address: EntityAddress of the CongestionPoint
    """
    class Meta:
        name = "DSOPortfolioQuery"

    time_zone: str = field(
        default=DEFAULT_TIME_ZONE,
        metadata={
            "name": "TimeZone",
            "type": "Attribute",
            "required": True,
            "pattern": r"(Africa|America|Australia|Europe|Pacific)/[a-zA-Z0-9_/]{3,}",
        }
    )
    period: XmlDate = field(
        metadata={
            "name": "Period",
            "type": "Attribute",
            "format": "%Y-%m-%d",
            "required": True,
        }
    )
    entity_address: str = field(
        metadata={
            "name": "EntityAddress",
            "type": "Attribute",
            "required": True,
            "pattern": r"(ea1\.[0-9]{4}-[0-9]{2}\..{1,244}:.{1,244}|ean\.[0-9]{12,34})",
        }
    )
