47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING: # pragma no cover
|
|
from orm.models import Model
|
|
|
|
|
|
class RelationshipManager:
|
|
|
|
def __init__(self):
|
|
self._relations = dict()
|
|
|
|
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_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 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, 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__()
|