support sending mail without pdf attachment

1 files changed, 13 insertions, 3 deletions

diff --git a/invoice/generate-invoice.py b/invoice/generate-invoice.py
index eb6c07e..3ac65b2 100755
--- a/invoice/generate-invoice.py
+++ b/invoice/generate-invoice.py
@@ -135,9 +135,10 @@ def generate_mail(receiver, subject, message, pdfdata, cc = None):
 
 	msg.attach(MIMEText(message, 'plain', 'utf-8'))
 
-	pdf = MIMEApplication(pdfdata, 'pdf')
-	pdf.add_header('Content-Disposition', 'attachment', filename = 'rechnung.pdf')
-	msg.attach(pdf)
+	if pdfdata is not None:
+		pdf = MIMEApplication(pdfdata, 'pdf')
+		pdf.add_header('Content-Disposition', 'attachment', filename = 'rechnung.pdf')
+		msg.attach(pdf)
 
 	return msg
 
@@ -196,4 +197,13 @@ def daily(timestamp = time.time()):
 		else:
 			print("Can't send invoice for missing user with the following id:", user)
 
+def monthly(timestamp = time.time()):
+	print("monthly invoice()")
+
+def backup(timestamp = time.time()):
+	print("backup()")
+
+def stock(timestamp = time.time()):
+	print("stock()")
+
 daily()