Browse Source

enhancement mongodb

Steve Nyemba 3 năm trước cách đây
mục cha
commit
fd9523e99a
2 tập tin đã thay đổi với 5 bổ sung5 xóa
  1. 1 1
      setup.py
  2. 4 4
      transport/mongo.py

+ 1 - 1
setup.py

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

+ 4 - 4
transport/mongo.py

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