Hadrian Burkhardt
62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
"""Potsdam parser/provider implementation."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from openmensa_parsers.config import Canteen
|
|
from openmensa_parsers.parsers.base import BaseOpenMensaParser, FeedDefinition
|
|
from openmensa_parsers.webspeiseplan_api import (
|
|
WebspeiseplanAPI,
|
|
WebspeiseplanData,
|
|
)
|
|
from openmensa_parsers.webspeiseplan_parser import WebspeiseplanParser
|
|
from openmensa_parsers.xml_types.canteen_xml import CanteenXML
|
|
|
|
|
|
class PotsdamParser(BaseOpenMensaParser):
|
|
"""Parser for Studentenwerk Potsdam's Webspeiseplan source."""
|
|
|
|
id = "potsdam"
|
|
BASE_URL = "https://swp.webspeiseplan.de"
|
|
feed = FeedDefinition(source=BASE_URL)
|
|
|
|
def __init__(
|
|
self,
|
|
api: WebspeiseplanAPI | None = None,
|
|
parser: WebspeiseplanParser | None = None,
|
|
) -> None:
|
|
"""Initialize the Potsdam parser with fetch and parse helpers."""
|
|
self.api = WebspeiseplanAPI(self.BASE_URL) if api is None else api
|
|
self.parser = WebspeiseplanParser() if parser is None else parser
|
|
self.logger = logging.getLogger(__name__)
|
|
|
|
def fetch(self) -> WebspeiseplanData:
|
|
"""Download all data required by the Potsdam parser."""
|
|
return self.api.fetch_all()
|
|
|
|
def parse(
|
|
self,
|
|
config: dict[str, Canteen],
|
|
raw_data: WebspeiseplanData,
|
|
) -> dict[str, CanteenXML]:
|
|
"""Convert Potsdam Webspeiseplan data into canteen structures."""
|
|
parsed: dict[str, CanteenXML] = {}
|
|
for canteen_key, configured_canteen in config.items():
|
|
source_name = configured_canteen.name
|
|
if source_name not in raw_data.outlets:
|
|
self.logger.warning("%s not found in keys", source_name)
|
|
continue
|
|
|
|
outlet = dict(raw_data.outlets[source_name])
|
|
menus = raw_data.menus[source_name]
|
|
categories = raw_data.meal_categories[source_name]
|
|
locations = raw_data.locations[source_name]
|
|
outlet["isPublic"] = locations["isPublic"]
|
|
|
|
canteen = self.parser.parse_canteen_meta_times(outlet)
|
|
for meal_data in self.parser.parse_meals(menus, categories):
|
|
canteen.add_meal(**meal_data)
|
|
parsed[canteen_key] = canteen
|
|
return parsed
|