checktheroads-hyperglass/hyperglass/parsing/juniper.py

"""Parse Juniper XML Response to Structured Data."""

# Third Party
import xmltodict

# Project
from hyperglass.log import log
from hyperglass.exceptions import ParsingError, ResponseEmpty
from hyperglass.configuration import params
from hyperglass.parsing.models.juniper import JuniperRoute


def parse_juniper(output):
    """Parse a Juniper BGP XML response."""
    data = {}
    for i, response in enumerate(output):
        try:
            parsed = xmltodict.parse(response, force_list=("rt", "rt-entry"))

            if "rpc-reply" in parsed.keys():
                parsed = parsed["rpc-reply"]["route-information"]["route-table"]
            elif "route-information" in parsed.keys():
                parsed = parsed["route-information"]["route-table"]

            if "rt" not in parsed:
                raise ResponseEmpty(params.messages.no_output)

            validated = JuniperRoute(**parsed)
            serialized = validated.serialize().export_dict()

            if i == 0:
                data.update(serialized)
            else:
                data["routes"].extend(serialized["routes"])

        except xmltodict.expat.ExpatError as err:
            log.critical(str(err))
            raise ParsingError("Error parsing response data")

        except KeyError as err:
            log.critical(f"'{str(err)}' was not found in the response")
            raise ParsingError("Error parsing response data")

    return data