1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#!/usr/bin/env python
"""
mutt-notmuch-py
This is a Gmail-only version of the original mutt-notmuch script.
It will interactively ask you for a search query and then symlink the matching
messages to $HOME/.cache/mutt_results.
Add this to your muttrc.
macro index / "<enter-command>unset wait_key<enter><shell-escape>mutt-notmuch-py<enter><change-folder-readonly>~/.cache/mutt_results<enter>" \
"search mail (using notmuch)"
This script overrides the $HOME/.cache/mutt_results each time you run a query.
Install this by adding this file somewhere on your PATH.
Only tested on OSX Lion.
(c) 2012 - Honza Pokorny
Licensed under BSD
"""
import hashlib, sys
from subprocess import getoutput
from mailbox import Maildir
from optparse import OptionParser
def digest(filename):
with open(filename, "r", encoding='utf-8') as f:
return hashlib.sha1(f.read()).hexdigest()
def pick_all_mail(messages):
for m in messages:
if 'All Mail' in m:
return m
def empty_dir(directory):
box = Maildir(directory)
box.clear()
def command(cmd):
return getoutput(cmd)
def main(dest_box):
query = input('Query: ').encode('utf-8')
command('mkdir -p %s' % dest_box)
command('mkdir -p %s/cur' % dest_box)
command('mkdir -p %s/new' % dest_box)
empty_dir(dest_box)
files = command('notmuch search --output=files %s' % query).split('\n')
files = filter(None, files)
data = {}
messages = []
for f in files:
sha = digest(f)
if sha not in data.keys():
data[sha] = [f]
else:
data[sha].append(f)
for sha in data.keys():
if is_gmail and len(data[sha]) > 1:
messages.append(pick_all_mail(data[sha]))
else:
messages.append(data[sha][0])
for m in messages:
command('ln -s "%s" %s/cur/' % (m, dest_box))
if __name__ == '__main__':
global is_gmail
p = OptionParser("usage: %prog [OPTIONS] [RESULTDIR]")
p.add_option('-g', '--gmail', dest='gmail',
action='store_true', default=True,
help='gmail-specific behavior')
p.add_option('-G', '--not-gmail', dest='gmail',
action='store_false',
help='gmail-specific behavior')
(options, args) = p.parse_args()
is_gmail = options.gmail
if args:
dest = args[0]
else:
dest = '~/.cache/mutt_results'
main(dest.rstrip('/'))
|
