notmuch/devel/nmbug/nmbug-status

#!/usr/bin/python
#
# Copyright (c) 2011-2012 David Bremner <david@tethera.net>
# License: Same as notmuch
# dependencies
#       - python 2.6 for json
#       - argparse; either python 2.7, or install separately

from __future__ import print_function

import codecs
import datetime
import email.utils
import locale
import urllib
import json
import argparse
import os
import sys
import subprocess


_ENCODING = locale.getpreferredencoding() or sys.getdefaultencoding()


def read_config(path=None, encoding=None):
    "Read config from json file"
    if not encoding:
        encoding = _ENCODING
    if path:
        fp = open(path)
    else:
        nmbhome = os.getenv('NMBGIT', os.path.expanduser('~/.nmbug'))

        # read only the first line from the pipe
        sha1_bytes = subprocess.Popen(
            ['git', '--git-dir', nmbhome, 'show-ref', '-s', 'config'],
            stdout=subprocess.PIPE).stdout.readline()
        sha1 = sha1_bytes.decode(encoding).rstrip()

        fp_byte_stream = subprocess.Popen(
            ['git', '--git-dir', nmbhome, 'cat-file', 'blob',
             sha1+':status-config.json'],
            stdout=subprocess.PIPE).stdout
        fp = codecs.getreader(encoding=encoding)(stream=fp_byte_stream)

    return json.load(fp)


class Thread:
    def __init__(self, last, lines):
        self.last = last
        self.lines = lines

    def join_utf8_with_newlines(self):
        return '\n'.join( (line.encode('utf-8') for line in self.lines) )


def output_with_separator(threadlist, sep):
    outputs = (thread.join_utf8_with_newlines() for thread in threadlist)
    print(sep.join(outputs))


def print_view(database, title, query, comment,
               headers=('date', 'from', 'subject')):

    query_string = ' and '.join(query)
    q_new = notmuch.Query(database, query_string)
    q_new.set_sort(notmuch.Query.SORT.OLDEST_FIRST)

    last_thread_id = ''
    threads = {}
    threadlist = []
    out = {}
    last = None
    lines = None

    if output_format == 'html':
        print('<h3><a name="%s" />%s</h3>' % (title, title))
        print(comment)
        print('The view is generated from the following query:')
        print('<blockquote>')
        print(query_string)
        print('</blockquote>')
        print('<table>\n')

    for m in q_new.search_messages():

        thread_id = m.get_thread_id()

        if thread_id != last_thread_id:
            if threads.has_key(thread_id):
                last = threads[thread_id].last
                lines = threads[thread_id].lines
            else:
                last = {}
                lines = []
                thread = Thread(last, lines)
                threads[thread_id] = thread
                for h in headers:
                    last[h] = ''
                threadlist.append(thread)
            last_thread_id = thread_id

        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) = email.utils.parseaddr(val)
                if val == '':
                    val = addr.split('@')[0]

            if header != 'subject' and last[header] == val:
                out[header] = ''
            else:
                out[header] = val
                last[header] = val

        mid = m.get_message_id()
        out['id'] = 'id:"%s"' % mid

        if output_format == 'html':

            out['subject'] = '<a href="http://mid.gmane.org/%s">%s</a>' \
                % (urllib.quote(mid), out['subject'])

            lines.append(' <tr><td>%s' % out['date'])
            lines.append('</td><td>%s' % out['id'])
            lines.append('</td></tr>')
            lines.append(' <tr><td>%s' % out['from'])
            lines.append('</td><td>%s' % out['subject'])
            lines.append('</td></tr>')
        else:
            lines.append('%(date)-10.10s %(from)-20.20s %(subject)-40.40s\n%(id)72s' % out)

    if output_format == 'html':
        output_with_separator(threadlist,
                              '\n<tr><td colspan="2"><br /></td></tr>\n')
        print('</table>')
    else:
        output_with_separator(threadlist, '\n\n')


# 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',
                    metavar='PATH')
parser.add_argument('--list-views', help='list views',
                    action='store_true')
parser.add_argument('--get-query', help='get query for view',
                    metavar='VIEW')

args = parser.parse_args()

config = read_config(path=args.config)

if args.list_views:
    for view in config['views']:
        print(view['title'])
    sys.exit(0)
elif args.get_query != None:
    for view in config['views']:
        if args.get_query == view['title']:
            print(' and '.join(view['query']))
    sys.exit(0)
else:
    # only import notmuch if needed
    import notmuch

if args.text:
    output_format = 'text'
else:
    output_format = 'html'

# main program

db = notmuch.Database(mode=notmuch.Database.MODE.READ_WRITE)

if output_format == 'html':
    print('''<?xml version="1.0" encoding="utf-8" ?>
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
<head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
<title>Notmuch Patches</title>
</head>
<body>''')
    print('<h2>Notmuch Patches</h2>')
    print('Generated: %s<br />' % datetime.datetime.utcnow().date())
    print('For more infomation see <a href="http://notmuchmail.org/nmbug">nmbug</a>')

    print('<h3>Views</h3>')
    print('<ul>')
    for view in config['views']:
        print('<li><a href="#%(title)s">%(title)s</a></li>' % view)
    print('</ul>')

for view in config['views']:
    print_view(database=db, **view)

if output_format == 'html':
    print('</body>\n</html>')