Coverage for src/kwai/events/v1/identity/user_recovery_tasks.py: 0%

1"""Module that defines entry points for tasks for user recoveries.""" 

2 

3from typing import Any 

4 

5import inject 

6 

7from loguru import logger 

8 

9from kwai.core.db.database import Database 

10from kwai.core.db.uow import UnitOfWork 

11from kwai.core.domain.exceptions import UnprocessableException 

12from kwai.core.domain.value_objects.email_address import EmailAddress 

13from kwai.core.events.event_router import EventRouter 

14from kwai.core.mail.mailer import Mailer 

15from kwai.core.mail.recipient import Recipient, Recipients 

16from kwai.core.settings import Settings 

17from kwai.core.template.mail_template import MailTemplate 

18from kwai.core.template.template_engine import TemplateEngine 

19from kwai.modules.identity.mail_user_recovery import ( 

20 MailUserRecovery, 

21 MailUserRecoveryCommand, 

22) 

23from kwai.modules.identity.user_recoveries.user_recovery_db_repository import ( 

24 UserRecoveryDbRepository, 

25) 

26from kwai.modules.identity.user_recoveries.user_recovery_events import ( 

27 UserRecoveryCreatedEvent, 

28) 

29from kwai.modules.identity.user_recoveries.user_recovery_repository import ( 

30 UserRecoveryNotFoundException, 

31) 

32 

33 

34@inject.autoparams() 

35async def email_user_recovery_task( 

36 event: dict[str, Any], 

37 settings: Settings, 

38 database: Database, 

39 mailer: Mailer, 

40 template_engine: TemplateEngine, 

41): 

42 """Actor for sending a user recovery mail.""" 

43 command = MailUserRecoveryCommand(uuid=event["data"]["uuid"]) 

44 

45 try: 

46 async with UnitOfWork(database): 

47 await MailUserRecovery( 

48 UserRecoveryDbRepository(database), 

49 mailer, 

50 Recipients( 

51 from_=Recipient( 

52 email=EmailAddress(settings.website.email), 

53 name=settings.website.name, 

54 ) 

55 ), 

56 MailTemplate( 

57 template_engine.create("mail/identity/recover_html"), 

58 template_engine.create("mail/identity/recover_txt"), 

59 ), 

60 ).execute(command) 

61 except UnprocessableException as ex: 

62 logger.error(f"Task not processed: {ex}") 

63 except UserRecoveryNotFoundException: 

64 logger.error( 

65 f"Mail not send because user recovery does not exist " 

66 f"with uuid {command.uuid}" 

67 ) 

68 

69 

70router = ( 

71 EventRouter(event=UserRecoveryCreatedEvent, callback=email_user_recovery_task), 

72)