fix for issue 73

2020-12-14 15:36:04 +01:00
parent ef4b687957
commit 6e67b69385
15 changed files with 306 additions and 80 deletions
--- a/ormar/relations/relation_manager.py
+++ b/ormar/relations/relation_manager.py
@ -56,8 +56,14 @@ class RelationsManager:
        return None

    @staticmethod
-    def add(parent: "Model", child: "Model", child_name: str, virtual: bool) -> None:
-        to_field: Type[BaseField] = child.resolve_relation_field(child, parent)
+    def add(
+        parent: "Model",
+        child: "Model",
+        child_name: str,
+        virtual: bool,
+        relation_name: str,
+    ) -> None:
+        to_field: Type[BaseField] = child.Meta.model_fields[relation_name]

        (parent, child, child_name, to_name,) = get_relations_sides_and_names(
            to_field, parent, child, child_name, virtual