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, be inserted are required parameters. The host and port are optional,
defaulting to 'localhost' and 27017, repectively. 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__() super(MongoDBEmitter, self).__init__()
from pymongo.connection import Connection from pymongo.connection import Connection
conn = Connection(host, port) 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): def emit_record(self, record):
self.collection.insert(record) self.collection.insert(record)