enhancement mongodb

pull/1/head
Steve Nyemba 3 years ago
parent e9a6779a9b
commit fd9523e99a

@ -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"]}

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

Loading…
Cancel
Save