Hadrian Burkhardt
130 lines
4.7 KiB
Python
130 lines
4.7 KiB
Python
import logging
|
|
import re
|
|
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
|
|
|
|
|
|
EURO_PRICE_PATTERN = re.compile(r"(\d+(?:[,.]\d{1,2})?)\s*€")
|
|
|
|
|
|
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_price(self, value):
|
|
if value in (None, "", {}):
|
|
return None
|
|
return float(str(value).replace(",", "."))
|
|
|
|
def _parse_embedded_prices(
|
|
self, name: str, price: dict[str, float | None]
|
|
) -> tuple[str, dict[str, float | None]]:
|
|
if any(price.values()):
|
|
return name, price
|
|
|
|
matches = EURO_PRICE_PATTERN.findall(name)
|
|
if len(matches) < 2:
|
|
return name, price
|
|
|
|
parsed = [self._parse_price(match) for match in matches]
|
|
if len(parsed) >= 3:
|
|
price = {
|
|
"student": parsed[0],
|
|
"employee": parsed[1],
|
|
"other": parsed[2],
|
|
}
|
|
elif "Stud" in name and ("Gäste" in name or "Gaeste" in name):
|
|
price = {
|
|
"student": parsed[0],
|
|
"employee": price["employee"],
|
|
"other": parsed[1],
|
|
}
|
|
else:
|
|
return name, price
|
|
|
|
name = EURO_PRICE_PATTERN.sub("", name)
|
|
name = re.sub(r"\s*/\s*", " ", name)
|
|
name = re.sub(r"\s+", " ", name).strip()
|
|
return name, price
|
|
|
|
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": self._parse_price(
|
|
additional_info["mitarbeiterpreisDecimal2"]
|
|
),
|
|
"employee": self._parse_price(
|
|
additional_info["price3Decimal2"]
|
|
),
|
|
"other": self._parse_price(
|
|
additional_info["gaestepreisDecimal2"]
|
|
),
|
|
}
|
|
name, price = self._parse_embedded_prices(
|
|
info["gerichtname"], price
|
|
)
|
|
meal = MealXML(name=name, 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
|