wraps database error in update_many.

Author: WaVEV

django_mongodb/compiler.py

@@ -608,18 +608,18 @@ def execute_sql(self, result_type):
         except EmptyResultSet:
             return 0
         is_empty = not bool(values)
-        rows = (
-            0
-            if is_empty
-            else self.collection.update_many(criteria, [{"$set": values}]).matched_count
-        )
+        rows = 0 if is_empty else self.update(criteria, [{"$set": values}])
         for query in self.query.get_related_updates():
             aux_rows = query.get_compiler(self.using).execute_sql(result_type)
             if is_empty and aux_rows:
                 rows = aux_rows
                 is_empty = False
         return rows
 
+    @wrap_database_errors
+    def update(self, criteria, pipeline):
+        return self.collection.update_many(criteria, pipeline).matched_count
+
     def get_where(self):
         return self.query.where