3 # Copyright (c) 2011-2012 David Bremner <david@tethera.net>
4 # License: Same as notmuch
6 # - python 2.6 for json
7 # - argparse; either python 2.7, or install separately
9 from __future__ import print_function
16 from urllib.parse import quote
17 except ImportError: # Python 2
18 from urllib import quote
26 _ENCODING = locale.getpreferredencoding() or sys.getdefaultencoding()
29 def read_config(path=None, encoding=None):
30 "Read config from json file"
36 nmbhome = os.getenv('NMBGIT', os.path.expanduser('~/.nmbug'))
38 # read only the first line from the pipe
39 sha1_bytes = subprocess.Popen(
40 ['git', '--git-dir', nmbhome, 'show-ref', '-s', 'config'],
41 stdout=subprocess.PIPE).stdout.readline()
42 sha1 = sha1_bytes.decode(encoding).rstrip()
44 fp_byte_stream = subprocess.Popen(
45 ['git', '--git-dir', nmbhome, 'cat-file', 'blob',
46 sha1+':status-config.json'],
47 stdout=subprocess.PIPE).stdout
48 fp = codecs.getreader(encoding=encoding)(stream=fp_byte_stream)
54 def __init__(self, last, lines):
58 def join_utf8_with_newlines(self):
59 return '\n'.join( (line.encode('utf-8') for line in self.lines) )
62 def output_with_separator(threadlist, sep):
63 outputs = (thread.join_utf8_with_newlines() for thread in threadlist)
64 print(sep.join(outputs))
67 def print_view(database, title, query, comment,
68 headers=('date', 'from', 'subject')):
70 query_string = ' and '.join(query)
71 q_new = notmuch.Query(database, query_string)
72 q_new.set_sort(notmuch.Query.SORT.OLDEST_FIRST)
81 if output_format == 'html':
82 print('<h3><a name="%s" />%s</h3>' % (title, title))
84 print('The view is generated from the following query:')
87 print('</blockquote>')
90 for m in q_new.search_messages():
92 thread_id = m.get_thread_id()
94 if thread_id != last_thread_id:
95 if threads.has_key(thread_id):
96 last = threads[thread_id].last
97 lines = threads[thread_id].lines
101 thread = Thread(last, lines)
102 threads[thread_id] = thread
105 threadlist.append(thread)
106 last_thread_id = thread_id
108 for header in headers:
109 val = m.get_header(header)
112 val = str.join(' ', val.split(None)[1:4])
113 val = str(datetime.datetime.strptime(val, '%d %b %Y').date())
114 elif header == 'from':
115 (val, addr) = email.utils.parseaddr(val)
117 val = addr.split('@')[0]
119 if header != 'subject' and last[header] == val:
125 mid = m.get_message_id()
126 out['id'] = 'id:"%s"' % mid
128 if output_format == 'html':
130 out['subject'] = '<a href="http://mid.gmane.org/%s">%s</a>' % (
131 quote(mid), out['subject'])
133 lines.append(' <tr><td>%s' % out['date'])
134 lines.append('</td><td>%s' % out['id'])
135 lines.append('</td></tr>')
136 lines.append(' <tr><td>%s' % out['from'])
137 lines.append('</td><td>%s' % out['subject'])
138 lines.append('</td></tr>')
140 lines.append('%(date)-10.10s %(from)-20.20s %(subject)-40.40s\n%(id)72s' % out)
142 if output_format == 'html':
143 output_with_separator(threadlist,
144 '\n<tr><td colspan="2"><br /></td></tr>\n')
147 output_with_separator(threadlist, '\n\n')
150 # parse command line arguments
152 parser = argparse.ArgumentParser()
153 parser.add_argument('--text', help='output plain text format',
155 parser.add_argument('--config', help='load config from given file',
157 parser.add_argument('--list-views', help='list views',
159 parser.add_argument('--get-query', help='get query for view',
162 args = parser.parse_args()
164 config = read_config(path=args.config)
167 for view in config['views']:
170 elif args.get_query != None:
171 for view in config['views']:
172 if args.get_query == view['title']:
173 print(' and '.join(view['query']))
176 # only import notmuch if needed
180 output_format = 'text'
182 output_format = 'html'
186 db = notmuch.Database(mode=notmuch.Database.MODE.READ_ONLY)
188 if output_format == 'html':
189 print('''<?xml version="1.0" encoding="utf-8" ?>
190 <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
191 <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
193 <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
194 <title>Notmuch Patches</title>
197 <h2>Notmuch Patches</h2>
198 Generated: {date}<br />
199 For more infomation see <a href="http://notmuchmail.org/nmbug">nmbug</a>
201 <ul>'''.format(date=datetime.datetime.utcnow().date()))
202 for view in config['views']:
203 print('<li><a href="#%(title)s">%(title)s</a></li>' % view)
206 for view in config['views']:
207 print_view(database=db, **view)
209 if output_format == 'html':
210 print('</body>\n</html>')