Refactor Encoder

BeanieODM · Sep 17, 2023 · 28712de · 28712de
1 parent 60dc39c
commit 28712de
Show file tree

Hide file tree

Showing 4 changed files with 166 additions and 280 deletions.
diff --git a/beanie/odm/queries/find.py b/beanie/odm/queries/find.py
@@ -940,9 +940,7 @@ async def replace_one(
                 Operation(
                     operation=ReplaceOne,
                     first_query=self.get_filter_query(),
-                    second_query=Encoder(
-                        by_alias=True, exclude={"_id"}
-                    ).encode(document),
+                    second_query=Encoder(exclude={"_id"}).encode(document),
                     object_class=self.document_model,
                     pymongo_kwargs=self.pymongo_kwargs,
                 )

diff --git a/beanie/odm/utils/dump.py b/beanie/odm/utils/dump.py
@@ -18,9 +18,8 @@ def get_dict(
         exclude.add("_id")
     if not document.get_settings().use_revision:
         exclude.add("revision_id")
-    return Encoder(
-        by_alias=True, exclude=exclude, to_db=to_db, keep_nulls=keep_nulls
-    ).encode(document)
+    encoder = Encoder(exclude=exclude, to_db=to_db, keep_nulls=keep_nulls)
+    return encoder.encode(document)
 
 
 def get_nulls(