rebuild the registry of relationships

2020-08-05 18:32:13 +02:00
parent a371c48959
commit 475dafb6c9
8 changed files with 204 additions and 81 deletions
--- a/orm/relations.py
+++ b/orm/relations.py
@ -1,44 +1,46 @@
-from typing import Dict, Union, List
+from typing import TYPE_CHECKING

-from orm.exceptions import RelationshipNotFound
-
-
-class Relationship:
-
-    def __init__(self, name: str, parent: 'Model', child: 'Model', fk_side: str = 'child'):
-        self.fk_side = fk_side
-        self.child = child
-        self.parent = parent
-        self.name = name
+if TYPE_CHECKING:  # pragma no cover
+    from orm.models import Model


 class RelationshipManager:

-    def __init__(self, model: 'Model'):
-        self._orm_id: str = model._orm_id
-        self._relations: Dict[str, Union[Relationship, List[Relationship]]] = dict()
+    def __init__(self):
+        self._relations = dict()

-    def __contains__(self, item):
-        return item in self._relations
+    def add_relation_type(self, relations_key, reverse_key):
+        print(relations_key, reverse_key)
+        if relations_key not in self._relations:
+            self._relations[relations_key] = {'type': 'primary'}
+        if reverse_key not in self._relations:
+            self._relations[reverse_key] = {'type': 'reverse'}

-    def add_related(self, relation: Relationship):
-        if relation.fk_side == 'child' and relation.parent._orm_id == self._orm_id:
-            new_relation = Relationship(name=relation.parent.__class__.__name__.lower(),
-                                        child=relation.parent,
-                                        parent=relation.child,
-                                        fk_side='parent')
-            relation.child._orm_relationship_manager.add(new_relation)
+    def add_relation(self, parent_name: str, child_name: str, parent: 'Model', child: 'Model', virtual: bool = False):
+        parent_id = parent._orm_id
+        child_id = child._orm_id
+        if virtual:
+            child_name, parent_name = parent_name, child_name
+            child_id, parent_id = parent_id, child_id
+            child, parent = parent, child
+        self._relations[parent_name.title() + '_' + child_name + 's'].setdefault(parent_id, []).append(
+            child)
+        self._relations[child_name.title() + '_' + parent_name].setdefault(child_id, []).append(parent)

-    def add(self, relation: Relationship):
-        if relation.name in self._relations:
-            self._relations[relation.name].append(relation)
-        else:
-            self._relations[relation.name] = [relation]
-        self.add_related(relation)
+    def contains(self, relations_key: str, object: 'Model'):
+        if relations_key in self._relations:
+            return object._orm_id in self._relations[relations_key]
+        return False

-    def get(self, name: str):
-        for rel, relations in self._relations.items():
-            if rel == name:
-                if relations and relations[0].fk_side == 'parent':
-                    return relations[0].child
-                return [rela.child for rela in relations]
+    def get(self, relations_key: str, object: 'Model'):
+        if relations_key in self._relations:
+            if object._orm_id in self._relations[relations_key]:
+                if self._relations[relations_key]['type'] == 'primary':
+                    return self._relations[relations_key][object._orm_id][0]
+                return self._relations[relations_key][object._orm_id]
+
+    def __str__(self):  # pragma no cover
+        return ''.join(self._relations[rel].__str__() for rel in self._relations)
+
+    def __repr__(self):  # pragma no cover
+        return self.__str__()