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
121
|
#!/usr/bin/env python3
"""
Modified from github:honza/mutt-notmuch-py
This is a recoll 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-recoll.py<enter><change-folder-readonly>~/.cache/mutt_results<enter>" \
"search mail (using recoll)"
This script overrides the $HOME/.cache/mutt_results each time you run a query.
Install this by adding this file somewhere on your PATH.
(c) 2012 - Honza Pokorny
(c) 2014 - Jean-Francois Dockes
Licensed under BSD
"""
import os
import hashlib
from commands import getoutput
from mailbox import Maildir
from optparse import OptionParser
from collections import defaultdict
def digest(filename):
with open(filename) 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, is_gmail):
query = raw_input("Query: ")
command("mkdir -p %s/cur" % dest_box)
command("mkdir -p %s/new" % dest_box)
empty_dir(dest_box)
files = command("recoll -t -b -q %s" % query).split("\n")
data = defaultdict(list)
messages = []
for f in files:
# Recoll outputs file:// urls
f = f[7:]
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):
print("symlink [%s] -> [%s]" % (m, 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 = os.path.expanduser("~/.cache/mutt_results")
if not os.path.exists(dest):
os.makedirs(dest)
# Use expanduser() so that os.symlink() won't get weirded out by tildes.
main(os.path.expanduser(dest).rstrip("/"), options.gmail)
|
