summaryrefslogtreecommitdiff
path: root/jimbrella/sms.py
blob: e83dc540b7799e846d25e8673fb4b9e1cad0251b (plain)
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
import logging
import json
from alibabacloud_dysmsapi20170525.client import Client as DysmsapiClient
from alibabacloud_dysmsapi20170525 import models as dysmsapi_models
from alibabacloud_tea_openapi import models as open_api_models
from .config import config


class SMS:
    """SMS notification client.

    This client talks to Aliyun's shit-in-a-box API when JImbrella needs to
    remind a tenant to return their overdue umbrella.
    """

    def __init__(self):
        """Initialize Aliyun client"""
        self.config = open_api_models.Config(
            access_key_id=config.get("sms", "access_key_id"),
            access_key_secret=config.get("sms", "access_key_secret"),
        )
        self.client = DysmsapiClient(self.config)

    def _send(
        self,
        phone_numbers: str,
        sign_name: str,
        template_code: str,
        template_param: dict,
    ):
        """Call API to send generic SMS"""
        req = dysmsapi_models.SendSmsRequest(
            phone_numbers=phone_numbers,
            sign_name=sign_name,
            template_code=template_code,
            template_param=json.dumps(template_param),
        )
        resp = self.client.send_sms(req)
        if resp.body.code != "OK":
            logging.warning(
                "API call to send SMS notification to %s failed", phone_numbers
            )

    def remind_overdue(self, tenant_phone: str, tenant_name: str, key: int):
        self._send(tenant_phone, "TBD", "TBD", {"name": tenant_name, "key": key})