ical-parser: stringify an event label before printing

[etc/mutt.git] / .mutt / icalparser

diff --git a/.mutt/icalparser b/.mutt/icalparser
index 38e2fe7c9fbd8472558a73cbaf8ce996d2f6d25d..50ed4726865ec22c2525918a8fd5796f9df710ce 100755 (executable)
--- a/.mutt/icalparser
+++ b/.mutt/icalparser
@@ -13,8 +13,13 @@ COLUMNS = int(os.getenv('COLUMNS', 80))-10
 WIDTH = max(len(i) for i in FIELDS)
 INDENT = WIDTH + len(COLSEP)
 
-def dtstrs(start, end):
-    start, end = start.dt, end.dt
+def dtstrs(event):
+    start, end = event.get('dtstart').dt, event.get('dtend')
+    if not end:
+        duration = event.get('duration').dt
+        end = start + duration
+    else:
+        end = end.dt
 
     def strftime(dt, strf):
         return dt.astimezone().strftime(strf)
@@ -39,7 +44,7 @@ def parse_ics_file(fp):
 
     for event in cal.walk():
         if event.name != "VEVENT": continue
-        print('{} → {}'.format(*dtstrs(event.get('dtstart'), event.get('dtend'))))
+        print('{} → {}'.format(*dtstrs(event)))
         flags = []
         for k in ('status', 'class', 'transp', 'priority'):
             t = event.get(k)
@@ -47,7 +52,7 @@ def parse_ics_file(fp):
         event['flags'] = ' '.join('{}:{}'.format(*f) for f in flags)
         for label in FIELDS:
             if label in event:
-                text = fill(event[label], width=COLUMNS-INDENT,
+                text = fill(str(event[label]), width=COLUMNS-INDENT,
                             initial_indent='', subsequent_indent=' '*INDENT)
                 print(f'{label.capitalize():>{WIDTH}s}{COLSEP}{text}')