diff --git a/brian_polling_manager/interface_stats/cli.py b/brian_polling_manager/interface_stats/cli.py index 2203647d03ef66dbfca015d2c5ed2b10e504e5a9..5829c07e12ea5051ec8adacee712a3a04d35830f 100644 --- a/brian_polling_manager/interface_stats/cli.py +++ b/brian_polling_manager/interface_stats/cli.py @@ -85,9 +85,13 @@ def get_netconf(router_name, vendor=Vendor.JUNIPER, **kwargs): source_dir = _APP_CONFIG_PARAMS.get("testing", {}).get("netconf-source-dir") if source_dir: if vendor == Vendor.JUNIPER: - return juniper.get_netconf_interface_info_from_source_dir(router_name, source_dir) + return juniper.get_netconf_interface_info_from_source_dir( + router_name, source_dir + ) else: - return nokia.get_netconf_interface_info_from_source_dir(router_name, source_dir) + return nokia.get_netconf_interface_info_from_source_dir( + router_name, source_dir + ) ssh_params = _APP_CONFIG_PARAMS[vendor.value] if vendor == Vendor.JUNIPER: @@ -97,7 +101,10 @@ def get_netconf(router_name, vendor=Vendor.JUNIPER, **kwargs): def validate_router_hosts( - hostnames: List[str], vendor: Vendor, inprov_hosts=None, load_interfaces_=load_interfaces + hostnames: List[str], + vendor: Vendor, + inprov_hosts=None, + load_interfaces_=load_interfaces, ): if inprov_hosts is None: return True @@ -207,6 +214,8 @@ def main( logger.exception( f"Error while processing {vendor_str} {router}", exc_info=e ) + error_count += 1 + if raise_errors: raise