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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
# -*- coding: utf-8 -*-
from base import ModuleBase
import re
from datetime import date, timedelta
import urllib
class Module(ModuleBase):
def __init__(self, manager):
ModuleBase.__init__(self, manager)
def commands(self):
return [('!mensa [tomorrow]', 'Zeigt das aktuelle Menü der Mensa der Uni Lübeck an (oder das des nächsten Tages)')]
def groupchat(self, room, nick, text, handler):
if not re.match(r'!mensa(?:\W|\Z)', text):
return
tomorrow = re.match(r'!mensa\W+tomorrow(\W|\Z)', text)
daystr = 'heute'
day = date.today()
if tomorrow:
day += timedelta(1)
daystr = 'morgen'
datestr = day.strftime('%d.%m.</div>')
try:
f = urllib.urlopen(self.manager.config['mensa_url'], 'r')
except IOError:
handler.reply('Hmm, es scheint gerade keine Daten über das Mensa-Menü zu geben.')
return
found = False
line = f.readline()
while line:
if re.search(datestr, line):
found = True
break
line = f.readline()
if not found:
f.close()
handler.reply('%s kein Mensa-Menü... ist vielleicht Wochenende?' % daystr)
return
meals = []
r = re.compile(r'<td class="schrift_gerichte">([^<]*)</td>')
line = f.readline()
while line:
match = r.search(line)
if match:
meals.append(match.group(1))
if len(meals) >= 4:
break
if re.search('<td rowspan="3" class="schrift_fett"><div>', line):
break
line = f.readline()
prices = []
r = re.compile(r'<td class="schrift_preise">([^<]*)</td>')
line = f.readline()
while line:
match = r.search(line)
if match:
prices.append(match.group(1))
if len(prices) >= 4:
break
if re.search('<td rowspan="3" class="schrift_fett"><div>', line):
break
line = f.readline()
f.close()
if len(meals) < 4 or len(prices) < 4:
handler.reply('%s kein Mensa-Menü... ist vielleicht Wochenende?' % daystr)
return
def unescape(s):
s = s.replace(' ', '')
s = s.replace('€', '€')
s = s.replace('ä', 'ä')
s = s.replace('Ä', 'Ä')
s = s.replace('ö', 'ö')
s = s.replace('Ö', 'Ö')
s = s.replace('ü', 'ü')
s = s.replace('Ü', 'Ü')
s = s.replace('ß', 'ß')
s = s.replace('&', '&')
s = s.replace('<', '<')
s = s.replace('>', '>')
s = s.replace('"', '"')
s = re.sub('\s+', ' ', s)
return s
meals = map(unescape, meals)
prices = map(unescape, prices)
reply = 'Mensa-Menü für den %s\n\n' % day.strftime('%d.%m.%y')
if meals[0]:
reply += 'Eintopf: %s (%s)\n' % (meals[0], prices[0])
if meals[1]:
reply += 'Hauptgericht 1: %s (%s)\n' % (meals[1], prices[1])
if meals[2]:
reply += 'Hauptgericht 2: %s (%s)\n' % (meals[2], prices[2])
if meals[3]:
reply += 'Vegetarisches Hauptgericht: %s (%s)\n' % (meals[3], prices[3])
handler.reply(reply.strip())
|