Cache the relation map dictionary

collerek · May 3, 2023 · aa0dbe3 · aa0dbe3
1 parent bcc3af0
commit aa0dbe3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/ormar/models/helpers/models.py b/ormar/models/helpers/models.py
@@ -78,6 +78,7 @@ def populate_default_options_values(  # noqa: CCR001
     }
 
     new_model.__relation_map__ = None
+    new_model.__relation_map_dict__ = None
 
 
 class Connection(sqlite3.Connection):

diff --git a/ormar/models/newbasemodel.py b/ormar/models/newbasemodel.py
@@ -87,6 +87,7 @@ class NewBaseModel(pydantic.BaseModel, ModelTableProxy, metaclass=ModelMetaclass
         __metadata__: sqlalchemy.MetaData
         __database__: databases.Database
         __relation_map__: Optional[List[str]]
+        __relation_map_dict__: Optional[dict[str, Any]]
         __cached_hash__: Optional[int]
         _orm_relationship_manager: AliasManager
         _orm: RelationsManager
@@ -734,6 +735,16 @@ def _extract_nested_models(  # noqa: CCR001, CFQ002
             except ReferenceError:
                 dict_instance[field] = None
         return dict_instance
+
+
+    @classmethod
+    def _related_models_dict(cls) -> dict:
+        if not cls.__relation_map_dict__:
+            cls.__relation_map_dict__ = translate_list_to_dict(
+                cls._iterate_related_models()
+                )
+        return cls.__relation_map_dict__
+
 
     def dict(  # type: ignore # noqa A003
         self,
@@ -813,7 +824,7 @@ def dict(  # type: ignore # noqa A003
         relation_map = (
             relation_map
             if relation_map is not None
-            else translate_list_to_dict(self._iterate_related_models())
+            else self._related_models_dict()
         )
         pk_only = getattr(self, "__pk_only__", False)
         if relation_map and not pk_only:
-Original file line number
+Diff line change
@@ Expand Up / @@ -78,6 +78,7 @@ def populate_default_options_values( # noqa: CCR001 @@
         }
         new_model.__relation_map__ = None
+        new_model.__relation_map_dict__ = None
     class Connection(sqlite3.Connection):
@@ Expand Down @@