handle events with duration rather than end time

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

diff --git a/.mutt/icalparser b/.mutt/icalparser
index 38e2fe7c9fbd8472558a73cbaf8ce996d2f6d25d..77e7fe2880cf4b1090ddf9be14ca416ea2304714 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)
 
 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)
 
     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
 
     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)
         flags = []
         for k in ('status', 'class', 'transp', 'priority'):
             t = event.get(k)