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

whitespace changes after autopep8

Browse Source
This commit is contained in:
Logan Lipke
2021-11-29 10:14:28 -08:00
parent 4a97a90f28
commit 6fab304140
1 changed files with 65 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

124
py-scripts/lf_webpage.py
View File

@@ -17,7 +17,7 @@ import paramiko
from datetime import datetime from datetime import datetime
import pandas as pd import pandas as pd
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../"))) sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
LFUtils = importlib.import_module("py-json.LANforge.LFUtils") LFUtils = importlib.import_module("py-json.LANforge.LFUtils")
@@ -57,8 +57,6 @@ class HttpDownload(Realm):
self.http_profile.debug = _debug_on self.http_profile.debug = _debug_on
self.created_cx = {} self.created_cx = {}
def set_values(self): def set_values(self):
# This method will set values according user input # This method will set values according user input
if self.bands == "5G": if self.bands == "5G":
@@ -67,7 +65,7 @@ class HttpDownload(Realm):
self.radio = [self.twog_radio] self.radio = [self.twog_radio]
elif self.bands == "Both": elif self.bands == "Both":
self.radio = [self.fiveg_radio, self.twog_radio] self.radio = [self.fiveg_radio, self.twog_radio]
print( self.radio) print(self.radio)
self.num_sta = self.num_sta // 2 self.num_sta = self.num_sta // 2
def precleanup(self): def precleanup(self):
@@ -121,7 +119,7 @@ class HttpDownload(Realm):
self.station_profile.create(radio=rad, sta_names_=self.station_list, debug=self.local_realm.debug) self.station_profile.create(radio=rad, sta_names_=self.station_list, debug=self.local_realm.debug)
self.local_realm.wait_until_ports_appear(sta_list=self.station_list) self.local_realm.wait_until_ports_appear(sta_list=self.station_list)
self.station_profile.admin_up() self.station_profile.admin_up()
if self.local_realm.wait_for_ip(self.station_list,timeout_sec=60): if self.local_realm.wait_for_ip(self.station_list, timeout_sec=60):
self.local_realm._pass("All stations got IPs") self.local_realm._pass("All stations got IPs")
else: else:
self.local_realm._fail("Stations failed to get IPs") self.local_realm._fail("Stations failed to get IPs")
@@ -181,7 +179,7 @@ class HttpDownload(Realm):
LFUtils.wait_until_ports_disappear(base_url=self.local_realm.lfclient_url, port_list=self.station_profile.station_names, LFUtils.wait_until_ports_disappear(base_url=self.local_realm.lfclient_url, port_list=self.station_profile.station_names,
debug=self.debug) debug=self.debug)
def file_create(self,ssh_port): def file_create(self, ssh_port):
ip = self.host ip = self.host
user = "root" user = "root"
pswd = "lanforge" pswd = "lanforge"
@@ -211,13 +209,13 @@ class HttpDownload(Realm):
time.sleep(1) time.sleep(1)
return output return output
def download_time_in_sec(self,result_data): def download_time_in_sec(self, result_data):
self.resullt_data = result_data self.resullt_data = result_data
download_time = dict.fromkeys(result_data.keys()) download_time = dict.fromkeys(result_data.keys())
for i in download_time: for i in download_time:
try: try:
download_time[i] = result_data[i]['dl_time'] download_time[i] = result_data[i]['dl_time']
except: except BaseException:
download_time[i] = [] download_time[i] = []
print(download_time) print(download_time)
lst = [] lst = []
@@ -253,13 +251,13 @@ class HttpDownload(Realm):
dataset.append(dwnld_time["Both"]) dataset.append(dwnld_time["Both"])
return dataset return dataset
def speed_in_Mbps(self,result_data): def speed_in_Mbps(self, result_data):
self.resullt_data = result_data self.resullt_data = result_data
speed = dict.fromkeys(result_data.keys()) speed = dict.fromkeys(result_data.keys())
for i in speed: for i in speed:
try: try:
speed[i] = result_data[i]['speed'] speed[i] = result_data[i]['speed']
except: except BaseException:
speed[i] = [] speed[i] = []
print(speed) print(speed)
lst = [] lst = []
@@ -341,7 +339,7 @@ class HttpDownload(Realm):
pass_fail_list.append("PASS") pass_fail_list.append("PASS")
sumry2.append("PASS") sumry2.append("PASS")
# BOTH # BOTH
if float(z11[2]) == 0.0 or float(z11[2]) > float(threshold_both) : if float(z11[2]) == 0.0 or float(z11[2]) > float(threshold_both):
var = "FAIL" var = "FAIL"
pass_fail_list.append(var) pass_fail_list.append(var)
sumryB.append("FAIL") sumryB.append("FAIL")
@@ -446,35 +444,36 @@ class HttpDownload(Realm):
def generate_graph(self, dataset, lis, bands): def generate_graph(self, dataset, lis, bands):
graph = lf_graph.lf_bar_graph(_data_set=dataset, _xaxis_name="Stations", _yaxis_name="Time in Seconds", graph = lf_graph.lf_bar_graph(_data_set=dataset, _xaxis_name="Stations", _yaxis_name="Time in Seconds",
_xaxis_categories=lis, _label=bands, _xticks_font=8, _xaxis_categories=lis, _label=bands, _xticks_font=8,
_graph_image_name="webpage download time graph", _graph_image_name="webpage download time graph",
_color=['forestgreen', 'darkorange', 'blueviolet'], _color_edge='black', _figsize=(14, 5), _color=['forestgreen', 'darkorange', 'blueviolet'], _color_edge='black', _figsize=(14, 5),
_grp_title="Download time taken by each client", _xaxis_step=1, _show_bar_value=True, _grp_title="Download time taken by each client", _xaxis_step=1, _show_bar_value=True,
_text_font=6, _text_rotation=60, _text_font=6, _text_rotation=60,
_legend_loc="upper right", _legend_loc="upper right",
_legend_box=(1, 1.15), _legend_box=(1, 1.15),
_enable_csv=True _enable_csv=True
) )
graph_png = graph.build_bar_graph() graph_png = graph.build_bar_graph()
print("graph name {}".format(graph_png)) print("graph name {}".format(graph_png))
return graph_png return graph_png
def graph_2(self,dataset2, lis, bands): def graph_2(self, dataset2, lis, bands):
graph_2 = lf_graph.lf_bar_graph(_data_set=dataset2, _xaxis_name="Stations", _yaxis_name="Download Rate in Mbps", graph_2 = lf_graph.lf_bar_graph(_data_set=dataset2, _xaxis_name="Stations", _yaxis_name="Download Rate in Mbps",
_xaxis_categories=lis, _label=bands, _xticks_font=8, _xaxis_categories=lis, _label=bands, _xticks_font=8,
_graph_image_name="webpage_speed_graph", _graph_image_name="webpage_speed_graph",
_color=['forestgreen', 'darkorange', 'blueviolet'], _color_edge='black', _color=['forestgreen', 'darkorange', 'blueviolet'], _color_edge='black',
_figsize=(14, 5), _figsize=(14, 5),
_grp_title="Download rate for each client (Mbps)", _xaxis_step=1, _show_bar_value=True, _grp_title="Download rate for each client (Mbps)", _xaxis_step=1, _show_bar_value=True,
_text_font=6, _text_rotation=60, _text_font=6, _text_rotation=60,
_legend_loc="upper right", _legend_loc="upper right",
_legend_box=(1, 1.15), _legend_box=(1, 1.15),
_enable_csv=True _enable_csv=True
) )
graph_png = graph_2.build_bar_graph() graph_png = graph_2.build_bar_graph()
return graph_png return graph_png
def generate_report(self,date, num_stations,duration, test_setup_info,dataset,lis,bands,threshold_2g,threshold_5g,threshold_both,dataset2,summary_table_value,result_data,test_input_infor): def generate_report(self, date, num_stations, duration, test_setup_info, dataset, lis, bands, threshold_2g,
threshold_5g, threshold_both, dataset2, summary_table_value, result_data, test_input_infor):
report = lf_report.lf_report(_results_dir_name="webpage_test", _output_html="Webpage.html", _output_pdf="Webpage.pdf") report = lf_report.lf_report(_results_dir_name="webpage_test", _output_html="Webpage.html", _output_pdf="Webpage.pdf")
report.set_title("WEBPAGE DOWNLOAD TEST") report.set_title("WEBPAGE DOWNLOAD TEST")
report.set_date(date) report.set_date(date)
@@ -594,7 +593,8 @@ class HttpDownload(Realm):
html_file = report.write_html() html_file = report.write_html()
print("returned file {}".format(html_file)) print("returned file {}".format(html_file))
print(html_file) print(html_file)
report.write_pdf() report.write_pdf_with_timestamp(_page_size='A4', _orientation='Landscape')
def main(): def main():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
@@ -614,11 +614,11 @@ def main():
parser.add_argument('--file_size', type=str, help='specify the size of file you want to download', default='5MB') parser.add_argument('--file_size', type=str, help='specify the size of file you want to download', default='5MB')
parser.add_argument('--bands', nargs="+", help='specify which band testing you want to run eg 5G OR 2.4G OR Both', default=["5G", "2.4G", "Both"]) parser.add_argument('--bands', nargs="+", help='specify which band testing you want to run eg 5G OR 2.4G OR Both', default=["5G", "2.4G", "Both"])
parser.add_argument('--duration', type=int, help='time to run traffic') parser.add_argument('--duration', type=int, help='time to run traffic')
parser.add_argument('--threshold_5g',help="Enter the threshold value for 5G Pass/Fail criteria", default="60") parser.add_argument('--threshold_5g', help="Enter the threshold value for 5G Pass/Fail criteria", default="60")
parser.add_argument('--threshold_2g',help="Enter the threshold value for 2.4G Pass/Fail criteria",default="90") parser.add_argument('--threshold_2g', help="Enter the threshold value for 2.4G Pass/Fail criteria", default="90")
parser.add_argument('--threshold_both',help="Enter the threshold value for Both Pass/Fail criteria" , default="50") parser.add_argument('--threshold_both', help="Enter the threshold value for Both Pass/Fail criteria", default="50")
parser.add_argument('--ap_name', help="specify the ap model ", default="TestAP") parser.add_argument('--ap_name', help="specify the ap model ", default="TestAP")
parser.add_argument('--ssh_port', type=int, help="specify the shh port eg 22",default=22) parser.add_argument('--ssh_port', type=int, help="specify the ssh port eg 22", default=22)
args = parser.parse_args() args = parser.parse_args()
test_time = datetime.now() test_time = datetime.now()
@@ -631,14 +631,14 @@ def main():
list2G_bytes = [] list2G_bytes = []
list2G_speed = [] list2G_speed = []
Both = [] Both = []
Both_bytes =[] Both_bytes = []
Both_speed =[] Both_speed = []
dict_keys = [] dict_keys = []
dict_keys.extend(args.bands) dict_keys.extend(args.bands)
# print(dict_keys) # print(dict_keys)
final_dict = dict.fromkeys(dict_keys) final_dict = dict.fromkeys(dict_keys)
# print(final_dict) # print(final_dict)
dict1_keys = ['dl_time', 'min', 'max', 'avg','bytes_rd', 'speed'] dict1_keys = ['dl_time', 'min', 'max', 'avg', 'bytes_rd', 'speed']
for i in final_dict: for i in final_dict:
final_dict[i] = dict.fromkeys(dict1_keys) final_dict[i] = dict.fromkeys(dict1_keys)
print(final_dict) print(final_dict)
@@ -720,7 +720,7 @@ def main():
final_dict['Both']['min'] = min_both final_dict['Both']['min'] = min_both
max_both.append(max(Both)) max_both.append(max(Both))
final_dict['Both']['max'] = max_both final_dict['Both']['max'] = max_both
avg_both.append((sum(Both) /args.num_stations)) avg_both.append((sum(Both) / args.num_stations))
final_dict['Both']['avg'] = avg_both final_dict['Both']['avg'] = avg_both
final_dict['Both']['bytes_rd'] = Both_bytes final_dict['Both']['bytes_rd'] = Both_bytes
final_dict['Both']['speed'] = Both_speed final_dict['Both']['speed'] = Both_speed
@@ -739,36 +739,41 @@ def main():
print("total test duration ", test_duration) print("total test duration ", test_duration)
date = str(datetime.now()).split(",")[0].replace(" ", "-").split(".")[0] date = str(datetime.now()).split(",")[0].replace(" ", "-").split(".")[0]
test_setup_info = { test_setup_info = {
"DUT Name": args.ap_name, "DUT Name": args.ap_name,
"SSID": args.ssid, "SSID": args.ssid,
"Test Duration": test_duration, "Test Duration": test_duration,
} }
test_input_infor={ test_input_infor = {
"LANforge ip":args.mgr, "LANforge ip": args.mgr,
"File Size" : args.file_size, "File Size": args.file_size,
"Bands" : args.bands, "Bands": args.bands,
"Upstream": args.upstream_port, "Upstream": args.upstream_port,
"Stations": args.num_stations, "Stations": args.num_stations,
"SSID" :args.ssid, "SSID": args.ssid,
"Security": args.security, "Security": args.security,
"Duration" : args.duration, "Duration": args.duration,
"Contact": "support@candelatech.com" "Contact": "support@candelatech.com"
} }
http1 = HttpDownload(lfclient_host=args.mgr, lfclient_port=args.mgr_port, http1 = HttpDownload(lfclient_host=args.mgr, lfclient_port=args.mgr_port,
upstream=args.upstream_port, num_sta=args.num_stations, upstream=args.upstream_port, num_sta=args.num_stations,
security=args.security, security=args.security,
ssid=args.ssid, password=args.passwd, ssid=args.ssid, password=args.passwd,
target_per_ten=args.target_per_ten, target_per_ten=args.target_per_ten,
file_size=args.file_size, bands=args.bands, file_size=args.file_size, bands=args.bands,
twog_radio=args.twog_radio, twog_radio=args.twog_radio,
fiveg_radio=args.fiveg_radio) fiveg_radio=args.fiveg_radio)
dataset = http1.download_time_in_sec(result_data=result_data) dataset = http1.download_time_in_sec(result_data=result_data)
lis = [] lis = []
for i in range(1, args.num_stations + 1): for i in range(1, args.num_stations + 1):
lis.append(i) lis.append(i)
dataset2= http1.speed_in_Mbps(result_data=result_data) dataset2 = http1.speed_in_Mbps(result_data=result_data)
data = http1.summary_calculation(result_data=result_data, bands=args.bands, threshold_5g=args.threshold_5g , threshold_2g= args.threshold_2g, threshold_both=args.threshold_both) data = http1.summary_calculation(
result_data=result_data,
bands=args.bands,
threshold_5g=args.threshold_5g,
threshold_2g=args.threshold_2g,
threshold_both=args.threshold_both)
summary_table_value = { summary_table_value = {
"": args.bands, "": args.bands,
@@ -780,5 +785,6 @@ def main():
threshold_both=args.threshold_both, dataset2=dataset2, threshold_both=args.threshold_both, dataset2=dataset2,
summary_table_value=summary_table_value, result_data=result_data, test_input_infor=test_input_infor) summary_table_value=summary_table_value, result_data=result_data, test_input_infor=test_input_infor)
if __name__ == '__main__': if __name__ == '__main__':
main() main()