瀏覽代碼

bug fix with mongodb

Steve Nyemba 6 年之前
父節點
當前提交
008ae6edf9
共有 1 個文件被更改,包括 4 次插入2 次删除
  1. 4 2
      transport/mongo.py

+ 4 - 2
transport/mongo.py

@@ -65,8 +65,10 @@ class MongoWriter(Mongo,Writer):
     def write(self,**args):
         # document  = self.db[self.uid].find()
         collection = self.db[self.uid]
-        collection.update_one()
-        self.db[self.uid].insert_one(args['row'])
+        if type(args['row']) == list :
+            self.db[self.uid].insert_many(args['row'])
+        else:
+            self.db[self.uid].insert_one(args['row'])
     def set(self,document):
         collection = self.db[self.uid]
         if collection.count_document() > 0 :