.config/mutt/icalparser

   1 #!/usr/bin/python3
   2
   3 import sys
   4 import os
   5 from icalendar import Calendar, Event
   6 from textwrap import fill, indent
   7
   8 FIELDS = ('summary','location','organiser',
   9           'flags','uid','rrule')
  10 COLSEP = ': '
  11
  12 COLUMNS = int(os.getenv('COLUMNS', 80))-10
  13 WIDTH = max(len(i) for i in FIELDS)
  14 INDENT = WIDTH + len(COLSEP)
  15
  16 def dtstrs(event):
  17     start, end = event.get('dtstart').dt, event.get('dtend')
  18     if not end:
  19         duration = event.get('duration').dt
  20         end = start + duration
  21     else:
  22         end = end.dt
  23
  24     def strftime(dt, strf):
  25         return dt.astimezone().strftime(strf)
  26
  27     ret = []
  28
  29     try:
  30         if start.tzinfo == end.tzinfo:
  31             ret.append(strftime(start, '%F %R'))
  32
  33         else:
  34             ret.append(strftime(start, '%F %R %Z'))
  35
  36         if start.date() == end.date():
  37             ret.append(strftime(end, '%T %Z'))
  38
  39         else:
  40             ret.append(strftime(end, '%F %R %Z'))
  41
  42         return ret
  43
  44     except AttributeError:
  45         return map(lambda d: d.strftime('%F'), (start, end))
  46
  47 def parse_ics_file(fp):
  48     cal = Calendar.from_ical(fp.read())
  49
  50     for event in cal.walk():
  51         if event.name != "VEVENT": continue
  52         print('{} → {}'.format(*dtstrs(event)))
  53         flags = []
  54         for k in ('status', 'class', 'transp', 'priority'):
  55             t = event.get(k)
  56             if t: flags.append((k, t))
  57         event['flags'] = ' '.join('{}:{}'.format(*f) for f in flags)
  58         for label in FIELDS:
  59             if label in event:
  60                 text = fill(str(event[label]), width=COLUMNS-INDENT,
  61                             initial_indent='', subsequent_indent=' '*INDENT)
  62                 print(f'{label.capitalize():>{WIDTH}s}{COLSEP}{text}')
  63
  64         if 'description' in event:
  65             output = ['']
  66             lines = event['description'].split('\n')
  67             for line in lines:
  68                 output.append(fill(line, width=COLUMNS, initial_indent=' ',
  69                                    subsequent_indent=' '))
  70             print('\n'.join(output))
  71
  72 if __name__ == '__main__':
  73
  74     if len(sys.argv) > 1 and sys.argv[1] != '-':
  75         for f in sys.argv[1:]:
  76             with open(f) as fp:
  77                 parse_ics_file(fp)
  78     else:
  79         parse_ics_file(sys.stdin)