fixed in general
This commit is contained in:
@ -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,
|
||||||
|
|||||||
@ -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)
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
@ -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(
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
@ -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)
|
||||||
|
|||||||
@ -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"]:
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user