#!/usr/bin/env python3
"""
Fuellt die Tabelle `calendar_days` mit Feiertagen und Schulferien.

Die Automatiken koennen mit "an Feiertagen ausfuehren" und "in den Ferien
ausfuehren" auf besondere Tage reagieren; woher das Wissen kommt, steht hier.
Nur besondere Tage werden eingetragen - ein fehlendes Datum ist ein
gewoehnlicher Tag.

Quelle ist openholidaysapi.org: ein offenes Verzeichnis der EU-Kommission, das
gesetzliche Feiertage und Schulferien fuer alle deutschen Bundeslaender
liefert. Ein Aufruf je Jahr genuegt, deshalb ist das ein Cronjob und kein
Dauerlaeufer:

    0 4 1 1 * /usr/bin/python3 /pfad/zu/fetch_calendar.py

Ein zusaetzlicher Lauf im Herbst schadet nicht - die Ferientermine des
uebernaechsten Schuljahres stehen erst spaeter fest.

Bereits eingetragene Tage werden ueberschrieben, nie doppelt angelegt.
"""

import argparse
import logging
import sys
from datetime import date, timedelta

import pymysql
import requests

from autoaction_runner import Config, verbinden

logger = logging.getLogger("kalender")

BASIS = "https://openholidaysapi.org"


def zeitraum(api, land, region, von, bis):
    """Eintraege einer Art (PublicHolidays oder SchoolHolidays) abholen."""
    antwort = requests.get(
        BASIS + "/" + api,
        params={"countryIsoCode": land, "subdivisionCode": region,
                "languageIsoCode": "DE", "validFrom": von.isoformat(),
                "validTo": bis.isoformat()},
        headers={"Accept": "application/json"}, timeout=20)
    antwort.raise_for_status()
    return antwort.json()


def name(eintrag):
    for teil in eintrag.get("name", []):
        if teil.get("text"):
            return teil["text"][:80]
    return "?"


def tage(eintrag):
    """Ferien erstrecken sich ueber Wochen; hier wird daraus Tag fuer Tag."""
    start = date.fromisoformat(eintrag["startDate"])
    ende = date.fromisoformat(eintrag["endDate"])
    while start <= ende:
        yield start
        start += timedelta(days=1)


def main():
    parser = argparse.ArgumentParser(description="Feiertage und Ferien in calendar_days schreiben.")
    parser.add_argument("--jahr", type=int, default=date.today().year)
    parser.add_argument("--jahre", type=int, default=2, help="wie viele Jahre ab --jahr")
    parser.add_argument("--config", default="config.ini")
    args = parser.parse_args()

    logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
    config = Config(args.config)
    land = config.text("calendar", "country", "DE")
    region = config.text("calendar", "subdivision", "DE-BY")

    von = date(args.jahr, 1, 1)
    bis = date(args.jahr + args.jahre - 1, 12, 31)
    logger.info("Hole %s bis %s fuer %s", von, bis, region)

    gesammelt = {}
    for eintrag in zeitraum("PublicHolidays", land, region, von, bis):
        for tag in tage(eintrag):
            gesammelt.setdefault(tag, {})["holiday"] = name(eintrag)
    for eintrag in zeitraum("SchoolHolidays", land, region, von, bis):
        for tag in tage(eintrag):
            gesammelt.setdefault(tag, {})["vacation"] = name(eintrag)

    db = verbinden(config)
    with db.cursor() as c:
        c.executemany(
            """INSERT INTO calendar_days (date, holiday, vacation) VALUES (%s, %s, %s)
               ON DUPLICATE KEY UPDATE holiday = VALUES(holiday), vacation = VALUES(vacation)""",
            [(tag, eintrag.get("holiday"), eintrag.get("vacation"))
             for tag, eintrag in sorted(gesammelt.items())])
    logger.info("%d Tage eingetragen", len(gesammelt))
    return 0


if __name__ == "__main__":
    sys.exit(main())