Prodream/ormar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cache visited relations in iterating_models

Browse Source
This commit is contained in:
collerek
2021-09-11 11:18:52 +02:00
parent beb43dd76c
commit e5e0ff7b78
1 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
ormar/models/mixins/relation_mixin.py
View File

@ -1,5 +1,6 @@
from typing import (
Callable,
Dict,
List,
Optional,
Set,
@ -123,6 +124,7 @@ class RelationMixin:
def _iterate_related_models( # noqa: CCR001
cls,
node_list: NodeList = None,
parsed_map: Dict = None,
source_relation: str = None,
recurrent: bool = False,
) -> List[str]:
@ -137,11 +139,12 @@ class RelationMixin:
if cls.__relation_map__:
return cls.__relation_map__
node_list = NodeList()
parsed_map = dict()
current_node = node_list.add(node_class=cls)
else:
current_node = node_list[-1]
relations = cls.extract_related_names()
processed_relations = []
relations = sorted(cls.extract_related_names())
processed_relations: List[str] = []
for relation in relations:
if not current_node.visited(relation):
target_model = cls.Meta.model_fields[relation].to
@ -150,9 +153,17 @@ class RelationMixin:
relation_name=relation,
parent_node=current_node,
)
deep_relations = target_model._iterate_related_models(
source_relation=relation, node_list=node_list, recurrent=True
)
relation_key = f"{cls.get_name()}_{relation}"
parsed_map = cast(Dict, parsed_map)
deep_relations = parsed_map.get(relation_key)
if not deep_relations:
deep_relations = target_model._iterate_related_models(
source_relation=relation,
node_list=node_list,
recurrent=True,
parsed_map=parsed_map,
)
parsed_map[relation_key] = deep_relations
processed_relations.extend(deep_relations)
result = cls._get_final_relations(processed_relations, source_relation)