hiplab/data-transport
hiplab
/
data-transport
mirror of https://github.com/lnyemba/data-transport
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

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

8
transport/mongo.py
View File

@ -137,7 +137,7 @@ class MongoWriter(Mongo,Writer):
pass 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) 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 @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) # self.db[self.uid].insert_many(info)
# else: # else:
try: try:
_uid = self.uid if 'doc' not in _args else _args['doc']
if self._lock : if self._lock :
Mongo.lock.acquire() Mongo.lock.acquire()
if type(info) == list or type(info) == pd.DataFrame : 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: else:
self.db[self.uid].insert_one(info) self.db[_uid].insert_one(info)
finally: finally:
if self._lock : if self._lock :
Mongo.lock.release() Mongo.lock.release()