1 files changed, 56 insertions, 0 deletions
diff --git a/invoice/mailhelper.py b/invoice/mailhelper.py
new file mode 100644
index 0000000..49721cc
--- /dev/null
+++ b/invoice/mailhelper.py
@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+#-*- coding: utf-8 -*-
+
+import time
+
+class MAIL(object):
+	def __init__(self, server, port, username, password):
+		self.__server = server
+		self.__port = port
+		self.__username = username
+		self.__password = password
+		pass
+
+	def generate_mail(self, receiver, subject, message, attachments = None, timestamp=time.time(), cc = None):
+		msg = MIMEMultipart()
+		msg["From"] = "KtT-Shopsystem <shop@kreativitaet-trifft-technik.de>"
+		msg["Date"] = email.utils.formatdate(timestamp, True)
+
+		try:
+			if receiver.encode("ascii"):
+				msg["To"] = receiver
+		except UnicodeError:
+			msg["To"] = Header(receiver, 'utf-8')
+
+		if cc != None:
+			msg["Cc"] = cc
+		msg["Subject"] = Header(subject, 'utf-8')
+		msg.preamble = "Please use a MIME aware email client!"
+
+		msg.attach(MIMEText(message, 'plain', 'utf-8'))
+
+		if isinstance(attachments, dict):
+			for name, data in attachments.items():
+				if name.endswith("pdf"):
+					pdf = MIMEApplication(data, 'pdf')
+					pdf.add_header('Content-Disposition', 'attachment', filename = name)
+					msg.attach(pdf)
+				else:
+					txt = MIMEText(data, 'plain', 'utf-8')
+					txt.add_header('Content-Disposition', 'attachment', filename = name)
+					msg.attach(txt)
+		elif attachments is not None:
+			pdf = MIMEApplication(attachments, 'pdf')
+			pdf.add_header('Content-Disposition', 'attachment', filename = 'rechnung.pdf')
+			msg.attach(pdf)
+
+		return msg
+
+	def send_mail(self, mail, receiver):
+		server = smtplib.SMTP(self.__server, self.__port)
+		server.starttls()
+		if self.__username != "":
+			server.login(self.__username, self.__password)
+		maildata = mail.as_string()
+		server.sendmail(mail["From"], receiver, maildata)
+		server.quit()