holiday.py

import csv
from datetime import date, timedelta
from functools import cache


# Get all holydays from the official federal list from:
# https://www.data.gv.at/katalog/en/dataset/3deb9da7-8394-4797-9f32-5ca95281ba5b
# Note: yes this might look as not as sustainable as downloading from an API,
# but since the csv includes dates for the next 10 years I think we will be
# good. Also the webscraping will break much faster than this.
@cache
def holidays() -> set[date]:
    with open("holiday.csv", encoding="utf-8") as f:
        reader = csv.DictReader(f)

        return {
            day
            for row in reader
            if row["TYP"] == "HF"
            and date.today()
            < (day := date.fromisoformat(row["DATUM"]))
            < (date.today() + timedelta(days=365))
        }