Coverage for bc/kwai-bc-club/src/kwai_bc_club/get_user_coach.py: 93%

1"""Use case for getting the coach from a user.""" 

2 

3from dataclasses import dataclass 

4 

5from kwai_core.domain.presenter import Presenter 

6from kwai_core.domain.value_objects.unique_id import UniqueId 

7 

8from kwai_bc_club.domain.club_coach import ClubCoachEntity 

9from kwai_bc_club.repositories.coach_repository import CoachRepository 

10 

11 

12@dataclass(frozen=True, slots=True, kw_only=True) 

13class GetUserCoachCommand: 

14 """The input for the use case.""" 

15 

16 uuid: str 

17 

18 

19class GetUserCoach: 

20 """The use case for getting the coach from a user.""" 

21 

22 def __init__( 

23 self, coach_repo: CoachRepository, presenter: Presenter[ClubCoachEntity] 

24 ) -> None: 

25 """Initialize the use case.""" 

26 self._coach_repo = coach_repo 

27 self._presenter = presenter 

28 

29 async def execute(self, command: GetUserCoachCommand) -> None: 

30 """Execute the use case. 

31 

32 Raises: 

33 CoachNotFoundException: If the user is not attached to a coach. 

34 """ 

35 query = self._coach_repo.create_query().filter_by_user_uuid( 

36 UniqueId.create_from_string(command.uuid) 

37 ) 

38 coach = await self._coach_repo.get(query) 

39 self._presenter.present(coach)