Add image generator

Author: Korusuke

automate.py

@@ -1,6 +1,7 @@
 from scrape import scraper
 from file_utils import read_csv, write_csv, read_file, write_file
 from generate import _render_template, preprocess
+from image import pass_gen
 from mail import sendmail
 import json
 
@@ -12,17 +13,20 @@
 participants = read_csv()
 participants = preprocess(participants)
 
-# Getting the list of mails to whom mails have already been sent 
+# Getting the list of mails to whom mails have already been sent
 sent_mails = read_file()
 
 # Looping over all participants
 for participant in participants:
     # Checking if the participant was sent a mail previously
     if participant['email'] not in sent_mails:
-        # Generating a message from the template 
+        # Generating a message from the template
         msg = _render_template(participant['name'], participant['payment'])
+        # Generating a custom image
+        image_path = pass_gen(participant['name'], participant['email'], participant['phone'])
         # Sending the message to the participant via mail
-        response = sendmail(to_email=participant['email'], msg=msg)
-        if response['email_status'] == "Success":
+        response = sendmail(to_email=participant['email'], message=msg, image_path=image_path)
+        print(response)
+        if response['email_status'] == "Success" and participant['payment'] == "paid":
             # if mail was sent successfully append the email to sentmails.txt
             write_file(participant['email'])

image.py

@@ -0,0 +1,19 @@
+from PIL import Image,ImageDraw,ImageFont
+import string, random
+
+def pass_gen(name, email, phone):
+    img = Image.open("templates/pass.png")
+    draw = ImageDraw.Draw(img)
+    font = ImageFont.truetype("templates/Quicksand-Bold.otf", 40)
+
+    s_path = "passes/" + email + ".png"
+    if len(name) >= 18:
+        name = name.split()
+        name = name[0] + " " + name[1][0]
+
+    draw.text((200, 300), name.capitalize(), font=font, fill=(255, 255, 255))
+    draw.text((650, 300), phone, font=font, fill=(255, 255, 255))
+    draw.text((200, 500), email.lower(), font=font, fill=(255, 255, 255))
+
+    img.save(s_path)
+    return s_path

mail.py

@@ -1,20 +1,28 @@
 from email.mime.text import MIMEText
 from email.mime.multipart import MIMEMultipart
+from email.mime.image import MIMEImage
 import smtplib
 from flask import render_template
 import os
 import sys
 
-def sendmail(to_email, msg):
-    
-    from_email = ''
+def sendmail(to_email, message, image_path):
+
+    from_email = ""
     msg = MIMEMultipart('alternative')
     msg['Subject'] = "Python Automation Workstop"
     msg['From'] = from_email
     msg['To'] = to_email
-    
-    content = msg
+
+    content = MIMEText(message, 'plain')
     msg.attach(content)
+
+    # This example assumes the image is in the given directory
+    fp = open(image_path, 'rb')
+    msgImage = MIMEImage(fp.read())
+    fp.close()
+    msg.attach(msgImage)
+
     response = {}
     try:
         with smtplib.SMTP("smtp.gmail.com", 587) as s:

scrape.py

@@ -10,7 +10,7 @@ def scraper(url):
     data = []
 
     # converting the html string to soup object
-    soup = bs4.BeautifulSoup(html)
+    soup = bs4.BeautifulSoup(html, "html.parser")
     li_tags = soup.select('li')
 
     for i in range(len(li_tags)):