Kai Fabian
81 lines
3.2 KiB
Python
81 lines
3.2 KiB
Python
import logging
|
|
from datetime import datetime, date
|
|
from stw_potsdam.xml_types.canteen_xml import CanteenMeta, CanteenXML
|
|
from stw_potsdam.xml_types.times_xml import CanteenOpenTimespec, TimesXML
|
|
from stw_potsdam.xml_types.meal_xml import MealXML
|
|
|
|
|
|
class SWPWebspeiseplanParser:
|
|
"""Class method to parse SWP_Webspeiseplan."""
|
|
|
|
def __init__(self) -> None:
|
|
"""Init SWPWebspeiseplanParser object."""
|
|
logging.basicConfig()
|
|
self.logger = logging.getLogger(__name__)
|
|
|
|
def parse_canteen_meta_times(self, outlet: dict):
|
|
"""Parse the outlet data from outlet.
|
|
|
|
Args:
|
|
outlet (dict): [description]
|
|
"""
|
|
self.logger.debug("parse_canteen_meta_times(): %s", outlet["name"])
|
|
addr_info = outlet["addressInfo"]
|
|
meta = {
|
|
"name": outlet["name"],
|
|
"address": f'{addr_info["street"]}, {addr_info["postalCode"]} '
|
|
+ f'{addr_info["city"]}',
|
|
"city": addr_info["city"],
|
|
"phone": outlet["contactInfo"][0]["phone"],
|
|
"email": outlet["contactInfo"][0]["email"],
|
|
"availability": outlet["isPublic"]
|
|
}
|
|
|
|
if outlet["positionInfo"]:
|
|
meta["location"] = (
|
|
outlet["positionInfo"]["longitude"],
|
|
outlet["positionInfo"]["latitude"],
|
|
)
|
|
canteen_meta = CanteenMeta(**meta)
|
|
weekday_dict = {
|
|
# this approach only lists the first (valid) opening time,
|
|
# since OpenMensa does not support multiple time specs
|
|
# (yet).
|
|
"monday": outlet['moZeit1'] or outlet['moZeit2'],
|
|
"tuesday": outlet['diZeit1'] or outlet['diZeit2'],
|
|
"wednesday": outlet['miZeit1'] or outlet['miZeit2'],
|
|
"thursday": outlet['doZeit1'] or outlet['doZeit2'],
|
|
"friday": outlet['frZeit1'] or outlet['frZeit2'],
|
|
"saturday": outlet['saZeit1'] or outlet['saZeit2'],
|
|
"sunday": outlet['soZeit1'] or outlet['soZeit2'],
|
|
}
|
|
|
|
canteen_times = TimesXML({
|
|
k: CanteenOpenTimespec(v) for k, v in weekday_dict.items()
|
|
})
|
|
canteen = CanteenXML(canteen_meta, canteen_times)
|
|
return canteen
|
|
|
|
def parse_meals(
|
|
self, menu_data, meal_categories
|
|
) -> list[tuple[date, str, MealXML]]:
|
|
"""Parse the menu and adds it to the builder."""
|
|
meals = []
|
|
for menu in menu_data:
|
|
for meal_data in menu["speiseplanGerichtData"]:
|
|
info = meal_data["speiseplanAdvancedGericht"]
|
|
additional_info = meal_data["zusatzinformationen"]
|
|
price = {
|
|
"student": additional_info["mitarbeiterpreisDecimal2"],
|
|
"employee": additional_info["price3Decimal2"],
|
|
"other": additional_info["gaestepreisDecimal2"],
|
|
}
|
|
meal = MealXML(name=info["gerichtname"], price=price)
|
|
day = datetime.fromisoformat(info["datum"]).date()
|
|
category = meal_categories[info["gerichtkategorieID"]]["name"]
|
|
meals.append(
|
|
{"day": day, "category": category, "meal": meal}
|
|
)
|
|
self.logger.debug("parse_meals(): %s meals parsed", len(meals))
|
|
return meals
|