浏览代码

bug fix with mongodb client

Steve Nyemba 4 年之前
父节点
当前提交
4b49b0a060
共有 2 个文件被更改,包括 4 次插入2 次删除
  1. 1 1
      setup.py
  2. 3 1
      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.2.6",
+    "version":"1.2.8",
     "author":"The Phi Technology LLC","author_email":"info@the-phi.com",
     "license":"MIT",
     "packages":["transport"]}

+ 3 - 1
transport/mongo.py

@@ -48,7 +48,7 @@ class Mongo :
         q = self.uid in self.client[self.dbname].list_collection_names()
         return p and q
     def close(self):
-        self.db.close()
+        self.client.close()
 
 class MongoReader(Mongo,Reader):
     """
@@ -158,5 +158,7 @@ class MongoWriter(Mongo,Writer):
         else:
             collection.delete_many({})
             self.write(info)
+    def close(self):
+        Mongo.close(self)
         # collecton.update_one({"_id":self.uid},document,True)