pre commit

Author: idocyabra

mongoengine/queryset/base.py

@@ -567,8 +567,8 @@ def update(
                 for u in update["__raw__"]
             ]
         else:
-            if 'array_filters' in update:
-                array_filters = update.pop('array_filters')
+            if "array_filters" in update:
+                array_filters = update.pop("array_filters")
             update = transform.update(queryset._document, **update)
         # If doing an atomic upsert on an inheritable class
         # then ensure we add _cls to the update operation
@@ -584,7 +584,9 @@ def update(
                 update_func = collection.update_one
                 if multi:
                     update_func = collection.update_many
-                result = update_func(query, update, upsert=upsert, array_filters=array_filters)
+                result = update_func(
+                    query, update, upsert=upsert, array_filters=array_filters
+                )
             if full_result:
                 return result
             elif result.raw_result:

tests/queryset/test_queryset.py

@@ -592,8 +592,7 @@ class Blog(Document):
         Blog.drop_collection()
 
     def test_update_array_filters(self):
-        """Ensure that updating by array_filters works.
-        """
+        """Ensure that updating by array_filters works."""
 
         class Comment(EmbeddedDocument):
             comment_tags = ListField(StringField())
@@ -605,48 +604,48 @@ class Blog(Document):
         Blog.drop_collection()
 
         # update one
-        Blog.objects.create(tags=['test1', 'test2', 'test3'])
+        Blog.objects.create(tags=["test1", "test2", "test3"])
 
-        Blog.objects().update_one(__raw__=
-                                  {'$set': {"tags.$[element]": 'test11111'}},
-                                  array_filters=[{"element": {'$eq': 'test2'}}],
-                                  )
+        Blog.objects().update_one(
+            __raw__={"$set": {"tags.$[element]": "test11111"}},
+            array_filters=[{"element": {"$eq": "test2"}}],
+        )
         testc_blogs = Blog.objects(tags="test11111")
 
         assert testc_blogs.count() == 1
 
         Blog.drop_collection()
 
         # update one inner list
-        comments = Comment(comment_tags=['test1', 'test2', 'test3'])
+        comments = Comment(comment_tags=["test1", "test2", "test3"])
         Blog.objects.create(comments=comments)
 
-        Blog.objects().update_one(__raw__=
-                                  {'$set': {"comments.comment_tags.$[element]": 'test11111'}},
-                                  array_filters=[{"element": {'$eq': 'test2'}}],
-                                  )
+        Blog.objects().update_one(
+            __raw__={"$set": {"comments.comment_tags.$[element]": "test11111"}},
+            array_filters=[{"element": {"$eq": "test2"}}],
+        )
         testc_blogs = Blog.objects(comments__comment_tags="test11111")
 
         assert testc_blogs.count() == 1
 
         # update many
         Blog.drop_collection()
 
-        Blog.objects.create(tags=['test1', 'test2', 'test3', 'test_all'])
-        Blog.objects.create(tags=['test4', 'test5', 'test6', 'test_all'])
+        Blog.objects.create(tags=["test1", "test2", "test3", "test_all"])
+        Blog.objects.create(tags=["test4", "test5", "test6", "test_all"])
 
-        Blog.objects().update(__raw__=
-                              {'$set': {"tags.$[element]": 'test11111'}},
-                              array_filters=[{"element": {'$eq': 'test2'}}],
-                              )
+        Blog.objects().update(
+            __raw__={"$set": {"tags.$[element]": "test11111"}},
+            array_filters=[{"element": {"$eq": "test2"}}],
+        )
         testc_blogs = Blog.objects(tags="test11111")
 
         assert testc_blogs.count() == 1
 
-        Blog.objects().update(__raw__=
-                              {'$set': {"tags.$[element]": 'test_all1234577'}},
-                              array_filters=[{"element": {'$eq': 'test_all'}}],
-                              )
+        Blog.objects().update(
+            __raw__={"$set": {"tags.$[element]": "test_all1234577"}},
+            array_filters=[{"element": {"$eq": "test_all"}}],
+        )
         testc_blogs = Blog.objects(tags="test_all1234577")
 
         assert testc_blogs.count() == 2