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
46
47
48
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 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
|