-
Notifications
You must be signed in to change notification settings - Fork 0
/
message_handler.py
54 lines (36 loc) · 1.27 KB
/
message_handler.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
from email.mime.image import MIMEImage
import json
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from dotenv import dotenv_values
ENV = dotenv_values(".env")
def load_template(path: str):
with open(path, "r", encoding="UTF-8") as file:
return file.read()
def generate_html(path: str, args: dict):
content = load_template(path)
for name, tag_cont in args.items():
tag = "{" + name + "}"
content = content.replace(tag, tag_cont)
return content
def add_images(msg: MIMEMultipart):
for name, path in json.loads(ENV["EMBEDDED_IMG"]).items():
with open(path, "rb") as image_file:
# Create an image MIME object
image = MIMEImage(image_file.read())
# Set the image filename
image.add_header("Content-ID", name)
# Attach the image to the message
msg.attach(image)
return msg
def generate_msg(
sender: str, receiver: str, subject: str, html_path: str, args: dict
) -> str:
msg = MIMEMultipart()
msg["From"] = sender
msg["To"] = receiver
msg["Subject"] = subject
content = MIMEText(generate_html(html_path, args), "html")
msg.attach(content)
msg = add_images(msg)
return msg.as_string()