diff options
-rwxr-xr-x | train_bogofilter | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/train_bogofilter b/train_bogofilter index 45d7c7f..48f20bc 100755 --- a/train_bogofilter +++ b/train_bogofilter @@ -11,7 +11,7 @@ logging.basicConfig(format='%(levelname)s:%(funcName)s:%(message)s', CMD_STR = "/usr/bin/bogofilter -%s" imaplib.Debug = 4 -def process_folder(proc_fld_name, bogofilter_param, end_fld_name): +def process_folder(proc_fld_name, bogofilter_param, end_fld_name, mark_seen=True): client.select(proc_fld_name) _, resp = client.search(None, "ALL") messages = resp[0].split() @@ -32,9 +32,12 @@ def process_folder(proc_fld_name, bogofilter_param, end_fld_name): 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)) + typ, newmsg = client.append(end_fld_name, '', '', + msg.as_string(True)) logging.debug("typ = %s", typ) + logging.debug("newmsg = %s", newmsg) + # if mark_seen: + # client.store(newmsg, '+FLAGS', r'(\Seen)') if typ != 'OK': raise IOError("Cannot store a message to the folder %s" % end_fld_name) @@ -61,9 +64,10 @@ server = config.get("imap-training", "server") client = imaplib.IMAP4_SSL(server) client.login(login, password) -for box in [('Junk', 's', 'Deleted Items'), ('Ham', 'n', 'INBOX')]: +for box in [('Junk', 's', 'Trash', True), ('Ham', 'n', 'INBOX', False)]: logging.debug('box = %s', box) - processedCounter += process_folder(box[0], box[1], box[2]) + # processedCounter += process_folder(box[0], box[1], box[2], box[3) + processedCounter += process_folder(*box) client.logout() |