Merge pull request #263 from chang-xiao-feng/master

修复mongodb批量更新时,多个满足条件的数据只更新一条的问题.
Boris-code · Feb 14, 2025 · f56b369 · f56b369
2 parents 2825933 + ef1cf00
commit f56b369
Showing 1 changed file with 31 additions and 31 deletions.
diff --git a/feapder/db/mongodb.py b/feapder/db/mongodb.py
@@ -12,7 +12,7 @@
 from urllib import parse
 
 import pymongo
-from pymongo import MongoClient, UpdateOne
+from pymongo import MongoClient, UpdateOne, UpdateMany
 from pymongo.collection import Collection
 from pymongo.database import Database
 from pymongo.errors import DuplicateKeyError, BulkWriteError
@@ -23,14 +23,14 @@
 
 class MongoDB:
     def __init__(
-        self,
-        ip=None,
-        port=None,
-        db=None,
-        user_name=None,
-        user_pass=None,
-        url=None,
-        **kwargs,
+            self,
+            ip=None,
+            port=None,
+            db=None,
+            user_name=None,
+            user_pass=None,
+            url=None,
+            **kwargs,
     ):
         if not ip:
             ip = setting.MONGO_IP
@@ -97,7 +97,7 @@ def get_collection(self, coll_name, **kwargs) -> Collection:
         return self.db.get_collection(coll_name, **kwargs)
 
     def find(
-        self, coll_name: str, condition: Optional[Dict] = None, limit: int = 0, **kwargs
+            self, coll_name: str, condition: Optional[Dict] = None, limit: int = 0, **kwargs
     ) -> List[Dict]:
         """
         @summary:
@@ -136,13 +136,13 @@ def find(
         return dataset
 
     def add(
-        self,
-        coll_name,
-        data: Dict,
-        replace=False,
-        update_columns=(),
-        update_columns_value=(),
-        insert_ignore=False,
+            self,
+            coll_name,
+            data: Dict,
+            replace=False,
+            update_columns=(),
+            update_columns_value=(),
+            insert_ignore=False,
     ):
         """
         添加单条数据
@@ -198,13 +198,13 @@ def add(
         return affect_count
 
     def add_batch(
-        self,
-        coll_name: str,
-        datas: List[Dict],
-        replace=False,
-        update_columns=(),
-        update_columns_value=(),
-        condition_fields: dict = None,
+            self,
+            coll_name: str,
+            datas: List[Dict],
+            replace=False,
+            update_columns=(),
+            update_columns_value=(),
+            condition_fields: dict = None,
     ):
         """
         批量添加数据
@@ -362,11 +362,11 @@ def update_many(self, coll_name, data: Dict, condition: Dict, upsert: bool = Fal
             return True
 
     def update_batch(
-        self,
-        coll_name: str,
-        update_data_list: List[Dict],
-        condition_field: str,
-        upsert: bool = False,
+            self,
+            coll_name: str,
+            update_data_list: List[Dict],
+            condition_field: str,
+            upsert: bool = False,
     ):
         """
         批量更新数据
@@ -387,7 +387,7 @@ def update_batch(
 
         for update_data in update_data_list:
             condition = {condition_field: update_data.get(condition_field)}
-            update_operation = UpdateOne(
+            update_operation = UpdateMany(
                 condition, {"$set": update_data}, upsert=upsert
             )
             bulk_operations.append(update_operation)
@@ -468,7 +468,7 @@ def get_index_key(self, coll_name, index_name):
         return index_keys
 
     def __get_update_condition(
-        self, coll_name: str, data: dict, duplicate_errmsg: str
+            self, coll_name: str, data: dict, duplicate_errmsg: str
     ) -> dict:
         """
         根据索引冲突的报错信息 获取更新条件