enhancement mongodb

This commit is contained in:
Steve Nyemba 2022-04-29 11:15:32 -05:00
parent e9a6779a9b
commit fd9523e99a
2 changed files with 5 additions and 5 deletions

View File

@ -8,7 +8,7 @@ def read(fname):
return open(os.path.join(os.path.dirname(__file__), fname)).read()
args = {
"name":"data-transport",
"version":"1.4.8",
"version":"1.4.10",
"author":"The Phi Technology LLC","author_email":"info@the-phi.com",
"license":"MIT",
"packages":["transport"]}

View File

@ -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()