Clean up code

ProjectClarity · Sep 13, 2014 · c8849fd · c8849fd
1 parent 2121142
commit c8849fd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/main.py b/main.py
@@ -9,7 +9,6 @@
 if __name__ == '__main__':
     while True:
         newest_notifications = get_notifications(MAX_NOTIFICATION_BATCH_SIZE)
-        print newest_notifications
         if len(newest_notifications) > 0:
             process_notifications(newest_notifications, process_notification)
         time.sleep(20)

diff --git a/mongo.py b/mongo.py
@@ -12,13 +12,13 @@
 
 @contextmanager
 def get_raw_email(object_id):
-    raw_email = raw_data.find_one({'_id':ObjectId(objectId)})
+    raw_email = raw_data.find_one({'_id':ObjectId(object_id)})
     if processed_data.find_one({'email_id':raw_email['id']}):
         raise Exception('dupe')
     del raw_email['_id']
     try:
         yield raw_email
-        raw_data.remove({'_id':ObjectId(objectId)})
+        raw_data.remove({'_id':ObjectId(object_id)})
     except Exception as e:
         raise e
 

diff --git a/pipeline.py b/pipeline.py
@@ -8,7 +8,7 @@ def process_email(raw_email):
     for extractor in extractors:
         results.update(extractor.extract(raw_email))
     return results
-    
+
 def process_notification(notification):
     object_id = notification['object_id']
     try:
@@ -17,4 +17,4 @@ def process_notification(notification):
             store_processed_data(processed_email)
         return processed_email
     except:
-        pass
+        raise