fa753239ad
Co-authored-by: Yongtao Huang <99629139+hyongtao-db@users.noreply.github.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
import logging
|
|
|
|
import sendgrid # type: ignore
|
|
from python_http_client.exceptions import ForbiddenError, UnauthorizedError
|
|
from sendgrid.helpers.mail import Content, Email, Mail, To # type: ignore
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SendGridClient:
|
|
def __init__(self, sendgrid_api_key: str, _from: str):
|
|
self.sendgrid_api_key = sendgrid_api_key
|
|
self._from = _from
|
|
|
|
def send(self, mail: dict):
|
|
logger.debug("Sending email with SendGrid")
|
|
|
|
try:
|
|
_to = mail["to"]
|
|
|
|
if not _to:
|
|
raise ValueError("SendGridClient: Cannot send email: recipient address is missing.")
|
|
|
|
sg = sendgrid.SendGridAPIClient(api_key=self.sendgrid_api_key)
|
|
from_email = Email(self._from)
|
|
to_email = To(_to)
|
|
subject = mail["subject"]
|
|
content = Content("text/html", mail["html"])
|
|
mail = Mail(from_email, to_email, subject, content)
|
|
mail_json = mail.get() # type: ignore
|
|
response = sg.client.mail.send.post(request_body=mail_json)
|
|
logger.debug(response.status_code)
|
|
logger.debug(response.body)
|
|
logger.debug(response.headers)
|
|
|
|
except TimeoutError as e:
|
|
logger.exception("SendGridClient Timeout occurred while sending email")
|
|
raise
|
|
except (UnauthorizedError, ForbiddenError) as e:
|
|
logger.exception(
|
|
"SendGridClient Authentication failed. "
|
|
"Verify that your credentials and the 'from' email address are correct"
|
|
)
|
|
raise
|
|
except Exception as e:
|
|
logger.exception("SendGridClient Unexpected error occurred while sending email to %s", _to)
|
|
raise
|