1 files changed, 83 insertions, 0 deletions
diff --git a/lib/mongo.py b/lib/mongo.py
new file mode 100644
index 0000000..b8f3b2f
--- /dev/null
+++ b/lib/mongo.py
@@ -0,0 +1,83 @@
+#-*- coding: utf-8 -*-
+
+import pymongo
+from bunch import Bunch
+
+connection = pymongo.Connection('localhost', 27017)
+
+def get_emails_thread(table, start_email, thread):
+    db = connection[table]
+    db.mails.create_index('Date')
+    db.mails.ensure_index('Date')
+    db.mails.create_index('In-Reply-To')
+    db.mails.ensure_index('In-Reply-To')
+    db.mails.create_index('Message-ID')
+    db.mails.ensure_index('Message-ID')
+    for el in db.mails.find({'In-Reply-To': start_email['Message-ID']},
+        sort=[('Date', pymongo.DESCENDING)]):
+        thread.append(el)
+        get_emails_thread(el, thread)
+    return thread
+
+
+def get_archives(table, start, end):
+    db = connection[table]
+    db.mails.create_index('Date')
+    db.mails.ensure_index('Date')
+    db.mails.create_index('In-Reply-To')
+    db.mails.ensure_index('In-Reply-To')
+    # Beginning of thread == No 'In-Reply-To' header
+    archives = []
+    for el in db.mails.find({'In-Reply-To': {'$exists':False},
+            "Date": {"$gte": start, "$lt": end}}, 
+            sort=[('Date', pymongo.DESCENDING)]):
+        archives.append(el)
+    return archives
+
+
+def get_thread(table, start, end):
+    db = connection[table]
+    db.mails.create_index('Date')
+    db.mails.ensure_index('Date')
+    db.mails.create_index('In-Reply-To')
+    db.mails.ensure_index('In-Reply-To')
+    # Beginning of thread == No 'In-Reply-To' header
+    archives = Bunch()
+    for el in db.mails.find({'In-Reply-To': {'$exists':False},
+            "Date": {"$gte": start, "$lt": end}}, 
+            sort=[('Date', pymongo.DESCENDING)]):
+        thread = get_emails_thread(el, [el])
+        #print el['Subject'], len(thread)
+        archives[el['Subject']] = thread
+    return archives
+
+
+def get_thread_length(table, thread_id):
+    db = connection[table]
+    db.mails.create_index('Thread-ID')
+    db.mails.ensure_index('Thread-ID')
+    return db.mails.find({'Thread-ID': thread_id}).count()
+
+
+def get_thread_participants(table, thread_id):
+    db = connection[table]
+    db.mails.create_index('Thread-ID')
+    db.mails.ensure_index('Thread-ID')
+    authors = set()
+    for mail in db.mails.find({'Thread-ID': thread_id}):
+        authors.add(mail['From'])
+    return len(authors)
+
+def get_archives_length(table):
+    db = connection[table]
+    archives = {}
+    for entry in db.mails.find():
+        date = entry['Date']
+        if date.year in archives:
+            archives[date.year].add(date.month)
+        else:
+            archives[date.year] = set([date.month])
+    for key in archives:
+        archives[key] = list(archives[key])
+    return archives
+