Notmuch Patches

#!/usr/bin/python # # Copyright (c) 2011-2012 David Bremner # License: Same as notmuch # dependencies # - python 2.6 for json # - argparse; either python 2.7, or install separately import datetime import notmuch import rfc822 import urllib import json import argparse import os import subprocess # parse command line arguments parser = argparse.ArgumentParser() parser.add_argument('--text', help='output plain text format', action='store_true') parser.add_argument('--config', help='load config from given file') args = parser.parse_args() # read config from json file if args.config != None: fp = open(args.config) else: nmbhome = os.getenv('NMBGIT', os.path.expanduser('~/.nmbug')) # read only the first line from the pipe sha1 = subprocess.Popen(['git', '--git-dir', nmbhome, 'show-ref', '-s', 'config'], stdout=subprocess.PIPE).stdout.readline() sha1 = sha1.rstrip() fp = subprocess.Popen(['git', '--git-dir', nmbhome, 'cat-file', 'blob', sha1+':status-config.json'], stdout=subprocess.PIPE).stdout config = json.load(fp) if args.text: output_format = 'text' else: output_format = 'html' headers = ['date', 'from', 'subject'] last = {} def clear_last(): for header in headers: last[header] = '' def print_view(title, query, comment): query_string = ' and '.join(query) q_new = notmuch.Query(db, query_string) q_new.set_sort(notmuch.Query.SORT.OLDEST_FIRST) last['thread_id'] = '' if output_format == 'html': print '

%s

' % title print comment print 'The view is generated from the following query:' print '

' print query_string print '

' print '\n' for m in q_new.search_messages(): out = {} thread_id = m.get_thread_id() if thread_id != last['thread_id']: clear_last() for header in headers: val = m.get_header(header) if header == 'date': val = str.join(' ', val.split(None)[1:4]) val = str(datetime.datetime.strptime(val, '%d %b %Y').date()) elif header == 'from': (val, addr) = rfc822.parseaddr(val) if val == '': val = addr.split('@')[0] if last[header] == val: out[header] = '' else: out[header] = val.encode('utf-8') last[header] = val mid = m.get_message_id() out['id'] = 'id:"%s"' % mid if output_format == 'html': # XXX using
is a hack, but ... // 20111216 too if thread_id != last['thread_id']: br = '
' else: br = '' out['subject'] = '%s' \ % (urllib.quote(mid), out['subject']) print ' ' print ' \n' else: print '%(date)-10.10s %(from)-20.20s %(subject)-40.40s\n%(id)72s\n' % out last['thread_id'] = thread_id if output_format == 'html': print '

%s %s' % (br, out['date']) print '	%s %s' % (br, out['id']) print '
%s' % out['from'] print '	%s' % out['subject'] print '

' # main program db = notmuch.Database(mode=notmuch.Database.MODE.READ_WRITE) if output_format == 'html': print ''' Notmuch Patches ''' print '

Notmuch Patches

' print 'Generated: %s
' % datetime.datetime.utcnow().date() print 'For more infomation see nmbug' for view in config['views']: print_view(**view) if output_format == 'html': print '\n'