1 files changed, 49 insertions, 0 deletions
diff --git a/mensa/backends/mathemensa.py b/mensa/backends/mathemensa.py
new file mode 100644
index 0000000..8f66805
--- /dev/null
+++ b/mensa/backends/mathemensa.py
@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+
+import urllib.request, urllib.error, urllib.parse
+from lxml import etree
+from lxml.cssselect import CSSSelector
+import re
+
+import  xml.sax.saxutils as saxutils
+import html5lib
+import datetime
+from mensa.base import *
+from yapsy.IPlugin import IPlugin
+
+class Mathemensa(IPlugin) :
+    def register_restaurants(self) :
+        r = Restaurant("Mathemensa", "Mathemensa", self, "dummy")
+        register_restaurant(r)
+        
+    def get_food_items(self, **kwargs) :
+        weekdays = ["MONTAG", "DIENSTAG", "MITTWOCH", "DONNERSTAG", "FREITAG"]
+        weekday = datetime.datetime.today().weekday()
+        if weekday > 4 :
+            print("Error: No food today")
+            return ""
+        user_agent = 'Mozilla/5.0 (Windows NT 6.1; Win64; x64)'
+        headers = {'User-Agent': user_agent}
+
+        req = urllib.request.Request('http://personalkantine.personalabteilung.tu-berlin.de/', headers=headers)
+        response = urllib.request.urlopen(req)
+        the_page = response.read()
+        document = html5lib.parse(the_page, treebuilder="lxml")
+        sel = CSSSelector('.Menu__accordion')
+        fl = []
+        for i in sel(document)[0] :
+            h2sel = CSSSelector('h2')
+            if not weekdays[weekday] in i[0].text.upper() :
+                continue
+            for k in i :
+                if k.tag.endswith("ul") :
+                    for j in k :
+                        price = j[1].text
+                        st = str(etree.tostring(j)).split("\\n")[2].split("<")[0]
+                        name = j[0].text + ", " + etree.fromstring("<p>%s</p>"%st).text.strip() # really extremely dirty hack
+                        veg = 0
+                        
+                        if "(v)" in name or "Gemüseplatte" in name :
+                            veg = 1
+                        fl.append(Food(name, price, "Menü", veg))
+        return fl