jpt-archive/saucebrush
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add ability to drop_collection on MongoDBEmitter

Browse Source
This commit is contained in:
Jeremy Carbaugh 2009-07-02 18:01:53 -04:00
parent 4680cb8e3d
commit 2b1c2d9d26
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
saucebrush/emitters.py
View File

@ -171,11 +171,14 @@ class MongoDBEmitter(Emitter):
be inserted are required parameters. The host and port are optional,
defaulting to 'localhost' and 27017, repectively.
"""
def __init__(self, database, collection, host='localhost', port=27017):
def __init__(self, database, collection, host='localhost', port=27017, drop_collection=False):
super(MongoDBEmitter, self).__init__()
from pymongo.connection import Connection
conn = Connection(host, port)
self.collection = conn[database][collection]
db = conn[database]
if drop_collection:
db.drop_collection(collection)
self.collection = db[collection]
def emit_record(self, record):
self.collection.insert(record)