hb_appointment_notification.py

import requests
import argparse
from requests.structures import CaseInsensitiveDict


def main():
    args = parse_args()
    cookie = extract_cookie()
    appointment_markup = get_appointment_markup(
        cookie, args["loc_nr"], args["cnc_nr"], args["startdate"], args["enddate"]
    )
    if appointment_available(appointment_markup):
        telegram_send_text(
            "🤖 Termin verfügbar: https://termin.bremen.de/termine/select2?md=5",
            args["telegramtoken"],
            args["telegramchatid"],
        )


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--loc_nr", required=True)
    parser.add_argument("--cnc_nr", required=True)
    parser.add_argument("--startdate", required=True)
    parser.add_argument("--enddate", required=True)
    parser.add_argument("--telegramtoken", required=True)
    parser.add_argument("--telegramchatid", required=True)
    return vars(parser.parse_args())


def extract_cookie():
    url = "https://termin.bremen.de/termine/select2?md=5"
    session = requests.Session()
    session.get(url)
    return session.cookies.get_dict()["TVWebSession"]


def get_appointment_markup(cookie, loc_nr, cnc_nr, startdate, enddate):
    url = f"https://termin.bremen.de/termine/suggest?loc={loc_nr}&mdt=0&cnc-{cnc_nr}=1&filter_date_from={startdate}&filter_date_to={enddate}&suggest_filter=Filtern"
    headers = CaseInsensitiveDict()
    headers["Cookie"] = f"TVWebSession={cookie}"
    resp = requests.get(url, headers=headers)
    return resp.text


def appointment_available(appointment_markup):
    return 'id="sugg_accordion"' in appointment_markup


def telegram_send_text(message, token, chat_id):
    send_text = (
        "https://api.telegram.org/bot"
        + token
        + "/sendMessage?chat_id="
        + chat_id
        + "&parse_mode=Markdown&text="
        + message
    )
    response = requests.get(send_text)
    return response.json()


if __name__ == "__main__":
    main()