From a6ecb48b548b46453f345a296e8d63998a4e1c4f Mon Sep 17 00:00:00 2001 From: "martin f. krafft" Date: Wed, 4 Sep 2019 10:22:50 +1200 Subject: [PATCH] prettify icalparser output --- .mutt/icalparser | 34 ++++++++++++++++++++++++++-------- 1 file changed, 26 insertions(+), 8 deletions(-) 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) -- 2.39.5