From 6202f768b15b7311bb3024da2b7f182c25f57043 Mon Sep 17 00:00:00 2001
From: Jefferson Stachelski <jstachel@thoughtworks.com>
Date: Wed, 29 Apr 2015 17:30:14 -0300
Subject: #337 Created a validation to just open file that the file name starts
 with 'mbox' and ends with a number

---
 service/pixelated/maintenance.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'service/pixelated/maintenance.py')

diff --git a/service/pixelated/maintenance.py b/service/pixelated/maintenance.py
index 909b2237..661f2576 100644
--- a/service/pixelated/maintenance.py
+++ b/service/pixelated/maintenance.py
@@ -20,6 +20,7 @@ import sys
 import json
 import argparse
 import email
+import re
 
 from os.path import join
 from mailbox import mboxMessage
@@ -147,17 +148,23 @@ def load_mails(args, mail_paths):
         for root, dirs, files in os.walk(path):
             mbx = account.getMailbox('INBOX')
             for file_name in files:
+                if not is_mail_file_name_valid(file_name):
+                    continue
                 with open(join(root, file_name), 'r') as email_file:
                     m = email.message_from_file(email_file)
                     flags = ("\\RECENT",)
-                    yield mbx.addMessage(m.as_string(), flags=flags, notify_on_disk=False)
                     print 'Added message %s' % m.get('subject')
                     print m.as_string()
+                    yield mbx.addMessage(m.as_string(), flags=flags, notify_on_disk=False)
 
     defer.returnValue(args)
     return
 
 
+def is_mail_file_name_valid(file_name):
+    return re.match('mbox[0-9]+$', file_name)
+
+
 def dump_soledad(args):
     leap_session, soledad = args
 
-- 
cgit v1.2.3