Coverage for src/kwai/modules/club/repositories/coach_db_repository.py: 100%

1"""Module for defining a coach repository using a database.""" 

2 

3from kwai.core.db.database import Database 

4from kwai.core.domain.entity import Entity 

5from kwai.modules.club.domain.coach import CoachEntity, CoachIdentifier 

6from kwai.modules.club.repositories._tables import CoachRow 

7from kwai.modules.club.repositories.coach_repository import CoachRepository 

8 

9 

10class CoachDbRepository(CoachRepository): 

11 """A coach repository using a database.""" 

12 

13 def __init__(self, database: Database) -> None: 

14 self._database = database 

15 

16 async def create(self, coach: CoachEntity): 

17 new_coach_id = await self._database.insert( 

18 CoachRow.__table_name__, CoachRow.persist(coach) 

19 ) 

20 return Entity.replace(coach, id_=CoachIdentifier(new_coach_id)) 

21 

22 async def delete(self, coach: CoachEntity): 

23 await self._database.delete(coach.id.value, CoachRow.__table_name__)