capture-test-data-nokia.py

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

ROUTERS = [
    'rt0.lon.uk.lab.office.geant.net',
    'rt0.ams.nl.lab.office.geant.net'
]

SSH_PARAMS = {
    'ssh_config': '~/.ssh/config.d/routers-jump'
}

def load_estate_interface_info():

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

        file = pathlib.Path(__file__).parent / f"{fqdn}-ports.xml"
        file.write_bytes(etree.tostring(raw_counter_docs['port']))

        file = pathlib.Path(__file__).parent / f"{fqdn}-lags.xml"
        file.write_bytes(etree.tostring(raw_counter_docs['lag']))


if __name__ == "__main__":
    load_estate_interface_info()