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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
#!/usr/bin/env python3
"""
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.
Tested on OSX Lion and Arch Linux.
(c) 2012 - Honza Pokorny
Licensed under BSD
"""
"""
ChangeLogs:
2015/02/02: Weitian LI
* replaced 'commands.getoutput' with 'subprocess'
* replaced 'raw_input' with 'input'
* fixed digest() with an Unicode encoding bug
* updated to use python 3.x
"""
import os
import hashlib
import subprocess
from mailbox import Maildir
from optparse import OptionParser
from collections import defaultdict
def digest(filename):
# Require python 3.x
with open(filename) as f:
return hashlib.sha1(f.read().encode('utf-8')).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 subprocess.check_output(cmd, shell=True)
def main(dest_box, is_gmail):
query = input('Query: ')
command('test -d %s/cur || mkdir -p %s/cur' % (dest_box, dest_box))
command('test -d %s/new || mkdir -p %s/new' % (dest_box, dest_box))
empty_dir(dest_box)
output = command('notmuch search --output=files %s' % query)
files = output.decode(encoding='UTF-8').split('\n')
data = defaultdict(list)
messages = []
for f in files:
if not f:
continue
try:
sha = digest(f)
data[sha].append(f)
except IOError:
print('File %s does not exist' % f)
for sha in data:
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:
if not m:
continue
target = os.path.join(dest_box, 'cur', os.path.basename(m))
if not os.path.exists(target):
os.symlink(m, target)
if __name__ == '__main__':
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()
if args:
dest = args[0]
else:
dest = '~/.cache/mutt_results'
# Use expanduser() so that os.symlink() won't get weirded out by tildes.
main(os.path.expanduser(dest).rstrip('/'), options.gmail)
|
