scottk/wlan-lanforge-scripts
1
0
Fork 0
mirror of https://github.com/Telecominfraproject/wlan-lanforge-scripts.git synced 2025-11-06 21:58:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

gen_cxprofile.py : changed prints to logger

Browse Source 
Signed-off-by: Chuck SmileyRekiere <chuck.smileyrekiere@candelatech.com>
This commit is contained in:
Chuck SmileyRekiere
2022-01-28 11:47:13 -07:00
committed by shivam
parent 22a4473c7d
commit eea68062c0
1 changed files with 46 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
py-json/gen_cxprofile.py
View File

@@ -8,6 +8,8 @@ import pandas as pd
import time import time
import datetime import datetime
import json import json
import logging
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../"))) sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
@@ -15,6 +17,8 @@ lfcli_base = importlib.import_module("py-json.LANforge.lfcli_base")
LFCliBase = lfcli_base.LFCliBase LFCliBase = lfcli_base.LFCliBase
pandas_extensions = importlib.import_module("py-json.LANforge.pandas_extensions") pandas_extensions = importlib.import_module("py-json.LANforge.pandas_extensions")
logger = logging.getLogger(__name__)
class GenCXProfile(LFCliBase): class GenCXProfile(LFCliBase):
def __init__(self, lfclient_host, lfclient_port, local_realm, debug_=False): def __init__(self, lfclient_host, lfclient_port, local_realm, debug_=False):
@@ -41,11 +45,12 @@ class GenCXProfile(LFCliBase):
if self.type == "lfping": if self.type == "lfping":
if self.dest and self.interval: if self.dest and self.interval:
self.cmd = "%s -i %s -I %s %s" % (self.type, self.interval, sta_name, self.dest) self.cmd = "%s -i %s -I %s %s" % (self.type, self.interval, sta_name, self.dest)
# print(self.cmd)
else: else:
logger.critical("Please ensure dest and interval have been set correctly")
raise ValueError("Please ensure dest and interval have been set correctly") raise ValueError("Please ensure dest and interval have been set correctly")
elif self.type == "generic": elif self.type == "generic":
if self.cmd == "": if self.cmd == "":
logger.critical("Please ensure cmd has been set correctly")
raise ValueError("Please ensure cmd has been set correctly") raise ValueError("Please ensure cmd has been set correctly")
elif self.type == "speedtest": elif self.type == "speedtest":
self.cmd = "vrf_exec.bash %s speedtest-cli --json --share" % sta_name self.cmd = "vrf_exec.bash %s speedtest-cli --json --share" % sta_name
@@ -60,36 +65,40 @@ class GenCXProfile(LFCliBase):
self.cmd = "./scripts/lf_curl.sh -p %s -i AUTO -o %s -n %s -d %s" % \ self.cmd = "./scripts/lf_curl.sh -p %s -i AUTO -o %s -n %s -d %s" % \
(sta_name, self.file_output, self.loop_count, self.dest) (sta_name, self.file_output, self.loop_count, self.dest)
else: else:
logger.critical("Please ensure file_output has been set correctly")
raise ValueError("Please ensure file_output has been set correctly") raise ValueError("Please ensure file_output has been set correctly")
else: else:
logger.critical("Unknown command type")
raise ValueError("Unknown command type") raise ValueError("Unknown command type")
def start_cx(self): def start_cx(self):
print("Starting CXs...") logger.info("Starting CXs...")
# print(self.created_cx) logger.debug(self.created_cx)
# print(self.created_endp) logger.debug(self.created_endp)
for cx_name in self.created_cx: for cx_name in self.created_cx:
self.json_post("/cli-json/set_cx_state", { self.json_post("/cli-json/set_cx_state", {
"test_mgr": "default_tm", "test_mgr": "default_tm",
"cx_name": cx_name, "cx_name": cx_name,
"cx_state": "RUNNING" "cx_state": "RUNNING"
}, debug_=self.debug) }, debug_=self.debug)
# this is for a visual affect someone watching the screen, leave as print
print(".", end='') print(".", end='')
print("") print("")
def stop_cx(self): def stop_cx(self):
print("Stopping CXs...") logger.info("Stopping CXs...")
for cx_name in self.created_cx: for cx_name in self.created_cx:
self.json_post("/cli-json/set_cx_state", { self.json_post("/cli-json/set_cx_state", {
"test_mgr": "default_tm", "test_mgr": "default_tm",
"cx_name": cx_name, "cx_name": cx_name,
"cx_state": "STOPPED" "cx_state": "STOPPED"
}, debug_=self.debug) }, debug_=self.debug)
# this is for a visual affect someone watching the screen, leave as print
print(".", end='') print(".", end='')
print("") print("")
def cleanup(self): def cleanup(self):
print("Cleaning up cxs and endpoints") logger.info("Cleaning up cxs and endpoints")
for cx_name in self.created_cx: for cx_name in self.created_cx:
req_url = "cli-json/rm_cx" req_url = "cli-json/rm_cx"
data = { data = {
@@ -124,11 +133,13 @@ class GenCXProfile(LFCliBase):
if self.type == "lfping": if self.type == "lfping":
if self.dest and self.interval: if self.dest and self.interval:
self.cmd = "%s -i %s -I %s %s" % (self.type, self.interval, sta_name, dest) self.cmd = "%s -i %s -I %s %s" % (self.type, self.interval, sta_name, dest)
# print(self.cmd) logger.debug(self.cmd)
else: else:
logger.critical("Please ensure dest and interval have been set correctly")
raise ValueError("Please ensure dest and interval have been set correctly") raise ValueError("Please ensure dest and interval have been set correctly")
elif self.type == "generic": elif self.type == "generic":
if self.cmd == "": if self.cmd == "":
logger.critical("Please ensure cmd has been set correctly")
raise ValueError("Please ensure cmd has been set correctly") raise ValueError("Please ensure cmd has been set correctly")
elif self.type == "speedtest": elif self.type == "speedtest":
self.cmd = "vrf_exec.bash %s speedtest-cli --json --share" % sta_name self.cmd = "vrf_exec.bash %s speedtest-cli --json --share" % sta_name
@@ -140,8 +151,10 @@ class GenCXProfile(LFCliBase):
self.cmd = "./scripts/lf_curl.sh -p %s -i AUTO -o %s -n %s -d %s" % \ self.cmd = "./scripts/lf_curl.sh -p %s -i AUTO -o %s -n %s -d %s" % \
(sta_name, self.file_output, self.loop_count, self.dest) (sta_name, self.file_output, self.loop_count, self.dest)
else: else:
logger.critical("Please ensure file_output has been set correctly")
raise ValueError("Please ensure file_output has been set correctly") raise ValueError("Please ensure file_output has been set correctly")
else: else:
logger.critical("Unknown command type")
raise ValueError("Unknown command type") raise ValueError("Unknown command type")
def create_gen(self, sta_port, dest, add, sleep_time=.5, debug_=False, suppress_related_commands_=None): def create_gen(self, sta_port, dest, add, sleep_time=.5, debug_=False, suppress_related_commands_=None):
@@ -168,7 +181,7 @@ class GenCXProfile(LFCliBase):
print(endp_tpls) print(endp_tpls)
elif type(sta_port) == list: elif type(sta_port) == list:
for port_name in sta_port: for port_name in sta_port:
print("hello............", sta_port) logger.info("hello............{sta_port}".format(sta_port=sta_port))
for i in range(0, 5): for i in range(0, 5):
port_info = self.local_realm.name_to_eid(port_name) port_info = self.local_realm.name_to_eid(port_name)
try: try:
@@ -176,7 +189,8 @@ class GenCXProfile(LFCliBase):
shelf = port_info[0] shelf = port_info[0]
name = port_info[2] name = port_info[2]
except ValueError: except ValueError:
raise ValueError("Unexpected name for port_name %s" % port_name) logger.critical("Unexpected name for port_name {port_name}".format(port_name=port_name))
raise ValueError("Unexpected name for port_name {port_name}".format(port_name=port_name))
# this naming convention follows what you see when you use # this naming convention follows what you see when you use
# lf_firemod.pl --action list_endp after creating a generic endpoint # lf_firemod.pl --action list_endp after creating a generic endpoint
@@ -184,8 +198,7 @@ class GenCXProfile(LFCliBase):
gen_name_b = "D_%s-%s" % (self.name_prefix, name) + "_" + str(i) + add gen_name_b = "D_%s-%s" % (self.name_prefix, name) + "_" + str(i) + add
endp_tpls.append((shelf, resource, name, gen_name_a, gen_name_b)) endp_tpls.append((shelf, resource, name, gen_name_a, gen_name_b))
# exit(1) logger.info(endp_tpls)
print(endp_tpls)
for endp_tpl in endp_tpls: for endp_tpl in endp_tpls:
shelf = endp_tpl[0] shelf = endp_tpl[0]
@@ -258,7 +271,7 @@ class GenCXProfile(LFCliBase):
self.created_endp.append(gen_name_b) self.created_endp.append(gen_name_b)
time.sleep(sleep_time) time.sleep(sleep_time)
print(self.created_cx) logger.info(self.created_cx)
for data in post_data: for data in post_data:
url = "/cli-json/add_cx" url = "/cli-json/add_cx"
@@ -423,6 +436,7 @@ class GenCXProfile(LFCliBase):
else: else:
return False, v['name'] return False, v['name']
# TODO monitor is broken
def monitor(self, def monitor(self,
duration_sec=60, duration_sec=60,
monitor_interval_ms=1, monitor_interval_ms=1,
@@ -442,30 +456,41 @@ class GenCXProfile(LFCliBase):
duration_sec = self.parse_time(duration_sec).seconds duration_sec = self.parse_time(duration_sec).seconds
except ValueError: except ValueError:
if not duration_sec or (duration_sec <= 1): if not duration_sec or (duration_sec <= 1):
logger.critical("GenCXProfile::monitor wants duration_sec > 1 second")
raise ValueError("GenCXProfile::monitor wants duration_sec > 1 second") raise ValueError("GenCXProfile::monitor wants duration_sec > 1 second")
if duration_sec <= monitor_interval_ms: if duration_sec <= monitor_interval_ms:
logger.critical("GenCXProfile::monitor wants duration_sec > monitor_interval")
raise ValueError("GenCXProfile::monitor wants duration_sec > monitor_interval") raise ValueError("GenCXProfile::monitor wants duration_sec > monitor_interval")
if not report_file: if not report_file:
logger.critical("Monitor requires an output file to be defined")
raise ValueError("Monitor requires an output file to be defined") raise ValueError("Monitor requires an output file to be defined")
if not systeminfopath: if not systeminfopath:
logger.critical("Monitor requires a system info path to be defined")
raise ValueError("Monitor requires a system info path to be defined") raise ValueError("Monitor requires a system info path to be defined")
if not created_cx: if not created_cx:
logger.critical("Monitor needs a list of Layer 3 connections")
raise ValueError("Monitor needs a list of Layer 3 connections") raise ValueError("Monitor needs a list of Layer 3 connections")
if not monitor_interval_ms or (monitor_interval_ms < 1): if not monitor_interval_ms or (monitor_interval_ms < 1):
logger.critical("GenCXProfile::monitor wants monitor_interval >= 1 second")
raise ValueError("GenCXProfile::monitor wants monitor_interval >= 1 second") raise ValueError("GenCXProfile::monitor wants monitor_interval >= 1 second")
if not generic_cols: if not generic_cols:
logger.critical("GenCXProfile::monitor wants a list of column names to monitor")
raise ValueError("GenCXProfile::monitor wants a list of column names to monitor") raise ValueError("GenCXProfile::monitor wants a list of column names to monitor")
if output_format: if output_format:
if output_format.lower() != report_file.split('.')[-1]: if output_format.lower() != report_file.split('.')[-1]:
logger.critical(
'Filename {report_file} has an extension that does not match output format {output_format}'.format(
report_file=report_file, output_format=output_format))
raise ValueError( raise ValueError(
'Filename %s has an extension that does not match output format %s .' % (report_file, output_format)) 'Filename {report_file} has an extension that does not match output format {output_format}'.format(
report_file=report_file, output_format=output_format))
else: else:
output_format = report_file.split('.')[-1] output_format = report_file.split('.')[-1]
# default save to csv first # default save to csv first
if report_file.split('.')[-1] != 'csv': if report_file.split('.')[-1] != 'csv':
report_file = report_file.replace(str(output_format), 'csv', 1) report_file = report_file.replace(str(output_format), 'csv', 1)
print("Saving rolling data into..." + str(report_file)) logger.info("Saving rolling data into..." + str(report_file))
# ================== Step 1, set column names and header row # ================== Step 1, set column names and header row
generic_cols = [self.replace_special_char(x) for x in generic_cols] generic_cols = [self.replace_special_char(x) for x in generic_cols]
@@ -511,14 +536,13 @@ class GenCXProfile(LFCliBase):
# for x in range(0,int(round(iterations,0))): # for x in range(0,int(round(iterations,0))):
initial_starttime = datetime.datetime.now() initial_starttime = datetime.datetime.now()
print("Starting Test...") logger.info("Starting Test...")
while datetime.datetime.now() < end_time: while datetime.datetime.now() < end_time:
passes = 0 passes = 0
expected_passes = 0 expected_passes = 0
time.sleep(15) time.sleep(15)
result = False result = False
cur_time = datetime.datetime.now()
if self.type == "lfping": if self.type == "lfping":
result = self.choose_ping_command() result = self.choose_ping_command()
elif self.type == "generic": elif self.type == "generic":
@@ -556,16 +580,16 @@ class GenCXProfile(LFCliBase):
port_mgr_response = self.json_get("/port/1/1/%s?fields=%s" % (sta_list, port_mgr_fields)) port_mgr_response = self.json_get("/port/1/1/%s?fields=%s" % (sta_list, port_mgr_fields))
# get info from port manager with list of values from cx_a_side_list # get info from port manager with list of values from cx_a_side_list
if "endpoints" not in generic_response or not generic_response: if "endpoints" not in generic_response or not generic_response:
print(generic_response) logger.critical(generic_response)
if debug: logger.critical("Json generic_response from LANforge... {generic_response}".format(generic_response=generic_response))
print("Json generic_response from LANforge... " + str(generic_response)) logger.critical("Cannot find columns requested to be searched. Exiting script, please retry.")
raise ValueError("Cannot find columns requested to be searched. Exiting script, please retry.") raise ValueError("Cannot find columns requested to be searched. Exiting script, please retry.")
if port_mgr_cols: if port_mgr_cols:
if "interfaces" not in port_mgr_response or not port_mgr_response: if "interfaces" not in port_mgr_response or not port_mgr_response:
print(port_mgr_response) logger.critical(port_mgr_response)
logger.critical("Cannot find columns requested to be searched. Exiting script, please retry.")
raise ValueError("Cannot find columns requested to be searched. Exiting script, please retry.") raise ValueError("Cannot find columns requested to be searched. Exiting script, please retry.")
if debug: logger.debug("Json port_mgr_response from LANforge... {port_mgr_response}".format(port_mgr_response=port_mgr_response))
print("Json port_mgr_response from LANforge... " + str(port_mgr_response))
for endpoint in generic_response["endpoints"]: # each endpoint is a dictionary for endpoint in generic_response["endpoints"]: # each endpoint is a dictionary
endp_values = list(endpoint.values())[0] endp_values = list(endpoint.values())[0]