Add Session Property

pymfdata/rdb/repository.py

@@ -12,7 +12,7 @@
 
 
 class AsyncRepository(Protocol[_MT, _T]):
-    session: AsyncSession
+    _session: AsyncSession
 
     @property
     def _model(self):
@@ -22,6 +22,11 @@ def _model(self):
     def _pk_column(self) -> str:
         return inspect(self._model).primary_key[0].name
 
+    @property
+    def session(self) -> AsyncSession:
+        assert self._session is not None
+        return self._session
+
     async def delete(self, item: _MT):
         await self.session.delete(item)
 
@@ -68,7 +73,7 @@ def update(self, item: _MT, req: dict):
 
 
 class SyncRepository(Protocol[_MT, _T]):
-    session: Session
+    _session: Session
 
     @property
     def _model(self):
@@ -78,6 +83,11 @@ def _model(self):
     def _pk_column(self) -> str:
         return inspect(self._model).primary_key[0].name
 
+    @property
+    def session(self) -> Session:
+        assert self._session is not None
+        return self._session
+
     @final
     def count(self, **kwargs) -> int:
         return self._gen_query_for_param(**kwargs).count()

tests/rdb/domain/repository.py

@@ -6,9 +6,9 @@
 
 class AsyncMemoRepository(AsyncRepository[MemoEntity, int]):
     def __init__(self, session: Optional[AsyncSession]) -> None:
-        self.session = session
+        self._session = session
 
 
 class SyncMemoRepository(SyncRepository[MemoEntity, int]):
     def __init__(self, session: Optional[Session]) -> None:
-        self.session = session
+        self._session = session