capture-test-data-nokia.py

import pathlib
from brian_polling_manager.interface_stats.vendors import nokia
from lxml import etree
import logging

logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.DEBUG)
ncclient_logger = logging.getLogger("ncclient")
ncclient_logger.level = logging.WARNING

ROUTERS = [
    "rt0.ams.nl.geant.net",
]

SSH_PARAMS = {
    "ssh_config": "~/.ssh/config",
    "hostkey_verify": False,
    "username": "inprov",
    "password": "<password>",
}


def load_estate_interface_info():

    for fqdn in ROUTERS:
        logger.info(fqdn)
        raw_counter_docs = nokia.get_netconf_interface_info(
            router_name=fqdn, ssh_params=SSH_PARAMS
        )

        assert set(raw_counter_docs.keys()) == {"port", "lag", "router-interface"}
        for key, doc in raw_counter_docs.items():
            file = pathlib.Path(__file__).parent / f"{fqdn}-{key}s.xml"
            file.write_bytes(etree.tostring(doc))


if __name__ == "__main__":
    load_estate_interface_info()