mirror of
https://git.bolliret.ch/pcs/pcs-website
synced 2026-01-18 20:41:38 +01:00
81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
from datetime import datetime, date, timedelta, timezone
|
|
import recurring_ical_events
|
|
import requests
|
|
from copy import deepcopy
|
|
from icalendar import Calendar, Event
|
|
import locale
|
|
import sys
|
|
|
|
|
|
#This is duplicated code from calendar-fetcher-main.py
|
|
def split_multiday_events(events):
|
|
"""
|
|
Split multi-day events into separate daily events.
|
|
|
|
Args:
|
|
events: Iterator of iCal events
|
|
|
|
Returns:
|
|
List of events where multi-day events are split into daily events
|
|
"""
|
|
split_events = []
|
|
|
|
for event in events:
|
|
# Get start and end dates
|
|
start = event.get('dtstart').dt
|
|
end = event.get('dtend').dt if event.get('dtend') else start
|
|
|
|
# Convert datetime to date if needed
|
|
if isinstance(start, datetime):
|
|
start = start.date()
|
|
if isinstance(end, datetime):
|
|
end = end.date()
|
|
|
|
# If it's a single day event or not a date/datetime, add as is
|
|
if not isinstance(start, date) or not isinstance(end, date) or start == end:
|
|
split_events.append(event)
|
|
continue
|
|
|
|
# For multi-day events, create separate events for each day
|
|
current_date = start
|
|
while current_date < end:
|
|
# Create a copy of the original event
|
|
daily_event = Event()
|
|
for key in event:
|
|
daily_event[key] = deepcopy(event[key])
|
|
|
|
|
|
# Update the date for this instance
|
|
daily_event['dtstart'].dt = current_date
|
|
|
|
if 'dtend' in daily_event:
|
|
daily_event['dtend'].dt = current_date + timedelta(days=1)
|
|
|
|
split_events.append(daily_event)
|
|
current_date += timedelta(days=1)
|
|
return split_events
|
|
|
|
|
|
def get_events(ics_url, tgt_locale):
|
|
# Set German locale
|
|
locale.setlocale(locale.LC_TIME, tgt_locale)
|
|
|
|
response = requests.get(ics_url)
|
|
calendar = Calendar.from_ical(response.content)
|
|
|
|
now=datetime.now(timezone.utc) # + timedelta(days=1)
|
|
|
|
raw_events = split_multiday_events(recurring_ical_events.of(calendar).after(now))
|
|
|
|
now_date = now.date()
|
|
now_datetime = now
|
|
events = []
|
|
for raw_event in raw_events:
|
|
start = raw_event.get('dtstart').dt
|
|
if isinstance(start, datetime):
|
|
my_now = now_datetime
|
|
else:
|
|
my_now = now_date
|
|
if start >= my_now:
|
|
events.append(raw_event)
|
|
return events
|