Create EC2_Code.py

Code:
The code below will execute the SMTP protocol as soon as a file reaches the folder where this code is located.

Instructions:
Place this code in the created folder. The folder name should be the same as the one mentioned in the lambda code.

Author: Atul-Kumar-Rana

EC2_Code.py

@@ -0,0 +1,71 @@
+from flask import Flask, request, jsonify
+import smtplib
+import csv
+import os
+import boto3
+from werkzeug.utils import secure_filename
+
+app = Flask(__name__)
+
+# Define the upload folder for the CSV file
+UPLOAD_FOLDER = '/tmp'
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+
+# Define the allowed file extensions
+ALLOWED_EXTENSIONS = {'csv'}
+
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
+@app.route('/send-emails', methods=['POST'])
+def send_emails():
+    data = request.get_json()
+    
+    # Extract data from the request
+    bucket = data['bucket']
+    key = data['key']
+    filename = data['filename']
+    sender_email = data['sender_email']
+    sender_password = data['sender_password']
+    subject = data['subject']
+    body = data['body']
+    
+    # Download the file from S3
+    s3 = boto3.client('s3')
+    s3.download_file(bucket, key, os.path.join(app.config['UPLOAD_FOLDER'], filename))
+    
+    if allowed_file(filename):
+        # Read the CSV file and send emails
+        with open(os.path.join(app.config['UPLOAD_FOLDER'], filename), 'r') as f:
+            reader = csv.reader(f)
+            for row in reader:
+                recipient_email = row[1]
+                if '@' in recipient_email:
+                    try:
+                        # Set up the SMTP server
+                        if 'outlook.com' in sender_email:
+                            host = 'smtp-mail.outlook.com
+                            port = 587
+                        elif 'gmail.com' in sender_email:
+                            host = 'smtp.gmail.com'
+                            port = 587
+
+                        smtp = smtplib.SMTP(host, port)
+                        smtp.ehlo()
+                        smtp.starttls()
+                        smtp.login(sender_email, sender_password)
+
+                        # Send the email
+                        msg = f'Subject: {subject}\n\n{body}'
+                        smtp.sendmail(sender_email, recipient_email, msg)
+                        smtp.quit()
+                        print(f'Email sent to {recipient_email}')
+                    except Exception as e:
+                        print(f'Error sending email to {recipient_email}: {str(e)}')
+
+        return jsonify({'message': 'Emails sent successfully'}), 200
+    else:
+        return jsonify({'error': 'Invalid file type'}), 400
+
+if __name__ == '__main__':
+    app.run(debug=True, host='0.0.0.0', port=5000)