Hadrian Burkhardt
119 lines
3.9 KiB
Python
119 lines
3.9 KiB
Python
import logging
|
|
from datetime import datetime
|
|
from stw_potsdam.builder import Builder
|
|
from stw_potsdam.swp_webspeiseplan_api import SWP_Webspeiseplan_API
|
|
|
|
|
|
class SWP_Webspeiseplan_Parser:
|
|
"""Class method to parse SWP_Webspeiseplan."""
|
|
|
|
def __init__(
|
|
self,
|
|
menu_data: list[dict],
|
|
meal_categories: list[dict],
|
|
outlet_data: dict,
|
|
url: str,
|
|
):
|
|
"""Initialize the parser .
|
|
|
|
Args:
|
|
menu_data (list[dict]): [description]
|
|
meal_categories (list[dict]): [description]
|
|
outlet_data (dict): [description]
|
|
url (str): [description]
|
|
"""
|
|
logging.basicConfig()
|
|
self.logger = logging.getLogger(__name__)
|
|
self.menu_data = menu_data
|
|
self.meal_categories = meal_categories
|
|
self.outlet_data = outlet_data
|
|
self.url = url
|
|
self._builder = Builder()
|
|
self.__parse_canteen(outlet_data)
|
|
self.__parse_feed()
|
|
self.__parse_meals()
|
|
|
|
def __parse_canteen(self, outlet: dict):
|
|
"""Parse the outlet data from outlet.
|
|
|
|
Args:
|
|
outlet (dict): [description]
|
|
"""
|
|
canteen = self._builder
|
|
canteen.name = outlet["name"]
|
|
canteen.address = (
|
|
outlet["addressInfo"]["street"],
|
|
outlet["addressInfo"]["postalCode"],
|
|
outlet["addressInfo"]["city"],
|
|
)
|
|
canteen.city = outlet["addressInfo"]["city"]
|
|
canteen.phone = outlet["contactInfo"][0]["phone"]
|
|
canteen.email = outlet["contactInfo"][0]["email"]
|
|
if outlet["positionInfo"]:
|
|
canteen.location = (
|
|
outlet["positionInfo"]["longitude"],
|
|
outlet["positionInfo"]["latitude"],
|
|
)
|
|
|
|
# TODO: availability via locations isPublic
|
|
|
|
times = {
|
|
"monday": f"{outlet['moZeit1']}, {outlet['moZeit2']}",
|
|
"tuesday": f"{outlet['diZeit1']}, {outlet['diZeit2']}",
|
|
"wednesday": f"{outlet['miZeit1']}, {outlet['miZeit2']}",
|
|
"thursday": f"{outlet['doZeit1']}, {outlet['doZeit2']}",
|
|
"friday": f"{outlet['frZeit1']}, {outlet['frZeit2']}",
|
|
"saturday": f"{outlet['saZeit1']}, {outlet['saZeit2']}",
|
|
"sunday": f"{outlet['soZeit1']}, {outlet['soZeit2']}",
|
|
}
|
|
|
|
times = {
|
|
k: v.replace("None, None", "")
|
|
.replace("None,", "")
|
|
.replace(", None", "")
|
|
for k, v in times.items()
|
|
}
|
|
|
|
canteen.times = times
|
|
|
|
def __parse_feed(self):
|
|
"""Parse feed and set feed."""
|
|
feed = {
|
|
"name": "full",
|
|
"priority": 0,
|
|
"hour": "8-14",
|
|
"retry": "30 1",
|
|
"url": self.url,
|
|
"source": SWP_Webspeiseplan_API.URL_BASE,
|
|
}
|
|
self._builder.feed = feed
|
|
|
|
def __parse_meals(self):
|
|
"""Parse the menu and adds it to the builder."""
|
|
for menu in self.menu_data:
|
|
for meal in menu["speiseplanGerichtData"]:
|
|
info = meal["speiseplanAdvancedGericht"]
|
|
date = datetime.fromisoformat(info["datum"]).date()
|
|
additional_info = meal["zusatzinformationen"]
|
|
self._builder.add_meal(
|
|
date=date,
|
|
category=self.meal_categories[info["gerichtkategorieID"]][
|
|
"name"
|
|
],
|
|
name=info["gerichtname"],
|
|
prices={
|
|
"student": additional_info["mitarbeiterpreisDecimal2"],
|
|
"employee": additional_info["price3Decimal2"],
|
|
"other": additional_info["gaestepreisDecimal2"],
|
|
},
|
|
)
|
|
|
|
@property
|
|
def xml_feed(self):
|
|
"""Return the XML string of the builder.
|
|
|
|
Returns:
|
|
[type]: [description]
|
|
"""
|
|
return self._builder.toXML()
|