From b60e5a005d911a77a3145d35d5a4234d3b1ca6a8 Mon Sep 17 00:00:00 2001
From: Matthias Bilger <matthias@bilger.info>
Date: Sun, 21 Oct 2018 09:07:04 +0200
Subject: [PATCH] enabled status information

---
 infomentor/__main__.py |  6 ++----
 infomentor/informer.py | 31 ++++++++++++++++++++++++++-----
 2 files changed, 28 insertions(+), 9 deletions(-)

diff --git a/infomentor/__main__.py b/infomentor/__main__.py
index d94a463..92e3356 100644
--- a/infomentor/__main__.py
+++ b/infomentor/__main__.py
@@ -107,7 +107,7 @@ def notify_users():
                 statusinfo['degraded_count'] = 1
             if user.apistatus.ok == False and statusinfo['ok'] == False:
                 if user.apistatus.degraded_count == 1 and user.wantstatus:
-                    send_status_update(user, statusinfo['info'])
+                    im.send_status_update(statusinfo['info'])
                 try:
                     statusinfo['degraded_count'] = user.apistatus['degraded_count'] + 1
                 except Exception as e:
@@ -116,13 +116,11 @@ def notify_users():
                 statusinfo['info'] = 'Works as expected, failed {} times'.format(user.apistatus.degraded_count)
                 statusinfo['degraded_count'] = 0
                 if user.wantstatus:
-                    send_status_update(user, statusinfo['info'])
+                    im.send_status_update(statusinfo['info'])
             user.apistatus.updateobj(statusinfo)
         logger.info('New API status: %s', user.apistatus)
         session.commit()
 
-def send_status_update(user, text):
-    pass
 
 def main():
     args = parse_args(sys.argv[1:])
diff --git a/infomentor/informer.py b/infomentor/informer.py
index 3e7d5f1..4a74c7a 100755
--- a/infomentor/informer.py
+++ b/infomentor/informer.py
@@ -15,6 +15,27 @@ class Informer(object):
         self.user = user
         self.im = im
 
+    def send_status_update(self, text):
+        try:
+            if self.user.notification.ntype == model.Notification.Types.PUSHOVER:
+                pushover.Client(self.user.notification.info).send_message(
+                    text,
+                    title='Status Infomentor',
+                    html=False,
+                )
+            elif self.user.notification.ntype == model.Notification.Types.EMAIL:
+                mail = MIMEText(text)
+                mail['Subject'] = f'Status Infomentor'
+                mail['From'] = 'infomentor@09a.de'
+                mail['To'] = self.user.notification.info
+                self._send_mail(mail)
+        except:
+            mail = MIMEText("Fehler bei Infomentor")
+            mail['Subject'] = f'Fehler bei infomentor'
+            mail['From'] = 'infomentor@09a.de'
+            mail['To'] = 'matthias@bilger.info'
+            self._send_mail(mail)
+
     def update_news(self):
         session = db.get_db()
         newslist = self.im.get_news_list()
@@ -107,10 +128,7 @@ class Informer(object):
                 encoders.encode_base64(msg)
             msg.add_header('Content-Disposition', 'attachment', filename=fname)
             outer.attach(msg)
-        s = smtplib.SMTP_SSL('09a.de')
-        s.login('infomentor@09a.de', '***REMOVED***')
-        s.send_message(outer)
-        s.quit()
+        self._send_mail(outer)
 
     def update_homework(self):
         session = db.get_db()
@@ -181,7 +199,10 @@ class Informer(object):
                 encoders.encode_base64(msg)
             msg.add_header('Content-Disposition', 'attachment', filename=fname)
             outer.attach(msg)
+        self._send_mail(outer)
+
+    def _send_mail(self, mail):
         s = smtplib.SMTP_SSL('09a.de')
         s.login('infomentor@09a.de', '***REMOVED***')
-        s.send_message(outer)
+        s.send_message(mail)
         s.quit()