|
|
|
@ -137,7 +137,7 @@ class MongoWriter(Mongo,Writer):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
pass
|
|
|
|
|
def write(self,info):
|
|
|
|
|
def write(self,info,**_args):
|
|
|
|
|
"""
|
|
|
|
|
This function will write to a given collection i.e add a record to a collection (no updates)
|
|
|
|
|
@param info new record in the collection to be added
|
|
|
|
@ -148,13 +148,13 @@ class MongoWriter(Mongo,Writer):
|
|
|
|
|
# self.db[self.uid].insert_many(info)
|
|
|
|
|
# else:
|
|
|
|
|
try:
|
|
|
|
|
|
|
|
|
|
_uid = self.uid if 'doc' not in _args else _args['doc']
|
|
|
|
|
if self._lock :
|
|
|
|
|
Mongo.lock.acquire()
|
|
|
|
|
if type(info) == list or type(info) == pd.DataFrame :
|
|
|
|
|
self.db[self.uid].insert_many(info if type(info) == list else info.to_dict(orient='records'))
|
|
|
|
|
self.db[_uid].insert_many(info if type(info) == list else info.to_dict(orient='records'))
|
|
|
|
|
else:
|
|
|
|
|
self.db[self.uid].insert_one(info)
|
|
|
|
|
self.db[_uid].insert_one(info)
|
|
|
|
|
finally:
|
|
|
|
|
if self._lock :
|
|
|
|
|
Mongo.lock.release()
|
|
|
|
|