Coverage for bc/kwai-bc-identity/src/kwai_bc_identity/user_recoveries/user_recovery_repository.py: 100%

1"""Module that defines an interface for a user recovery repository.""" 

2 

3from abc import abstractmethod 

4 

5from kwai_core.domain.value_objects.unique_id import UniqueId 

6 

7from kwai_bc_identity.user_recoveries.user_recovery import UserRecoveryEntity 

8 

9 

10class UserRecoveryRepository: 

11 """Interface for a user recovery repository.""" 

12 

13 @abstractmethod 

14 async def get_by_uuid(self, uuid: UniqueId) -> UserRecoveryEntity: 

15 """Get a user recovery with the given unique id.""" 

16 raise NotImplementedError 

17 

18 @abstractmethod 

19 async def create(self, user_recovery: UserRecoveryEntity) -> UserRecoveryEntity: 

20 """Save a new user recovery.""" 

21 raise NotImplementedError 

22 

23 @abstractmethod 

24 async def update(self, user_recovery: UserRecoveryEntity): 

25 """Save a user recovery.""" 

26 raise NotImplementedError 

27 

28 @abstractmethod 

29 async def delete(self, user_recovery: UserRecoveryEntity): 

30 """Deletes a user recovery.""" 

31 raise NotImplementedError 

32 

33 

34class UserRecoveryNotFoundException(Exception): 

35 """Raised when the user recovery could not be found."""