if1007-microservices/mail_manager.py at master · b0rba/if1007-microservices · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
from typing import Dict, List

import requests
from dateutil.relativedelta import relativedelta

from metrics.topics import MAILS_FAIL, MAILS_SENT
from schemas.mail_params import MailParams
from schemas.recipient import Recipient
from schemas.template import Template


class MailManager:
    def __init__(self, api_key: str, url: str):
        self.api_key = api_key
        self.url = url

    def _send_email(self, to: List[Recipient], template: Template, params: Dict[str, str] = None):
        json = {
            'templateId': template.value,
            'to': [user.dict() for user in to],
        }
        if params is not None:
            json['params'] = params
        response = requests.post(f'{self.url}/smtp/email', headers={'api-key': self.api_key}, json=json)

        if not 200 <= response.status_code < 300:
            MAILS_FAIL.inc()
            print(str(response.json()))
            return False
        MAILS_SENT.inc()
        return True

    def _send_confirmation_to_mentored(self, to: Recipient, mail_metadata: MailParams):
        params = {
            'MENTOR_NAME': mail_metadata.mentor_name,
            'MENTORING_DATE': mail_metadata.mentoring_datetime.strftime('%d/%m/%Y'),
            'MENTORING_HOUR': mail_metadata.mentoring_datetime.strftime('%H:%M'),
            'MENTORED_NAME': mail_metadata.mentored_name,
            'MENTOR_ID': mail_metadata.mentor_id
        }
        return self._send_email([to], Template.CONFIRMATION_TO_MENTORED, params)

    def _send_confirmation_to_mentor(self, to: Recipient, mail_metadata: MailParams):
        params = {
            'MENTOR_NAME': mail_metadata.mentor_name,
            'MENTORING_DATE': mail_metadata.mentoring_datetime.strftime('%d/%m/%Y'),
            'MENTORING_HOUR': mail_metadata.mentoring_datetime.strftime('%H:%M'),
            'MENTORED_NAME': mail_metadata.mentored_name,
            'MENTORED_ID': mail_metadata.mentored_id
        }
        return self._send_email([to], Template.CONFIRMATION_TO_MENTOR, params)

    def _notify_subscribed_users_not_selected(self, to: Recipient, mail_metadata: MailParams):
        params = {
            'MENTOR_NAME': mail_metadata.mentor_name,
            'MENTORING_DATE': mail_metadata.mentoring_datetime.strftime('%d/%m/%Y'),
            'MENTORING_HOUR': mail_metadata.mentoring_datetime.strftime('%H:%M'),
            'MENTORED_NAME': mail_metadata.mentored_name
        }
        return self._send_email([to], Template.NOTIFY_SUBSCRIBED_USERS_NOT_SELECTED, params)

    def _notify_subscribed_users_of_mentoring_canceled(self, to: Recipient, mail_metadata: MailParams):
        params = {
            'MENTOR_NAME': mail_metadata.mentor_name,
            'MENTOR_ID': mail_metadata.mentor_id,
            'MENTORING_DATE': mail_metadata.mentoring_datetime.strftime('%d/%m/%Y'),
            'MENTORING_HOUR': mail_metadata.mentoring_datetime.strftime('%H:%M'),
            'MENTORED_NAME': mail_metadata.mentored_name
        }
        return self._send_email([to], Template.NOTIFY_SUBSCRIBED_USERS_MENTORING_CANCELED, params)

    def _notify_mentor_of_mentored_leaving(self, to: Recipient, mail_metadata: MailParams):
        params = {
            'MENTOR_NAME': mail_metadata.mentor_name,
            'MENTORED_NAME': mail_metadata.mentored_name,
            'MENTORING_DATE_LIMIT': (mail_metadata.mentoring_datetime - relativedelta(days=1)).strftime('%d/%m/%Y'),
            'MENTORING_DATE': mail_metadata.mentoring_datetime.strftime('%d/%m/%Y'),
            'MENTORING_HOUR': mail_metadata.mentoring_datetime.strftime('%H:%M')
        }
        return self._send_email([to], Template.NOTIFY_MENTOR_OF_MENTORED_LEAVING, params)

    def get_available_credits(self):
        response = requests.get(f'{self.url}/account', headers={'api-key': self.api_key})
        if not 200 <= response.status_code < 300:
            print(str(response.json()))
            return
        return next((item['credits'] for item in response.json()['plan'] if item['type'] == 'free'), 0)

    def send_mail(self, recipient: Recipient, mail_params: MailParams, template: Template):
        if template is Template.CONFIRMATION_TO_MENTORED:
            return self._send_confirmation_to_mentored(recipient, mail_params)
        if template is Template.CONFIRMATION_TO_MENTOR:
            return self._send_confirmation_to_mentor(recipient, mail_params)
        if template is Template.NOTIFY_SUBSCRIBED_USERS_NOT_SELECTED:
            return self._notify_subscribed_users_not_selected(recipient, mail_params)
        if template is Template.NOTIFY_SUBSCRIBED_USERS_MENTORING_CANCELED:
            return self._notify_subscribed_users_of_mentoring_canceled(recipient, mail_params)
        if template is Template.NOTIFY_MENTOR_OF_MENTORED_LEAVING:
            return self._notify_mentor_of_mentored_leaving(recipient, mail_params)
        return False
        # raise Exception('Not a valid template')