Coverage for portality / events / consumers / account_passwordreset_email.py: 100%

1# ~~ AccountPasswordResetEmail:Consumer ~~ 

2from portality.util import url_for 

3from portality.events.consumer import EventConsumer 

4from portality import constants 

5from portality import app_email 

6from portality import models 

7from portality.core import app 

8from portality.bll.exceptions import NoSuchPropertyException 

9from portality.ui import templates 

10 

11 

12class AccountPasswordResetEmail(EventConsumer): 

13 ID = "account:password_reset:email" 

14 

15 @classmethod 

16 def should_consume(cls, event): 

17 return event.id == constants.EVENT_ACCOUNT_PASSWORD_RESET and event.context.get("account") is not None 

18 

19 @classmethod 

20 def consume(cls, event): 

21 acc = models.Account(**event.context.get("account")) 

22 if not acc.reset_token or not acc.email: 

23 raise NoSuchPropertyException("Account {x} does not have a reset_token and/or an email address".format(x=acc.id)) 

24 cls._send_password_reset_email(acc) 

25 

26 @classmethod 

27 def _send_password_reset_email(cls, account: models.Account): 

28 # ~~-> AccountPasswordReset:Email~~ 

29 reset_url = app.config.get('BASE_URL', "https://doaj.org") + url_for('account.reset', reset_token=account.reset_token) 

30 

31 to = [account.email] 

32 fro = app.config.get('SYSTEM_EMAIL_FROM', app.config['ADMIN_EMAIL']) 

33 subject = app.config.get("SERVICE_NAME", "") + " - password reset" 

34 

35 # ~~-> Email:Library ~~ 

36 app_email.send_mail(to=to, 

37 fro=fro, 

38 subject=subject, 

39 template_name=templates.EMAIL_PASSWORD_RESET, 

40 email=account.email, 

41 reset_url=reset_url, 

42 forgot_pw_url=app.config.get('BASE_URL', "https://doaj.org") + url_for('account.forgot') 

43 )