diff --git a/python_code/mailboxes__load_json_mbox.py b/python_code/mailboxes__load_json_mbox.py index 30b1e3d..3928bd3 100644 --- a/python_code/mailboxes__load_json_mbox.py +++ b/python_code/mailboxes__load_json_mbox.py @@ -2,16 +2,23 @@ import sys import os -import couchdb +import pymongo +from pymongo import Connection + try: import jsonlib2 as json except ImportError: import json JSON_MBOX = sys.argv[1] # i.e. enron.mbox.json -DB = os.path.basename(JSON_MBOX).split('.')[0] +DB_NAME = os.path.basename(JSON_MBOX).split('.')[0] + +connection = Connection('localhost', 27017) +db = connection[DB_NAME] -server = couchdb.Server('http://localhost:5984') -db = server.create(DB) docs = json.loads(open(JSON_MBOX).read()) -db.update(docs, all_or_nothing=True) + +collection = db['messages'] +for doc in docs: + print str(doc) + collection.insert(doc) \ No newline at end of file