vytvor_ics.py

# --- Standardní knihovny ---
from datetime import datetime, timedelta, UTC
from uuid import uuid4

# --- Externí knihovny ---
from playwright.sync_api import sync_playwright

# --- Lokální moduly ---
from data import URL

# Slovník převodu měsíců
MESICE = {
    "ledna": 1, "února": 2, "března": 3, "dubna": 4,
    "května": 5, "června": 6, "července": 7, "srpna": 8,
    "září": 9, "října": 10, "listopadu": 11, "prosince": 12,
}

FILENAME = "zavod.ics"

# CSS selektory a lokátory
NAZEV_LOCATOR = (
    "body > div.min-h-screen.bg-gray-100.dark\\:bg-gray-900 > "
    "main > div.py-4 > div > div > div > div:nth-child(1) "
    "> div.justify-center.items-baseline.text-xl.font-bold.flex"
)

def nacti_text_ze_stranky(page, label_text: str) -> str:
    """Vrátí text hodnoty z páru 'label : value'."""
    label = page.locator(f"xpath=//div[normalize-space()='{label_text}']").first
    value = label.locator("xpath=following-sibling::div[1]")
    value.wait_for(timeout=5000)
    return value.inner_text().strip()

def ziskej_data(page):
    """Získá název, datum a střelnici ze stránky závodu."""
    nazev = page.inner_text(NAZEV_LOCATOR).strip()
    datum_raw = nacti_text_ze_stranky(page, "Datum závodu:")
    strelnice = nacti_text_ze_stranky(page, "Střelnice:")

    # 1. Odstraníme čárky (v "2026, sobota" by dělala neplechu)
    cisty_text = datum_raw.replace(",", "")
    
    # 2. Rozdělíme a vezmeme jen první 3 prvky, zbytek (den v týdnu) ignorujeme pomocí *_
    casti = cisty_text.split()
    if len(casti) < 3:
        raise ValueError(f"Neočekávaný formát data: {datum_raw}")
        
    den_str, mesic_str, rok_str = casti[0], casti[1], casti[2]
    
    den = int(den_str.replace(".", ""))
    mesic = MESICE[mesic_str.lower()] # Přidáno .lower() pro jistotu
    rok = int(rok_str)

    datum = datetime(rok, mesic, den)
    return nazev, datum, strelnice, datum_raw

def _fmt_date(d: datetime) -> str:
    """YYYYMMDD pro celodenní události."""
    return d.strftime("%Y%m%d")

def _fmt_dt_utc(dt: datetime) -> str:
    """YYYYMMDDTHHMMSSZ – UTC pro DTSTAMP."""
    return dt.astimezone(UTC).strftime("%Y%m%dT%H%M%SZ")

def vytvor_ics_google(nazev: str, datum: datetime, strelnice: str, filename: str = FILENAME) -> None:
    """
    Vytvoří ICS kompatibilní s Google Kalendářem.
    Celodenní událost: DTSTART;VALUE=DATE a DTEND;VALUE=DATE (o den později).
    """
    dtstart = _fmt_date(datum)
    dtend = _fmt_date(datum + timedelta(days=1))  # end-exclusive, o den později
    dtstamp = _fmt_dt_utc(datetime.now(UTC))
    uid = f"{uuid4()}@local"

    lines = [
        "BEGIN:VCALENDAR",
        "VERSION:2.0",
        "PRODID:-//KlikaTools//Event Generator//CZ",
        "CALSCALE:GREGORIAN",
        "METHOD:PUBLISH",
        "BEGIN:VEVENT",
        f"UID:{uid}",
        f"DTSTAMP:{dtstamp}",
        f"DTSTART;VALUE=DATE:{dtstart}",
        f"DTEND;VALUE=DATE:{dtend}",
        f"SUMMARY:{nazev}",
        f"LOCATION:{strelnice}",
        "END:VEVENT",
        "END:VCALENDAR",
    ]

    ics_text = "\r\n".join(lines) + "\r\n"
    with open(filename, "w", encoding="utf-8", newline="") as f:
        f.write(ics_text)

    # print(f"✅ ICS soubor byl vytvořen: {filename}")

def main():
    """Hlavní funkce – načte stránku, získá data a uloží ICS."""
    with sync_playwright() as p:
        browser = p.chromium.launch(headless=True)
        page = browser.new_page()

        try:
            page.goto(URL, timeout=10000)
        except Exception as e:
            print(f"❌ Nelze načíst stránku {URL}.\n{e}")
            return

        if page.title() == "Nenalezeno":
            print(f"❌ Stránka závodu {URL} nebyla nalezena (404).")
            return

        nazev, datum, strelnice, _ = ziskej_data(page)
        vytvor_ics_google(nazev, datum, strelnice, FILENAME)
        return FILENAME

if __name__ == "__main__":
    try:
        main()
    except Exception as e:
        print(f"⚠️ Neošetřená chyba: {e}")