Email business logic

src/registrar/utility/email.py

@@ -4,6 +4,10 @@ import boto3
 import logging
 from django.conf import settings
 from django.template.loader import get_template
+from email.mime.base import MIMEBase
+from email.mime.application import MIMEApplication
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
 
 
 logger = logging.getLogger(__name__)
@@ -15,7 +19,7 @@ class EmailSendingError(RuntimeError):
     pass
 
 
-def send_templated_email(template_name: str, subject_template_name: str, to_address: str, context={}):
+def send_templated_email(template_name: str, subject_template_name: str, to_address: str, context={}, file: str=None):
     """Send an email built from a template to one email address.
 
     template_name and subject_template_name are relative to the same template
@@ -40,74 +44,56 @@ def send_templated_email(template_name: str, subject_template_name: str, to_addr
     except Exception as exc:
         raise EmailSendingError("Could not access the SES client.") from exc
 
-    # Are we okay with passing in "attachment" var in as boolean parameter
-    # If so, TODO: add attachment boolean to other functions
     try:
-        #if not attachment: 
-        ses_client.send_email(
-            FromEmailAddress=settings.DEFAULT_FROM_EMAIL,
-            Destination={"ToAddresses": [to_address]},
-            Content={
-                "Simple": {
-                    "Subject": {"Data": subject},
-                    "Body": {"Text": {"Data": email_body}},
+        if file is None:
+            ses_client.send_email(
+                FromEmailAddress=settings.DEFAULT_FROM_EMAIL,
+                Destination={"ToAddresses": [to_address]},
+                Content={
+                    "Simple": {
+                        "Subject": {"Data": subject},
+                        "Body": {"Text": {"Data": email_body}},
+                    },
                 },
-            },
-        )
-        # else: # has attachment
-            # same as above but figure out how to attach a file
-            # via boto3 "boto3 SES file attachment"
-            # we also want this to only send to the help email
-        
-            # from email.mime.multipart import MIMEMultipart
-            # from email.mime.text import MIMEText
-            # from email.mime.application import MIMEApplication
+            )
+        if file is not None:
+            # TODO: Update sender email when we figure out
+            ses_client = boto3.client(
+                "ses",
+                region_name=settings.AWS_REGION,
+                aws_access_key_id=settings.AWS_ACCESS_KEY_ID,
+                aws_secret_access_key=settings.AWS_SECRET_ACCESS_KEY,
+                config=settings.BOTO_CONFIG,
+            )
 
-            # sender_email = 'sender@example.com'
-            # recipient_email = 'help@get.gov'
-            # subject = 'DOTGOV-Full Domain Metadata'
-            # body = 'Domain metadata email, should have an attachment included change here later.'
-            # attachment_path = 'path/to/attachment/file.pdf'
-            # aws_region = 'sesv2'
-
-            # response = send_email_with_attachment(sender_email, recipient_email, subject, body, attachment_path, aws_region)
-            # print(response)
+            #TODO: Update sender to settings.DEFAULT_FROM_EMAIL
+            response = send_email_with_attachment(settings.DEFAULT_FROM_EMAIL, to_address, subject, email_body, file, ses_client)
+            print("Response from send_email_with_attachment_is:", response)
     except Exception as exc:
         raise EmailSendingError("Could not send SES email.") from exc
 
+def send_email_with_attachment(sender, recipient, subject, body, attachment_file, ses_client):
+    # Create a multipart/mixed parent container
+    msg = MIMEMultipart('mixed')
+    msg['Subject'] = subject
+    msg['From'] = sender
+    msg['To'] = recipient
 
-# def send_email_with_attachment(sender, recipient, subject, body, attachment_path, aws_region):
-            #     # Create a multipart/mixed parent container
-            #     msg = MIMEMultipart('mixed')
-            #     msg['Subject'] = subject
-            #     msg['From'] = sender_email
-            #     msg['To'] = recipient_email
+    # Add the text part
+    text_part = MIMEText(body, 'plain')
+    msg.attach(text_part)
 
-            #     # Add the text part
-            #     text_part = MIMEText(body, 'plain')
-            #     msg.attach(text_part)
+    # Add the attachment part
 
-            #     # Add the attachment part
-            #     with open(attachment_path, 'rb') as attachment_file:
-            #         attachment_data = attachment_file.read()
-            #     attachment_part = MIMEApplication(attachment_data)
-            #     attachment_part.add_header('Content-Disposition', f'attachment; filename="{attachment_path}"')
-            #     msg.attach(attachment_part)
+    # set it into this "type"
+    attachment_part = MIMEApplication(attachment_file)
+    # Adding attachment header + filename that the attachment will be called
+    attachment_part.add_header('Content-Disposition', f'attachment; filename="encrypted_metadata.zip"')
+    msg.attach(attachment_part)
 
-            #     # Send the email
-            #     response = ses_client.send_raw_email(
-            #         Source=sender,
-            #         Destinations=[recipient],
-            #         RawMessage={'Data': msg.as_string()}
-            #     )
-
-            #     ses_client.send_email(
-            #     FromEmailAddress=settings.DEFAULT_FROM_EMAIL,
-            #     Destination={"ToAddresses": [to_address]},
-            #     Content={
-            #         "Simple": {
-            #             "Subject": {"Data": subject},
-            #             "Body": {"Text": {"Data": email_body}},
-            #         },
-            #     },
-            # )
+    response = ses_client.send_raw_email(
+        Source=sender,
+        Destinations=[recipient],
+        RawMessage={"Data": msg.as_string()}
+    )
+    return response