improve relation_proxy types

2021-03-19 17:13:59 +01:00
parent 929e979d37
commit 9c091afe35
6 changed files with 64 additions and 54 deletions
--- a/ormar/relations/relation.py
+++ b/ormar/relations/relation.py
@ -63,7 +63,7 @@ class Relation:
        self._through = through
        self.field_name: str = field_name
        self.related_models: Optional[Union[RelationProxy, "Model"]] = (
-            RelationProxy(relation=self, type_=type_, field_name=field_name)
+            RelationProxy(relation=self, type_=type_, to=to, field_name=field_name)
            if type_ in (RelationType.REVERSE, RelationType.MULTIPLE)
            else None
        )
@ -94,6 +94,7 @@ class Relation:
        self.related_models = RelationProxy(
            relation=self,
            type_=self._type,
+            to=self.to,
            field_name=self.field_name,
            data_=cleaned_data,
        )