fixed in general

This commit is contained in:
collerek
2020-12-15 14:11:46 +01:00
parent 1b42d321b9
commit daf47f891e
9 changed files with 71 additions and 27 deletions

View File

@ -164,7 +164,6 @@ class ForeignKeyField(BaseField):
) -> Optional[Union["Model", List["Model"]]]: ) -> Optional[Union["Model", List["Model"]]]:
if value is None: if value is None:
return None if not cls.virtual else [] return None if not cls.virtual else []
print('expanding', relation_name)
constructors = { constructors = {
f"{cls.to.__name__}": cls._register_existing_model, f"{cls.to.__name__}": cls._register_existing_model,
"dict": cls._construct_model_from_dict, "dict": cls._construct_model_from_dict,

View File

@ -89,18 +89,28 @@ def register_reverse_model_fields(
) -> None: ) -> None:
if issubclass(model_field, ManyToManyField): if issubclass(model_field, ManyToManyField):
model.Meta.model_fields[child_model_name] = ManyToMany( model.Meta.model_fields[child_model_name] = ManyToMany(
child, through=model_field.through, name=child_model_name, virtual=True, related_name=model_field.name child,
through=model_field.through,
name=child_model_name,
virtual=True,
related_name=model_field.name,
) )
# register foreign keys on through model # register foreign keys on through model
adjust_through_many_to_many_model(model, child, model_field) adjust_through_many_to_many_model(model, child, model_field, child_model_name)
else: else:
model.Meta.model_fields[child_model_name] = ForeignKey( model.Meta.model_fields[child_model_name] = ForeignKey(
child, real_name=child_model_name, virtual=True, related_name=model_field.name child,
real_name=child_model_name,
virtual=True,
related_name=model_field.name,
) )
def adjust_through_many_to_many_model( def adjust_through_many_to_many_model(
model: Type["Model"], child: Type["Model"], model_field: Type[ManyToManyField] model: Type["Model"],
child: Type["Model"],
model_field: Type[ManyToManyField],
child_model_name: str,
) -> None: ) -> None:
model_field.through.Meta.model_fields[model.get_name()] = ForeignKey( model_field.through.Meta.model_fields[model.get_name()] = ForeignKey(
model, real_name=model.get_name(), ondelete="CASCADE" model, real_name=model.get_name(), ondelete="CASCADE"
@ -148,6 +158,7 @@ def create_and_append_m2m_fk(
model.Meta.tablename + "." + model.get_column_alias(model.Meta.pkname), model.Meta.tablename + "." + model.get_column_alias(model.Meta.pkname),
ondelete="CASCADE", ondelete="CASCADE",
onupdate="CASCADE", onupdate="CASCADE",
related_name=model_field.name,
), ),
) )
model_field.through.Meta.columns.append(column) model_field.through.Meta.columns.append(column)

View File

@ -62,11 +62,17 @@ class ModelTableProxy:
@staticmethod @staticmethod
def get_clause_target_and_filter_column_name( def get_clause_target_and_filter_column_name(
parent_model: Type["Model"], target_model: Type["Model"], reverse: bool, related: str, parent_model: Type["Model"],
target_model: Type["Model"],
reverse: bool,
related: str,
) -> Tuple[Type["Model"], str]: ) -> Tuple[Type["Model"], str]:
if reverse: if reverse:
field = parent_model.Meta.model_fields[related] field_name = (
# field = target_model.resolve_relation_field(target_model, parent_model) parent_model.Meta.model_fields[related].related_name
or parent_model.get_name() + "s"
)
field = target_model.Meta.model_fields[field_name]
if issubclass(field, ormar.fields.ManyToManyField): if issubclass(field, ormar.fields.ManyToManyField):
sub_field = target_model.resolve_relation_field( sub_field = target_model.resolve_relation_field(
field.through, parent_model field.through, parent_model

View File

@ -262,9 +262,16 @@ class SqlJoin:
model_cls: Type["Model"], model_cls: Type["Model"],
part: str, part: str,
) -> Tuple[str, str]: ) -> Tuple[str, str]:
if join_params.prev_model.Meta.model_fields[part].virtual or is_multi: if is_multi:
to_field = model_cls.resolve_relation_name( to_field = join_params.prev_model.get_name()
model_cls, join_params.prev_model to_key = model_cls.get_column_alias(to_field)
from_key = join_params.prev_model.get_column_alias(
join_params.prev_model.Meta.pkname
)
elif join_params.prev_model.Meta.model_fields[part].virtual:
to_field = (
join_params.prev_model.Meta.model_fields[part].related_name
or join_params.prev_model.get_name() + "s"
) )
to_key = model_cls.get_column_alias(to_field) to_key = model_cls.get_column_alias(to_field)
from_key = join_params.prev_model.get_column_alias( from_key = join_params.prev_model.get_column_alias(

View File

@ -145,7 +145,11 @@ class PrefetchQuery:
) )
def _get_filter_for_prefetch( def _get_filter_for_prefetch(
self, parent_model: Type["Model"], target_model: Type["Model"], reverse: bool, related: str, self,
parent_model: Type["Model"],
target_model: Type["Model"],
reverse: bool,
related: str,
) -> List: ) -> List:
ids = self._extract_required_ids( ids = self._extract_required_ids(
parent_model=parent_model, target_model=target_model, reverse=reverse, parent_model=parent_model, target_model=target_model, reverse=reverse,
@ -155,7 +159,10 @@ class PrefetchQuery:
clause_target, clause_target,
filter_column, filter_column,
) = parent_model.get_clause_target_and_filter_column_name( ) = parent_model.get_clause_target_and_filter_column_name(
parent_model=parent_model, target_model=target_model, reverse=reverse, related=related parent_model=parent_model,
target_model=target_model,
reverse=reverse,
related=related,
) )
qryclause = QueryClause( qryclause = QueryClause(
model_cls=clause_target, select_related=[], filter_clauses=[], model_cls=clause_target, select_related=[], filter_clauses=[],
@ -246,7 +253,10 @@ class PrefetchQuery:
parent_model = target_model parent_model = target_model
filter_clauses = self._get_filter_for_prefetch( filter_clauses = self._get_filter_for_prefetch(
parent_model=parent_model, target_model=target_field.to, reverse=reverse, related=related parent_model=parent_model,
target_model=target_field.to,
reverse=reverse,
related=related,
) )
if not filter_clauses: # related field is empty if not filter_clauses: # related field is empty
return return

View File

@ -49,7 +49,10 @@ class Relation:
if i not in self._to_remove if i not in self._to_remove
] ]
self.related_models = RelationProxy( self.related_models = RelationProxy(
relation=self, type_=self._type, field_name=self.field_name, data_=cleaned_data relation=self,
type_=self._type,
field_name=self.field_name,
data_=cleaned_data,
) )
relation_name = self._owner.resolve_relation_name(self._owner, self.to) relation_name = self._owner.resolve_relation_name(self._owner, self.to)
self._owner.__dict__[relation_name] = cleaned_data self._owner.__dict__[relation_name] = cleaned_data

View File

@ -65,17 +65,17 @@ class RelationsManager:
relation_name: str, relation_name: str,
) -> None: ) -> None:
to_field: Type[BaseField] = child.Meta.model_fields[relation_name] to_field: Type[BaseField] = child.Meta.model_fields[relation_name]
print('comming', child_name, relation_name) # print('comming', child_name, relation_name)
(parent, child, child_name, to_name,) = get_relations_sides_and_names( (parent, child, child_name, to_name,) = get_relations_sides_and_names(
to_field, parent, child, child_name, virtual to_field, parent, child, child_name, virtual
) )
print('adding', parent.get_name(), child.get_name(), child_name) # print('adding', parent.get_name(), child.get_name(), child_name)
parent_relation = parent._orm._get(child_name) parent_relation = parent._orm._get(child_name)
if parent_relation: if parent_relation:
parent_relation.add(child) # type: ignore parent_relation.add(child) # type: ignore
print('adding', child.get_name(), parent.get_name(), child_name) # print('adding', child.get_name(), parent.get_name(), child_name)
child_relation = child._orm._get(to_name) child_relation = child._orm._get(to_name)
if child_relation: if child_relation:
child_relation.add(parent) child_relation.add(parent)

View File

@ -1,4 +1,4 @@
from typing import Any, TYPE_CHECKING from typing import Any, Optional, TYPE_CHECKING
import ormar import ormar
from ormar.exceptions import NoMatch, RelationshipInstanceError from ormar.exceptions import NoMatch, RelationshipInstanceError
@ -12,7 +12,11 @@ if TYPE_CHECKING: # pragma no cover
class RelationProxy(list): class RelationProxy(list):
def __init__( def __init__(
self, relation: "Relation", type_: "RelationType", field_name: str, data_: Any = None self,
relation: "Relation",
type_: "RelationType",
field_name: str,
data_: Any = None,
) -> None: ) -> None:
super().__init__(data_ or ()) super().__init__(data_ or ())
self.relation: "Relation" = relation self.relation: "Relation" = relation
@ -20,8 +24,17 @@ class RelationProxy(list):
self.field_name = field_name self.field_name = field_name
self._owner: "Model" = self.relation.manager.owner self._owner: "Model" = self.relation.manager.owner
self.queryset_proxy = QuerysetProxy(relation=self.relation, type_=type_) self.queryset_proxy = QuerysetProxy(relation=self.relation, type_=type_)
self._related_field_name: Optional[str] = None
@property
def related_field_name(self) -> str:
if self._related_field_name:
return self._related_field_name
owner_field = self._owner.Meta.model_fields[self.field_name] owner_field = self._owner.Meta.model_fields[self.field_name]
self.related_field_name = owner_field.related_name or self._owner.get_name() + 's' self._related_field_name = (
owner_field.related_name or self._owner.get_name() + "s"
)
return self._related_field_name
def __getattribute__(self, item: str) -> Any: def __getattribute__(self, item: str) -> Any:
if item in ["count", "clear"]: if item in ["count", "clear"]:

View File

@ -94,11 +94,6 @@ async def test_add_students():
assert user.attending is not None assert user.attending is not None
assert len(user.attending) > 0 assert len(user.attending) > 0
query = Session.objects.prefetch_related( query = Session.objects.prefetch_related(["students", "teacher",])
[
"students",
"teacher",
]
)
sessions = await query.all() sessions = await query.all()
assert len(sessions) == 5 assert len(sessions) == 5