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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
# Copyright (C) 2017 Gabriel Perez-Cerezo <gabriel@gpcf.eu>
# -*- coding: utf-8 -*-
import urllib.request, urllib.error, urllib.parse
from lxml import etree
from lxml.cssselect import CSSSelector
import html5lib
from mensa.base import *
from yapsy.IPlugin import IPlugin
import multiprocessing
import datetime
from yapsy import NormalizePluginNameForModuleName as normalize
mensenliste = {"TU Hardenbergstraße" : ["mensa-tu-hardenbergstra%C3%9Fe", (52.5097684, 13.3259478)],
"TU Marchstraße": ["cafeteria-tu-marchstra%C3%9Fe", ( 52.5166071, 13.3234066)],
"TU Skyline": ["cafeteria-tu-skyline", (52.5128648, 13.3200313)],
"TU Architektur": ["cafeteria-tu-architektur", (52.5137508, 13.3234541)],
"TU Ackerstraße": ["cafeteria-tu-ackerstra%C3%9Fe", (52.5386545, 13.3845294)],
"HU Nord": ["mensa-hu-nord", (52.52816,13.38208)],
"HU Oase Adlershof" : ["mensa-hu-oase-adlershof", ( 52.4293965, 13.5300404)],
"HU Süd": ["mensa-hu-sued", (52.5185929, 13.3928965)],
"HU Spandauer Straße": ["mensa-hu-spandauer-stra%C3%9Fe", (52.52096,13.40258)],
"HU „Jacob und Wilhelm Grimm Zentrum“": ["cafeteria-hu-im-jacob-und-wilhelm-grimm-zentrum", (52.52033,13.39083)],
"FU Herrenhaus Düppel": ["mensa-fu-herrenhaus-d%C3%BCppel", (52.4299794,13.2352233)],
"Mensa FU II Otto-von-Simson-Straße": ["mensa-fu-ii", (52.4531000, 13.2890712)],
}
class Studentenwerk(IPlugin) :
def fetch_page(self, mensa) :
user_agent = 'Mozilla/5.0 (Windows NT 6.1; Win64; x64)'
headers = {'User-Agent': user_agent}
req = urllib.request.Request('https://www.stw.berlin/mensen/%s.html' % mensa, headers=headers)
response = urllib.request.urlopen(req)
the_page = response.read()
open("/tmp/the_page","w").write(str(the_page))
document = html5lib.parse(the_page, treebuilder="lxml")
return document
def register_restaurants (self) :
for h,n in mensenliste.items() :
r = Restaurant(normalize(h), h, self, "dummy", [n[0]], pos=n[1])
register_restaurant(r)
def get_food_items(self, mensa="mensa-tu-hardenbergstra%C3%9Fe", ignore_nudelauswahl=False) :
document = self.fetch_page(mensa)
groupsel = CSSSelector('.splGroupWrapper')
groups = [e for e in groupsel(document)]
fl = []
for i in groups :
try:
name = CSSSelector('.splGroup')(i)[0].text
except:
raise NoMenuError from None
sel = CSSSelector('.splMeal')
meals = [e for e in sel(i)]
for m in meals :
namesel = CSSSelector('.bold')
nm = namesel(m)[0].text
if ignore_nudelauswahl and "Nudelauswahl" in nm :
continue
pricesel = CSSSelector('.col-md-3')
veg = 0
if len(pricesel(m)[0]) >= 2 :
if "15" in pricesel(m)[0][1].attrib["src"] :
veg = 2
elif "1.png" in pricesel(m)[0][1].attrib["src"] :
veg = 1
price = pricesel(m)[-1].text.strip()
fl.append(Food(nm, price, name, veg))
return fl
def get_opening_hours(self, mensa) :
#### Rudiment of a function for getting opening hours. Does NOT work yet due to unknown issues.
doc = self.fetch_page(mensa)
groupsel = CSSSelector('div.col-xs-10')
groups = [e for e in groupsel(doc)]
print(groups)
return doc
# format:
# if __name__ == "__main__":
# pool = multiprocessing.Pool(4)
# k = pool.map(pr_f, list(mensenliste.items()))
# k.sort(key=lambda x: x[0])
# for i in k :
# print(i[1])
|
