Source code for sr.comp.match_period

"""Classes that are useful for dealing with match periods."""

from __future__ import annotations

import dataclasses
import datetime
from collections.abc import Mapping
from enum import Enum, unique
from typing import Literal, NewType

from .types import ArenaName, MatchNumber, TLA




[docs]
@dataclasses.dataclass(frozen=True, slots=True)
class Delay:
    delay: datetime.timedelta
    time: datetime.datetime






[docs]
@unique
class MatchType(Enum):
    league = 'league'
    knockout = 'knockout'
    tiebreaker = 'tiebreaker'






[docs]
@dataclasses.dataclass(frozen=True)
class Match:
    num: MatchNumber
    display_name: str
    arena: ArenaName
    teams: list[TLA | None]
    start_time: datetime.datetime
    end_time: datetime.datetime
    type: MatchType  # noqa:A003
    use_resolved_ranking: bool






[docs]
@dataclasses.dataclass(frozen=True)
class KnockoutMatch(Match):
    type: Literal[MatchType.knockout]  # noqa:A003
    knockout_bracket: str




MatchSlot = NewType('MatchSlot', Mapping[ArenaName, Match])




[docs]
@dataclasses.dataclass(frozen=True, slots=True)
class MatchPeriod:
    start_time: datetime.datetime
    end_time: datetime.datetime
    max_end_time: datetime.datetime
    description: str
    matches: list[MatchSlot]
    type: MatchType  # noqa:A003

    def __str__(self) -> str:
        return "{} ({}–{})".format(
            self.description,
            self.start_time.strftime('%H:%M'),
            self.end_time.strftime('%H:%M'),
        )