fix ps per batch

gisce · Jun 14, 2024 · bbe4bb3 · bbe4bb3
1 parent 7478682
commit bbe4bb3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sippers/backends/mongodb.py b/sippers/backends/mongodb.py
@@ -42,8 +42,8 @@ def insert(self, document):
             if batch_insert:
                 ps_data = []
                 for x in docuemnt_list:
-                    x.get('measure_cnmc').backend = self
-                    ps_data.append(x.get('measure_cnmc').backend_data)
+                    x.get('ps').backend = self
+                    ps_data.append(x.get('ps').backend_data)
             else:
                 ps.backend = self
                 ps_data = ps.backend_data
@@ -81,6 +81,7 @@ def insert_ps(self, ps, collection=None):
             key = 'cups'
 
         if isinstance(ps, list):
+            oid = False
             for doc in ps:
                 oid = self.db[collection].update(
                     {key: doc[key]}, doc, upsert=True