Sfoglia il codice sorgente

bug fix with mongodb

Steve Nyemba 6 anni fa
parent
commit
008ae6edf9
1 ha cambiato i file con 4 aggiunte e 2 eliminazioni
  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 :