Skip to content

Commit

Permalink
enable email api send alerts to multiple email addresses
Browse files Browse the repository at this point in the history
  • Loading branch information
JackZhao516 committed Jul 22, 2024
1 parent 14c02cd commit 2bd6259
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
8 changes: 4 additions & 4 deletions smrti_quant_alerts/email_api/email_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def __init__(self, sender_email: str = None, receiver_email: str = None, passwor
self.port = 465 # For SSL
self.smtp_server = "smtp.gmail.com"
self.sender_email = self.email_tokens["SENDER_EMAIL"] if not sender_email else sender_email
self.receiver_email = self.email_tokens["RECEIVER_EMAIL"] if not receiver_email else receiver_email
self.receiver_emails = self.email_tokens["RECEIVER_EMAIL"] if not receiver_email else receiver_email
self.password = self.email_tokens["PASSWORD"] if not password else password

@error_handling("email", default_val=None)
Expand All @@ -30,13 +30,13 @@ def send_email(self, subject: str, body: str, csv_file_name: str = None,
:param csv_file_name: csv file path
:param pdf_file_name: pdf file path
"""
if not self.sender_email or not self.receiver_email or not self.password:
if not self.sender_email or not self.receiver_emails or not self.password:
return

message = MIMEMultipart("alternative")
message["Subject"] = subject
message["From"] = self.sender_email
message["To"] = self.receiver_email
message["To"] = ','.join(self.receiver_emails)
message.attach(MIMEText(body, "plain"))

if csv_file_name:
Expand All @@ -55,4 +55,4 @@ def send_email(self, subject: str, body: str, csv_file_name: str = None,
context = ssl.create_default_context()
with smtplib.SMTP_SSL(self.smtp_server, self.port, context=context) as server:
server.login(self.sender_email, self.password)
server.sendmail(self.sender_email, self.receiver_email, message.as_string())
server.sendmail(self.sender_email, self.receiver_emails, message.as_string())
2 changes: 1 addition & 1 deletion token.json.example
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,6 @@
"GMAIL": {
"SENDER_EMAIL": "<gmail address>",
"PASSWORD": "<gmail password>",
"RECEIVER_EMAIL": "<email address>"
"RECEIVER_EMAIL": ["<email address>", "<email address>"]
}
}

0 comments on commit 2bd6259

Please sign in to comment.