From f81cce73e8218c3981146675dece49e145172d52 Mon Sep 17 00:00:00 2001 From: everysoftware Date: Sat, 21 Dec 2024 19:17:15 +0300 Subject: [PATCH] Delete Identity Map --- src/enterprise/unit_of_work.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/enterprise/unit_of_work.py b/src/enterprise/unit_of_work.py index 7b5df52..da80e4a 100644 --- a/src/enterprise/unit_of_work.py +++ b/src/enterprise/unit_of_work.py @@ -59,7 +59,6 @@ def __init__(self, source: DataSource) -> None: self._new: list[Entity] = [] self._dirty: list[Entity] = [] self._removed: list[Entity] = [] - self._all: dict[str, dict[int, Entity]] = {} @property def source(self) -> DataSource: @@ -76,7 +75,6 @@ def get[T: Entity](self, entity_type: type[T], id: int) -> T | None: if data is None: return None entity = entity_type(**data) # noqa - self._all.setdefault(entity_type.__name__, {})[id] = entity return entity def update(self, model: Entity) -> None: @@ -108,6 +106,7 @@ def commit(self) -> None: def rollback(self) -> None: self._source.rollback() + self._identity_map.clear() self._new.clear() self._dirty.clear() self._removed.clear()