From 5736cbe1abc7a18d65f7b2abc8556b657db503ce Mon Sep 17 00:00:00 2001
From: panki27 <panki312@gmail.com>
Date: Thu, 16 Nov 2017 17:33:01 +0100
Subject: [PATCH] quick train commit, added comments and a bit of refactoring
 and exception handling

---
 certAlert.py | 53 ++++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 43 insertions(+), 10 deletions(-)

diff --git a/certAlert.py b/certAlert.py
index 13395c0..04e5cb9 100644
--- a/certAlert.py
+++ b/certAlert.py
@@ -1,18 +1,24 @@
 #!/usr/bin/env python
 import datetime, sys, re, urllib2, logging
 import pushnotify
+#TODO: replace pushnotify with a simple HTTP_POST
 from bs4 import BeautifulSoup
 
 ERRSTR = '!!!!!!!!!!!!! '
 
-PUSHOVER_DEVICE = 'chromehome'
 
+
+
+# REMEMBER TO CHANGE THESE!!!
 TARGET_URL = 'https://www.cert-bund.de/overview/AdvisoryShort'
 MEMORY_PATH = 'C:\Users\Panki\Desktop\Privat\Dev\certAlert\out.txt'
 
+#TODO: Put all this in a single file
 USER_KEY_PATH = 'C:\Users\Panki\Desktop\Privat\Dev\pushover_user'
 API_KEY_PATH = 'C:\Users\Panki\Desktop\Privat\Dev\pushover_key'
+PUSHOVER_DEVICE = 'chromehome'
 
+# To monitor more programs, simply add a string here
 PROGRAMS = [u'Chrome', u'OpenSSH', u'Java', u'Linux', u'Apache', u'Windows']
 
 with open(USER_KEY_PATH, 'r') as userKeyFile:
@@ -22,6 +28,7 @@ with open(API_KEY_PATH, 'r') as apiKeyFile:
     API_KEY = apiKeyFile.read()
     userKeyFile.close()
 
+# object to store a single cert alert
 class Advisory:
     def __init__(self, html):
         self.date = datetime.datetime.strptime(html.td.text, '%d.%m.%y').date()
@@ -37,6 +44,7 @@ class Advisory:
         print('link: ' + self.link)
 
 def startLogger():
+    # thanks to whoever i stole this from
     logger = logging.getLogger('pushnotify')
     logger.setLevel(logging.DEBUG)
     formatter = logging.Formatter('%(name)s-%(levelname)s: %(message)s')
@@ -44,31 +52,54 @@ def startLogger():
     handler.setFormatter(formatter)
     logger.addHandler(handler)
 
+def getHTML(url):
+    try:
+        response = urllib2.open(url)
+    except urllib2.URLError:
+        print(ERRSTR + 'Failed getting webpage!')
+        print(ERRSTR + 'Check your internet connection or TARGET_URL.')
+        sys.exit(ERRSTR + 'Stopping execution!')
+    except:
+        e = sys.exc_info()[0]
+        print(ERRSTR + 'Error getting Webpage!')
+        print(e)
+        sys.exit(ERRSTR + 'Stopping execution!')
+    result = response.read();
+    return result
    
 def main():
     import pushnotify
     startLogger()
-    #readKeys()
     client = pushnotify.get_client('pushover', API_KEY, 'certAlert')
     client.add_key(USER_KEY, PUSHOVER_DEVICE)
-    try:
-        response = urllib2.urlopen(TARGET_URL)
-    except:
-        e = sys.exc_info()[0]
-        print(ERRSTR + 'Error getting Webpage!\r\n' + e)
-        sys.exit(ERRSTR + 'Stopping execution!')
-    html = response.read()
+    #try:
+    #    response = urllib2.urlopen(TARGET_URL)
+    #except urllib2.URLError:
+    #    print(ERRSTR + 'Failed getting webpage!')
+    #    print(ERRSTR + 'Check your internet connection or TARGET_URL.')
+    #    sys.exit(ERRSTR + 'Stopping execution!')
+    #except:
+    #    e = sys.exc_info()[0]
+    #   print(ERRSTR + 'Error getting Webpage!')
+    #    print(e)
+    #    sys.exit(ERRSTR + 'Stopping execution!')
+    #html = response.read()
+    html = getHTML(TARGET_URL)
     soup = BeautifulSoup(html, 'html.parser')
-
+    # create a list of results and add objects created with the data of each table row
     results = [] 
     for adv in soup.find_all('tr', {'class' : re.compile('search-result-*')}):
         x = Advisory(adv)
         results.append(x)
+    # here we're checking which advisory IDs we've already seen, 
+    # so we don't send multiple notifications for the same advisory
+    # TODO: refactor into functions writeMemory(checkeIDs), readMemory()
     try:
         with open(MEMORY_PATH, 'r') as memFile:
             checkedIDs = memFile.read() 
             memFile.close()
     except IOError:
+        # this most likely means file not found. this can happen during the first run
         print(ERRSTR + 'Error reading memory file!')
         print(ERRSTR + 'Continuing without list of checked IDs...')
         checkedIDs = ''
@@ -78,6 +109,7 @@ def main():
         print(e)
     for result in results:
         if result.risk > 3:
+            # here we're checking if the is related to our programs
             for prog in PROGRAMS:
                 if re.match(prog, result.description, re.IGNORECASE):
                     if result.identifier not in checkedIDs:
@@ -87,6 +119,7 @@ def main():
                         print('========================================================================')
                     else:
                         print('Already sent an alert for ' + result.identifier +', skipping...')
+    # now we overwrite our memory file with the IDs we just checked
     with open(MEMORY_PATH, 'w') as memFile:
         for result in results:
             memFile.write(result.identifier + '\r\n')