Coverage for src/kwai/modules/identity/get_user_accounts.py: 73%

1"""Module for the get user accounts use case.""" 

2 

3from dataclasses import dataclass 

4 

5from kwai.core.domain.presenter import AsyncPresenter, IterableResult 

6from kwai.modules.identity.users.user_account import UserAccountEntity 

7from kwai.modules.identity.users.user_account_repository import UserAccountRepository 

8 

9 

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

11class GetUserAccountsCommand: 

12 """Input for the use case. 

13 

14 Attributes: 

15 offset: Offset from where to start the query of user accounts. 

16 limit: The maximum number of users to return. 

17 """ 

18 

19 offset: int | None = None 

20 limit: int | None = None 

21 

22 

23class GetUserAccounts: 

24 """Implementation of the get user accounts use case.""" 

25 

26 def __init__( 

27 self, 

28 user_account_repo: UserAccountRepository, 

29 presenter: AsyncPresenter[IterableResult[UserAccountEntity]], 

30 ): 

31 """Initialize the get users use case.""" 

32 self._user_account_repo = user_account_repo 

33 self._presenter = presenter 

34 

35 async def execute(self, command: GetUserAccountsCommand): 

36 """Execute the get users use case. 

37 

38 Args: 

39 command: Input for the use case. 

40 """ 

41 query = self._user_account_repo.create_query() 

42 

43 await self._presenter.present( 

44 IterableResult( 

45 count=await query.count(), 

46 limit=command.limit, 

47 offset=command.offset, 

48 iterator=self._user_account_repo.get_all( 

49 query, command.limit, command.offset 

50 ), 

51 ) 

52 )