modsde module

modsde.py

@@ -0,0 +1,98 @@
+#!/usr/bin/env python
+
+import sys
+import json
+from datetime import datetime,timedelta
+import urllib, urllib2
+import re
+import cookielib
+import xml.etree.ElementTree as ET
+
+class ModsDeChecker:
+    """ 
+    This class returns i3status parsable output of the number of
+    unread posts in any bookmark in the mods.de forums.
+    """
+
+    last_checked = datetime.now()
+    unread_cache = 0
+    login_url = 'http://login.mods.de/'
+    bookmark_url = "http://forum.mods.de/bb/xml/bookmarks.php"
+    opener = None
+    cj = None
+    logged_in = False
+    
+    settings =  {
+        'color': '#7181fe',
+        'pause': 20,
+        'username': "",
+        'password': ""
+    }
+
+    def __init__(self, settings = None):
+        self.settings.update(settings)
+        self.cj = cookielib.CookieJar()
+        self.last_checked = \
+            datetime.now() - timedelta(seconds=self.settings['pause'])
+        self.opener = urllib2.build_opener(
+            urllib2.HTTPCookieProcessor(self.cj))
+
+    def get_unread_count(self):
+        delta = datetime.now() - self.last_checked
+
+        if delta.total_seconds() > self.settings['pause']:
+            if not self.logged_in:
+                try:
+                    self.login()
+                except Exception:
+                    pass
+            
+            try:
+                f = self.opener.open(self.bookmark_url)
+                root = ET.fromstring(f.read())
+                self.last_checked = datetime.now()
+                self.unread_cache = int(root.attrib['newposts'])
+            except Exception:
+                self.cj.clear()
+                self.opener = urllib2.build_opener(
+                    urllib2.HTTPCookieProcessor(self.cj))
+                self.logged_in = False
+
+        return self.unread_cache
+        
+
+    def login(self):
+
+        data = urllib.urlencode({
+            "login_username": self.settings["username"],
+            "login_password": self.settings["password"],
+            "login_lifetime": "31536000"
+        })
+
+        response = self.opener.open(self.login_url, data)
+        m = re.search("http://forum.mods.de/SSO.php[^']*", response.read())
+        self.cj.clear()
+
+        if m and m.group(0):
+            # get the cookie
+            response = self.opener.open(m.group(0))
+            for cookie in self.cj:
+                self.cj.clear
+                self.logged_in = True
+                self.opener.addheaders.append(('Cookie', 
+                    '{}={}'.format(cookie.name, cookie.value)))
+                return True
+
+        return False
+
+    def output(self):
+        
+        unread = self.get_unread_count()
+
+        if not unread:
+            return None
+
+        return {'full_text' : '%d new posts in bookmarks' % unread, 
+                'name' : 'modsde',
+                'urgent' : 'true',
+                'color' : self.settings['color']}

wrapper.py

@@ -2,6 +2,7 @@
 # -*- coding: utf-8 -*-
 
 import mailchecker
+import modsde
 from statushandler import I3statusHandler
 
 if __name__ == '__main__':
@@ -33,5 +34,12 @@ if __name__ == '__main__':
     mailchecker = mailchecker.MailChecker(mailsettings)
     status.register_module(mailchecker)
 
+    mdesettings = {
+        'username': "your_username",
+        'password': "your_password"
+    }
+    mde = modsde.ModsDeChecker(mdesettings)
+    status.register_module(mde)
+
     # start the handler
     status.run()