handle events with duration rather than end time

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

diff --git a/.mutt/icalparser b/.mutt/icalparser
index b7a28c96589ea14fc910347878cf31e4eafc4632..77e7fe2880cf4b1090ddf9be14ca416ea2304714 100755 (executable)
--- a/.mutt/icalparser
+++ b/.mutt/icalparser
@@ -3,9 +3,23 @@
 import sys
 import os
 from icalendar import Calendar, Event
+from textwrap import fill, indent
 
-def dtstrs(start, end):
-    start, end = start.dt, end.dt
+FIELDS = ('summary','location','organiser',
+          'flags','uid','rrule')
+COLSEP = ': '
+
+COLUMNS = int(os.getenv('COLUMNS', 80))-10
+WIDTH = max(len(i) for i in FIELDS)
+INDENT = WIDTH + len(COLSEP)
+
+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)
@@ -30,22 +44,31 @@ def parse_ics_file(fp):
 
     for event in cal.walk():
         if event.name != "VEVENT": continue
-        print('=' * (int(os.getenv('COLUMNS', 80))-10))
-        print('{} → {}'.format(*dtstrs(event.get('dtstart'), event.get('dtend'))))
-        for i in ('summary','organizer','location','uid','rrule'):
-            print('  {:9s}: {}'.format(i.capitalize(), event.get(i, f'(no {i})')))
+        print('{} → {}'.format(*dtstrs(event)))
         flags = []
         for k in ('status', 'class', 'transp', 'priority'):
             t = event.get(k)
             if t: flags.append((k, t))
-        print('  Flags    : ' + ' '.join('{}:{}'.format(*f) for f in flags))
+        event['flags'] = ' '.join('{}:{}'.format(*f) for f in flags)
+        for label in FIELDS:
+            if label in event:
+                text = fill(event[label], width=COLUMNS-INDENT,
+                            initial_indent='', subsequent_indent=' '*INDENT)
+                print(f'{label.capitalize():>{WIDTH}s}{COLSEP}{text}')
+
+        if 'description' in event:
+            output = ['']
+            lines = event['description'].split('\n')
+            for line in lines:
+                output.append(fill(line, width=COLUMNS, initial_indent=' ',
+                                   subsequent_indent=' '))
+            print('\n'.join(output))
 
 if __name__ == '__main__':
 
-    if len(sys.argv) > 1:
+    if len(sys.argv) > 1 and sys.argv[1] != '-':
         for f in sys.argv[1:]:
-            fp = open(f)
-            parse_ics_file(fp)
-            fp.close()
+            with open(f) as fp:
+                parse_ics_file(fp)
     else:
         parse_ics_file(sys.stdin)