Modularized program

1 files changed, 44 insertions, 0 deletions

diff --git a/backends/mathemensa.py b/backends/mathemensa.py
new file mode 100644
index 0000000..b98ab1d
--- /dev/null
+++ b/backends/mathemensa.py
@@ -0,0 +1,44 @@
+# -*- 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 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) :
+        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 k in sel(document)[0][weekday] :
+            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