refresh Mensa

This commit is contained in:
2023-12-04 21:42:58 +01:00
parent 96a116e240
commit dbcf3cc0ce
4 changed files with 115 additions and 139 deletions

8
app.py
View File

@ -96,8 +96,9 @@ def getRapla():
return redirect(url_for("error", ecode=file + 900))
return redirect(url_for("welcome"))
@login_required
@app.route("/backendpoc/plan", methods=["GET"])
@login_required
def displayRapla():
week = request.args.get("week")
if week:
@ -215,5 +216,10 @@ def logout():
return red
@app.errorhandler(AttributeError)
def handle_anon(e):
return redirect(url_for("login"), code=e)
if __name__ == "__main__":
app.run(host='0.0.0.0', port=2024, debug=True)

View File

@ -1,11 +1,7 @@
import time
import icalendar
import datetime
import requests
import recurring_ical_events
import json
from init import app, db, Meals
from fetchMENSA import getMeals
def getWeek(weekstart: datetime, file: str, showsat: bool):
@ -69,50 +65,3 @@ def daylist(weekstart: datetime, showsat: bool):
return dayl
def getMeals(day: datetime):
print (day)
if day.day < 10:
tag = "0" + str(day.day)
else:
tag = str(day.day)
day = str(day.year) + "-" + str(day.month) + "-" + tag
essen = []
query = Meals.query.filter_by(date=day).all()
if len(query) != 0:
for i in query:
essen += [i.name]
return essen
url = "https://dh-api.paulmartin.cloud/plans/" + day + "?canteens=erzberger"
response = requests.request("GET", url)
response = response.content
jres = json.loads(response.decode("utf-8"))
try:
num = len(jres["data"][0]["lines"])
for i in range(num):
try:
jsmeal = jres["data"][0]["lines"][i]["meals"][0]
except IndexError:
return ["Essen nicht (mehr) verfügbar"]
name = jsmeal["name"]
vegan = jsmeal["classifiers"].count("VG") == 1
schwein = jsmeal["classifiers"].count("S") == 1
if vegan:
veget = True
else:
veget = jsmeal["classifiers"].count("VEG") == 1
if veget:
if name.count("Reibekäse") > 0:
vegan = True
if name != "Tagesdessert":
essen += [name]
mid = int(time.time()*1000) % 100000
neu = Meals(date=day, name=name, id=mid, vegan=vegan, vegetarian=veget, schwein=schwein)
db.session.add(neu)
db.session.commit()
except KeyError:
essen = ["Essen nicht (mehr) verfügbar"]
return essen

107
fetchMENSA.py Normal file
View File

@ -0,0 +1,107 @@
import json
from init import db, Meals, scheduler
import datetime
import requests
import hashlib
import time
nomeal = 'Essen nicht (mehr) verfügbar'
def getMeals(day: datetime):
day = formatDay(day)
essen = []
query = Meals.query.filter_by(date=day).all()
if len(query) != 0:
for i in query:
essen += [i.name]
essen.sort(key=len, reverse=True)
return essen
return getMealsFromAPI(day, dbentry=True)
def getMealsFromAPI(day: str, dbentry: bool = False):
url = "https://dh-api.paulmartin.cloud/plans/" + day + "?canteens=erzberger"
response = requests.request("GET", url)
response = response.content
jres = json.loads(response.decode("utf-8"))
essen = []
try:
num = len(jres["data"][0]["lines"])
for i in range(num):
try:
jsmeal = jres["data"][0]["lines"][i]["meals"]
cont = True
except IndexError:
essen = []
cont = False
if cont:
for e in range(len(jsmeal)):
ji = jsmeal[e]
name = ji["name"]
if pricetofloat(ji["price"]) >= 1.1:
vegan = ji["classifiers"].count("VG") == 1
schwein = ji["classifiers"].count("S") == 1
if vegan:
veget = True
else:
veget = ji["classifiers"].count("VEG") == 1
if veget:
if name.count("Reibekäse") > 0:
vegan = True
essen += [name]
if dbentry:
mid = int(hashlib.sha1((day + name).encode("utf-8")).hexdigest(), 16) % (8 ** 8)
neu = Meals(date=day, name=name, id=mid, vegan=vegan, vegetarian=veget, schwein=schwein)
db.session.add(neu)
db.session.commit()
if not essen:
essen = [nomeal]
except KeyError:
essen = [nomeal]
return essen
def pricetofloat(price: str):
price = price[:-2]
price = price.replace(",", ".")
try:
return float(price)
except ValueError:
return 0
def formatDay(day: datetime):
if day.day < 10:
tag = "0" + str(day.day)
else:
tag = str(day.day)
day = str(day.year) + "-" + str(day.month) + "-" + tag
return day
@scheduler.task('cron', id="refreshMeals", hour='*', day_of_week='*', minute='1-59', week='*', second='30')
def refreshMeals():
print("Aktualisiere Essenspläne...\n")
with scheduler.app.app_context():
table = Meals.query.all()
dates = []
for i in table:
if i.date not in dates:
dates += [i.date]
for i in range(len(dates)):
dates[i] = formatDay(dates[i])
for i in dates:
apinames = getMealsFromAPI(i)
dbmeals = Meals.query.filter_by(date=i).all()
dbnames = []
for m in dbmeals:
dbnames += [m.name]
if set (dbnames) != set(apinames) and nomeal not in apinames:
print (dbnames, apinames)
for n in dbnames:
db.session.delete(Meals.query.filter_by(date=i, name=n).first())
db.session.commit()
getMealsFromAPI(i, True)

View File

@ -1,86 +0,0 @@
import icalendar
import datetime
import requests
import recurring_ical_events
import json
def getWeek(weekstart: datetime, file: str, showsat: bool):
if weekstart == "today":
start_date = datetime.date.today()
else:
start_date = weekstart
start_date -= datetime.timedelta(days=start_date.weekday() % 7)
end_date = start_date + datetime.timedelta(days=7)
with open("calendars/" + file) as f:
calendar = icalendar.Calendar.from_ical(f.read())
events = recurring_ical_events.of(calendar).between(start_date, end_date)
eventl = []
for event in events:
estart = event["DTSTART"].dt
formstart = str(estart.hour) + ":" + str(estart.minute)
eend = event["DTEND"].dt
formend = str(eend.hour) + ":" + str(eend.minute)
forml = [formstart, formend]
for i in range(2):
if len(forml[i]) != 5:
if forml[i][-2] == ":":
forml[i] = forml[i] + "0"
if forml[i][1] == ":":
forml[i] = "0" + forml[i]
formstart = forml[0]
formend = forml[1]
eventdict = {
"start": formstart,
"end": formend,
"dur": str(event["DTEND"].dt - event["DTSTART"].dt)[:-3],
"name": event["SUMMARY"],
"room": event["LOCATION"],
"weekday": estart.weekday(),
"day": estart.day
}
eventl += [eventdict]
return eventl, daylist(start_date, showsat)
shortnames = ["mon", "tue", "wed", "thu", "fri", "sat"]
longnames = ["Montag", "Dienstag", "Mittwoch", "Donnerstag", "Freitag", "Samstag"]
def daylist(weekstart: datetime, showsat: bool):
weekday = weekstart
dayl = []
if showsat:
r = 6
else:
r = 5
for i in range(r):
essen = getMeals(weekday)
dayl += [{
"day": weekday.day,
"short": shortnames[i],
"long": longnames[i],
"mensa": essen
}]
weekday += datetime.timedelta(days=1)
return dayl
def getMeals(day: datetime):
if day.day < 10:
tag = "0" + str(day.day)
else:
tag = str(day.day)
day = str(day.year) + "-" + str(day.month) + "-" + tag
url = "https://dh-api.paulmartin.cloud/plans/" + day + "?canteens=erzberger"
response = requests.request("GET", url)
response = response.content
jres = json.loads(response.decode("utf-8"))
essen = []
try:
num = len(jres["data"][0]["lines"])
for i in range(num):
essen += [jres["data"][0]["lines"][i]["meals"][0]["name"]]
except KeyError:
essen = ["Tag nicht (mehr) verfügbar"]
return essen