Skip to content

Commit a59d1f4

Browse files
committed
move pymongo patch code to the bottom of the file
1 parent 2ffd105 commit a59d1f4

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

aikido_zen/sinks/pymongo.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -79,13 +79,6 @@ def _bulk_write(func, instance, args, kwargs):
7979
)
8080

8181

82-
@on_import("pymongo", "pymongo", version_requirement="3.10.0")
83-
def patch(m):
84-
on_import("pymongo.collection")(patch_collection)
85-
# After 4.9, pymongo moved the collection module to `synchronous`
86-
on_import("pymongo.synchronous.collection")(patch_collection)
87-
88-
8982
def patch_collection(m):
9083
"""
9184
patching pymongo.collection
@@ -127,3 +120,10 @@ def patch_collection(m):
127120

128121
# bulk_write
129122
patch_function(m, "Collection.bulk_write", _bulk_write)
123+
124+
125+
@on_import("pymongo", "pymongo", version_requirement="3.10.0")
126+
def patch(m):
127+
on_import("pymongo.collection")(patch_collection)
128+
# After 4.9, pymongo moved the collection module to `synchronous`
129+
on_import("pymongo.synchronous.collection")(patch_collection)

0 commit comments

Comments
 (0)