94 lines
2.6 KiB
Python
Executable file
94 lines
2.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import logging
|
|
import locale
|
|
from dateutil.parser import parse
|
|
from datetime import datetime, timedelta
|
|
from dateutil.rrule import rruleset, rrulestr
|
|
|
|
import icalendar
|
|
|
|
|
|
def vevent_to_event(event, rrstart=None):
|
|
if rrstart == None:
|
|
begin = parse(event["DTSTART"].to_ical())
|
|
else:
|
|
begin = rrstart
|
|
|
|
return {
|
|
"name": event["SUMMARY"].to_ical(),
|
|
"url": event["URL"].to_ical(),
|
|
"begin": begin
|
|
}
|
|
|
|
|
|
def parse_single_event(event, start, end):
|
|
logging.info(f"Processing single event {event['SUMMARY'].to_ical().decode('utf-8')}")
|
|
dtstart = parse(event["DTSTART"].to_ical())
|
|
if dtstart >= start and dtstart < end:
|
|
return vevent_to_event(event)
|
|
|
|
|
|
def parse_recurring_event(event, start, end):
|
|
logging.info(f"Processing recurring event {event['SUMMARY'].to_ical().decode('utf-8')}")
|
|
dtstart = parse(event["DTSTART"].to_ical())
|
|
rs = rruleset()
|
|
rs.rrule(rrulestr(event["RRULE"].to_ical().decode("utf-8"), dtstart=dtstart))
|
|
if "EXDATE" in event.keys():
|
|
for exdate in event["EXDATE"]:
|
|
rs.exdate(parse(exdate.to_ical()))
|
|
|
|
events = []
|
|
for date in list(rs):
|
|
if date >= start and date < end:
|
|
events.append(vevent_to_event(event, date))
|
|
|
|
return events
|
|
|
|
|
|
def find_events(icsfilestr, start, end, num):
|
|
with open(icsfilestr, "r") as icsfile:
|
|
cal = icalendar.Calendar.from_ical(icsfile.read())
|
|
|
|
events = []
|
|
for event in cal.subcomponents:
|
|
if event.name == "VEVENT":
|
|
if "RRULE" in event.keys():
|
|
events.append(parse_recurring_event(event, start, end))
|
|
elif ev := parse_single_event(event, start, end) != None:
|
|
events.append(ev)
|
|
|
|
events = sorted(events, key=lambda k: k["begin"])
|
|
events = events[0:num]
|
|
|
|
return events
|
|
|
|
|
|
def format_events(events):
|
|
print("<table class=\"table table-condensed\">")
|
|
for event in events:
|
|
print(
|
|
"<tr>"
|
|
f"<td>{event['begin'].strftime('%A, %d.%m um %H:%M Uhr')}</td>"
|
|
f"<td><a href=\"{event['url'].decode('utf-8')}\">{event['name'].decode('utf-8')}</a></td>"
|
|
"</tr>"
|
|
)
|
|
print("</table><!--/.table .table-condensed-->")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < 3:
|
|
print(f"Usage: {sys.argv[0]} calendar max_days max_items")
|
|
sys.exit(-1)
|
|
|
|
locale.setlocale(locale.LC_TIME, "de_DE.UTF-8")
|
|
calendar = sys.argv[1]
|
|
max_days = int(sys.argv[2])
|
|
max_items = int(sys.argv[3])
|
|
|
|
now = datetime.now()
|
|
events = find_events(calendar, now, now + timedelta(days=max_days), max_items)
|
|
format_events(events)
|
|
|