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
|
#!/usr/bin/python
import email
import imaplib
import logging
import os.path
import subprocess
from ConfigParser import ConfigParser
logging.basicConfig(format='%(levelname)s:%(funcName)s:%(message)s',
level=logging.DEBUG)
CMD_STR = "/usr/bin/bogofilter -%s"
imaplib.Debug = 4
def process_folder(proc_fld_name, bogofilter_param, end_fld_name):
client.select(proc_fld_name)
_, resp = client.search(None, "ALL")
messages = resp[0].split()
logging.debug('messages = {0}'.format(messages))
proc_msg_count = 0
for msgId in messages:
logging.debug('msgId = {0}'.format(msgId))
typ, msg_data = client.fetch(msgId, '(RFC822)')
msg = hparser.parsestr(msg_data[0][1])
ret = subprocess.Popen(CMD_STR % bogofilter_param,
stdin=subprocess.PIPE,
shell=True)
ret.communicate(input=msg_data[0][1])
logging.debug("ret.returncode = %s", ret.returncode)
if ret.returncode == 0:
del msg['X-Bogosity']
typ, _ = client.append(end_fld_name, '', '',
msg.as_string(True))
logging.debug("typ = %s", typ)
if typ != 'OK':
raise IOError("Cannot store a message to the folder %s"
% end_fld_name)
else:
client.store(msgId, '+FLAGS', r'(\Deleted)')
proc_msg_count += 1
else:
raise OSError("bogofilter finished with the returncode: %d"
% ret)
client.expunge()
client.close()
return proc_msg_count
processedCounter = 0
hparser = email.Parser.Parser()
config = ConfigParser()
config.read(os.path.expanduser("~/.bogofilter-imap-train-rc"))
login = config.get("imap-training", "login")
password = config.get("imap-training", "password")
server = config.get("imap-training", "server")
client = imaplib.IMAP4_SSL(server)
client.login(login, password)
for box in [('Spam', 's', 'Deleted Items'), ('Ham', 'n', 'INBOX')]:
logging.debug('box = {0}'.format(box))
processedCounter += process_folder(box[0], box[1], box[2])
client.logout()
if processedCounter > 0:
logging.info("Processed %d spam messages.", processedCounter)
|
