agenda/agenda/types.py

"""Types."""

import datetime
import typing
from dataclasses import dataclass, field

from pycountry.db import Country

import agenda
from agenda import format_list_with_ampersand

StrDict = dict[str, typing.Any]
DateOrDateTime = datetime.datetime | datetime.date


def as_date(d: DateOrDateTime) -> datetime.date:
    """Convert datetime to date."""
    if isinstance(d, datetime.datetime):
        return d.date()
    assert isinstance(d, datetime.date)
    return d


@dataclass
class Trip:
    """Trip."""

    start: datetime.date
    travel: list[StrDict] = field(default_factory=list)
    accommodation: list[StrDict] = field(default_factory=list)
    conferences: list[StrDict] = field(default_factory=list)
    events: list[StrDict] = field(default_factory=list)

    @property
    def title(self) -> str:
        """Trip title."""
        return (
            format_list_with_ampersand(
                [conf["name"] for conf in self.conferences]
                + [event["title"] for event in self.events]
            )
            or "[unnamed trip]"
        )

    @property
    def end(self) -> datetime.date | None:
        """End date for trip."""
        max_conference_end = (
            max(as_date(item["end"]) for item in self.conferences)
            if self.conferences
            else datetime.date.min
        )
        assert isinstance(max_conference_end, datetime.date)

        arrive = [item["arrive"].date() for item in self.travel if "arrive" in item]
        travel_end = max(arrive) if arrive else datetime.date.min
        assert isinstance(travel_end, datetime.date)

        max_date = max(max_conference_end, travel_end)
        return max_date if max_date != datetime.date.min else None

    @property
    def countries(self) -> list[Country]:
        """Countries visited as part of trip, in order."""
        seen: set[str] = set()
        items: list[Country] = []
        for item in self.conferences + self.accommodation + self.events:
            if "country" not in item:
                continue
            if item["country"] in seen:
                continue
            seen.add(item["country"])
            country = agenda.get_country(item["country"])
            assert country
            items.append(country)

        return items

    @property
    def countries_str(self) -> str:
        """List of countries visited on this trip."""
        return format_list_with_ampersand(
            [f"{c.flag} {c.name}" for c in self.countries]
        )


@dataclass
class Holiday:
    """Holiay."""

    name: str
    country: str
    date: datetime.date


emojis = {
    "market": "🧺",
    "us_presidential_election": "🗳️🇺🇸",
    "bus_route_closure": "🚌❌",
    "meetup": "👥",
    "dinner": "🍷",
    "party": "🍷",
    "ba_voucher": "✈️",
    "accommodation": "🏨",  # alternative: 🧳
    "flight": "✈️",
    "conference": "🎤",
    "rocket": "🚀",
    "birthday": "🎈",
    "waste_schedule": "🗑️",
    "economist": "📰",
    "running": "🏃",
    "critical_mass": "🚴",
}


@dataclass
class Event:
    """Event."""

    name: str
    date: DateOrDateTime
    end_date: DateOrDateTime | None = None
    title: str | None = None
    url: str | None = None
    going: bool | None = None

    @property
    def as_datetime(self) -> datetime.datetime:
        """Date/time of event."""
        d = self.date
        t0 = datetime.datetime.min.time()
        return (
            d
            if isinstance(d, datetime.datetime)
            else datetime.datetime.combine(d, t0).replace(tzinfo=datetime.timezone.utc)
        )

    @property
    def has_time(self) -> bool:
        """Event has a time associated with it."""
        return isinstance(self.date, datetime.datetime)

    @property
    def as_date(self) -> datetime.date:
        """Date of event."""
        return (
            self.date.date() if isinstance(self.date, datetime.datetime) else self.date
        )

    @property
    def end_as_date(self) -> datetime.date:
        """Date of event."""
        return (
            (
                self.end_date.date()
                if isinstance(self.end_date, datetime.datetime)
                else self.end_date
            )
            if self.end_date
            else self.as_date
        )

    @property
    def display_time(self) -> str | None:
        """Time for display on web page."""
        return (
            self.date.strftime("%H:%M")
            if isinstance(self.date, datetime.datetime)
            else None
        )

    @property
    def display_timezone(self) -> str | None:
        """Timezone for display on web page."""
        return (
            self.date.strftime("%z")
            if isinstance(self.date, datetime.datetime)
            else None
        )

    def delta_days(self, today: datetime.date) -> str:
        """Return number of days from today as a string."""
        delta = (self.as_date - today).days

        match delta:
            case 0:
                return "today"
            case 1:
                return "1 day"
            case _:
                return f"{delta:,d} days"

    @property
    def display_date(self) -> str:
        """Date for display on web page."""
        if isinstance(self.date, datetime.datetime):
            return self.date.strftime("%a, %d, %b %Y %H:%M %z")
        else:
            return self.date.strftime("%a, %d, %b %Y")

    @property
    def display_title(self) -> str:
        """Name for display."""
        return self.title or self.name

    @property
    def emoji(self) -> str | None:
        """Emoji."""
        if self.title == "LHG Run Club":
            return "🏃🍻"
        return emojis.get(self.name)

    @property
    def title_with_emoji(self) -> str | None:
        """Title with optional emoji at the start."""
        title = self.title or self.name
        if title is None:
            return None
        emoji = self.emoji
        return f"{emoji} {title}" if emoji else title