def _extract(self, report_type: str, text: str) -> list[str]:
"""Extracts the raw_report element from XML response"""
resp = parsexml(text)
try:
data = resp["response"]["data"]
if data["@num_results"] == "0":
return ""
reports = data[self._targets[report_type]]
except KeyError as key_error:
raise InvalidRequest(
"Could not find report path in response") from key_error
target = "station_id" if report_type == "station" else "raw_text"
return list({r[target] for r in reports})
async def fetch(self, report_type: str, distance: float,
route: list[Coord]) -> list[str]:
"""Fetch reports from the service along a coordinate route"""
target_type = self._type_map.get(report_type, report_type)
if target_type not in self._valid_types:
raise InvalidRequest(
f"{report_type} is not a valid router report type")
flight_path = ";".join(f"{lon},{lat}" for lat, lon in route)
params = {
"dataSource": target_type + "s",
"requestType": "retrieve",
"format": "xml",
"flightPath": f"{distance};{flight_path}",
}
text = await self._call(self.url, params=params)
return self._extract(report_type, text)
def make_err(self, body: str, key: str = 'report path') -> InvalidRequest:
"""
Returns an InvalidRequest exception with formatted error message
"""
msg = f'Could not find {key} in {self.__class__.__name__} response\n'
return InvalidRequest(msg + body)