diff options
author | Gabriel Pérez-Cerezo <gabriel@gpcf.eu> | 2017-09-22 16:16:03 +0200 |
---|---|---|
committer | Gabriel Pérez-Cerezo <gabriel@gpcf.eu> | 2017-09-22 16:16:03 +0200 |
commit | 21ee7c5fde7e87de751dbfc5008e8ae2e41a4238 (patch) | |
tree | 0585e6c80eaa6bf54d51a1983548791a8329f9ce /mathemensa.py | |
download | mensa-21ee7c5fde7e87de751dbfc5008e8ae2e41a4238.tar.gz mensa-21ee7c5fde7e87de751dbfc5008e8ae2e41a4238.tar.bz2 mensa-21ee7c5fde7e87de751dbfc5008e8ae2e41a4238.zip |
First working program
Diffstat (limited to 'mathemensa.py')
-rw-r--r-- | mathemensa.py | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/mathemensa.py b/mathemensa.py new file mode 100644 index 0000000..b4f97af --- /dev/null +++ b/mathemensa.py @@ -0,0 +1,45 @@ +# -*- coding: utf-8 -*- + +import urllib2 +from lxml import etree +from lxml.cssselect import CSSSelector + +import xml.sax.saxutils as saxutils +import html5lib +import datetime +from common import Food + +def get_food_items() : + 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 = urllib2.Request('http://personalkantine.personalabteilung.tu-berlin.de/', headers=headers) + response = urllib2.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 + name = j[0].text + etree.fromstring("<p>"+etree.tostring(j).split("\n")[2].split("<")[0]+"</p>").text.strip() # really extremely dirty hack + veg = 0 + if "(v)" in name or u"Gemüseplatte" in name : + veg = 1 + fl.append(Food(name, price, u"Menü", veg)) + return fl +if __name__ == "__main__": + food = get_food_items() + cat = [] + vegkeys = [ "", "Vegetarian", "Vegan" ] + for i in food: + if not i.category in cat : + cat.append(i.category) + if not i.category == None : + print i.category + print "\t" + i.name.ljust(80) + "\t"+ i.price.ljust(20) + vegkeys[i.veggie] |