import icalendar
import requests
import datetime
import copy
import sys

calendar = icalendar.Calendar.from_ical(requests.get("https://www.mff.cuni.cz/cs/studenti/harmonogram-ak-roku/harmonogram-2024-2025?ical&study_type=study_type-B").text)

assert isinstance(calendar, icalendar.cal.Calendar)

new_calendar = icalendar.cal.Calendar()

for event in calendar.events:
    if event.get("SUMMARY") == "Událost harmonogramu akademického roku..." and "DESCRIPTION" in event:
        event["SUMMARY"] = event["DESCRIPTION"]
    if event.duration > datetime.timedelta(7):
        end_event = copy.copy(event)
        event.end = event.start + datetime.timedelta(1)
        end_event.start = end_event.end - datetime.timedelta(1)
        end_event["SUMMARY"] = "KONEC: " + end_event.get("SUMMARY", "(Bez názvu)")
        event["SUMMARY"] = "ZAČÁTEK: " + event.get("SUMMARY", "(Bez názvu)")
        new_calendar.add_component(end_event)
    new_calendar.add_component(event)

# for event in new_calendar.events:
#     print(event.duration, event.get("SUMMARY"))

sys.stdout.buffer.write(new_calendar.to_ical())

