Refactor send_email function to use Sendinblue API for sending emails

send_mail.py

@@ -1,20 +1,16 @@
-import os
-import smtplib
-from email.mime.text import MIMEText
-from email.mime.multipart import MIMEMultipart
-from email.mime.base import MIMEBase
 from email import encoders
+from email.mime.base import MIMEBase
+import os
+import sib_api_v3_sdk
+from sib_api_v3_sdk.rest import ApiException
 from dotenv import load_dotenv
 
 load_dotenv()
-smtp_port = os.getenv("PORT")
-smtp_server = os.getenv("SERVER")
-smtp_login = os.getenv("LOGIN")
-smtp_passwd = os.getenv("PASSWD")
+mail_api = os.getenv("MAIL_API")
 
-def mail_body(email, generation_details):
+def mail_body(generation_details):
     body = f"""
-    Hello {email},
+    Hello,
     Thankyou for using Project Gatekeeper to generate your SSL certificate.
     Your SSL certificate has been generated and is attached to this email. 
     Please find the attached file for your SSL certificate.
@@ -28,56 +24,36 @@ def mail_body(email, generation_details):
     """
     return body
 
-def make_attachment(private_key, csr, cert):
-    pvt = perpare_pvt(private_key)
-    csr = perpare_csr(csr)
-    ssl = perpare_ssl(cert)
-    return pvt, csr, ssl
-
-def perpare_pvt(pvt):
-    filename = "private_key.key"
-    pvt= pvt.encode('utf-8')
-    pvtkey = MIMEBase('application', 'octet-stream')
-    pvtkey.set_payload(pvt)
-    encoders.encode_base64(pvtkey)
-    pvtkey.add_header('Content-Disposition', "attachment; filename= " + filename)
-    return pvtkey
-
-def perpare_csr(csr):
-    filename = "domain.csr"
-    csr= csr.encode('utf-8')
-    domaincsr = MIMEBase('application', 'octet-stream')
-    domaincsr.set_payload(csr)
-    encoders.encode_base64(domaincsr)
-    domaincsr.add_header('Content-Disposition', "attachment; filename= " + filename)
-    return domaincsr
-
-def perpare_ssl(ssl):
-    filename = "ssl_certificate.crt"
-    ssl= ssl.encode('utf-8')
-    sslcrt = MIMEBase('application', 'octet-stream')
-    sslcrt.set_payload(ssl)
-    encoders.encode_base64(sslcrt)
-    sslcrt.add_header('Content-Disposition', "attachment; filename= " + filename)
-    return sslcrt
-
-def prepare_email(email, private_key, csr, cert, generation_details):
-    body = mail_body(email, generation_details)
-    msg = MIMEMultipart()
-    msg['From'] = "Project Gatekeeper <{}>".format(smtp_login)
-    msg['To'] = email
-    msg['Subject'] = "Project Gatekeeper - Your SSL Certificate is ready!"
-    msg.attach(MIMEText(body, 'plain'))
-    p, c, s = make_attachment(private_key, csr, cert)
-    for attachment_package in [p, c, s]:
-        msg.attach(attachment_package)
-    text = msg.as_string()
-    return text
+def create_attachment(content, filename):
+    attachment = MIMEBase('application', 'octet-stream')
+    attachment.set_payload(content)
+    encoders.encode_base64(attachment)
+    attachment.add_header('Content-Disposition', f'attachment; filename="{filename}"')
+    return attachment
 
 def send_email(email, private_key, csr, cert, generation_details):
-    data = prepare_email(email, private_key, csr, cert, generation_details)
-    TIE_server = smtplib.SMTP(smtp_server, smtp_port)
-    TIE_server.starttls()
-    TIE_server.login(smtp_login, smtp_passwd)
-    TIE_server.sendmail(from_addr=smtp_login, to_addrs=email, msg=data)
-    TIE_server.quit()
+    configuration = sib_api_v3_sdk.Configuration()
+    configuration.api_key['api-key'] = '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'
+    api_instance = sib_api_v3_sdk.TransactionalEmailsApi(sib_api_v3_sdk.ApiClient(configuration))
+
+    data = mail_body(generation_details)
+    p_attachment = create_attachment(private_key.encode('utf-8'), "private_key.txt")
+    c_attachment = create_attachment(csr.encode('utf-8'), "csr.txt")
+    s_attachment = create_attachment(cert.encode('utf-8'), "cert.txt")
+
+    subject = "Project Gatekeeper - Your SSL Certificate is ready!"
+    sender = {"name": "Project Gatekeeper", "email": "[email protected]"}
+    reply_to = {"name": "Project Gatekeeper", "email": "[email protected]"}
+    text_content = data
+    attachments = [
+        {"content": p_attachment.get_payload(), "name": p_attachment.get_filename()},
+        {"content": c_attachment.get_payload(), "name": c_attachment.get_filename()},
+        {"content": s_attachment.get_payload(), "name": s_attachment.get_filename()},
+    ]
+    to = [{"email": email}]
+    send_smtp_email = sib_api_v3_sdk.SendSmtpEmail(to=to, reply_to=reply_to, attachment=attachments, text_content=text_content, sender=sender, subject=subject)
+    try:
+        api_response = api_instance.send_transac_email(send_smtp_email)
+        print(api_response)
+    except ApiException as e:
+        print("Exception when calling SMTPApi->send_transac_email: %s\n" % e)