feat: add full pay path

src/infrastructure/database/unit_of_work.py

@@ -1,7 +1,9 @@
 from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
 from src.application.abstractions import IUnitOfWork
+from src.application.abstractions.repositories import IOrderRepository,IPaymentRepository
 from src.application.contracts import ILogger
-# from src.application.abstractions.repositories import IUserRepository, ISessionRepository
+from src.infrastructure.database.repositories.order_repository import OrderRepository
+from src.infrastructure.database.repositories.payment_repository import PaymentRepository
 # from src.infrastructure.database.repositories import UserRepository, SessionRepository
 
 
@@ -10,12 +12,16 @@ class UnitOfWork(IUnitOfWork):
     def __init__(self, session_factory: async_sessionmaker[AsyncSession], logger: ILogger):
         self.session_factory = session_factory
         self._session: AsyncSession = None
+        self._order_repository: IOrderRepository | None = None
+        self._payment_repository: IPaymentRepository | None = None
         # self._user_repository: IUserRepository = None
         # self._session_repository: ISessionRepository = None
         self._logger: ILogger = logger
 
     async def __aenter__(self):
         self._session = self.session_factory()
+        self._order_repository = None
+        self._payment_repository = None
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
@@ -29,6 +35,19 @@ class UnitOfWork(IUnitOfWork):
             self._logger.debug('Commit')
         await self._session.close()
 
+    @property
+    def order_repository(self) -> IOrderRepository:
+        if self._order_repository is None:
+            self._order_repository = OrderRepository(session=self._session, logger=self._logger)
+        return self._order_repository
+
+
+    @property
+    def payment_repository(self) -> IPaymentRepository:
+        if self._payment_repository is None:
+            self._payment_repository = PaymentRepository(session=self._session, logger=self._logger)
+        return self._payment_repository
+
     # @property
     # def user_repository(self) -> IUserRepository:
     #     if self._user_repository is None: