From: martin f. krafft <madduck@madduck.net>
Date: Tue, 3 Sep 2019 22:22:50 +0000 (+1200)
Subject: prettify icalparser output
X-Git-Url: https://git.madduck.net/etc/mutt.git/commitdiff_plain/a6ecb48b548b46453f345a296e8d63998a4e1c4f?hp=f71e8e88aab41e7cfdba3550239c9ad9a7326768

prettify icalparser output
---

diff --git a/.mutt/icalparser b/.mutt/icalparser
index b7a28c9..38e2fe7 100755
--- a/.mutt/icalparser
+++ b/.mutt/icalparser
@@ -3,6 +3,15 @@
 import sys
 import os
 from icalendar import Calendar, Event
+from textwrap import fill, indent
+
+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(start, end):
     start, end = start.dt, end.dt
@@ -30,22 +39,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})')))
         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)