blob: b4f97af39cacca8895821eaccfba54acb3ddc964 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
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]
|