mirror of
				https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
				synced 2025-10-31 18:58:01 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			99 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			99 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
| #!/usr/bin/env python3
 | |
| '''
 | |
| This script loads and builds a Chamber View Scenario, runs WiFi Capacity Test, and posts the results to Grafana
 | |
| '''
 | |
| import sys
 | |
| import os
 | |
| import argparse
 | |
| 
 | |
| if sys.version_info[0] != 3:
 | |
|     print("This script requires Python 3")
 | |
|     exit(1)
 | |
| 
 | |
| if 'py-json' not in sys.path:
 | |
|     sys.path.append(os.path.join(os.path.abspath('..'), 'py-json'))
 | |
|     sys.path.append(os.path.join(os.path.abspath('..'), 'py-dashboard'))
 | |
| 
 | |
| 
 | |
| from lf_wifi_capacity_test import WiFiCapacityTest
 | |
| from cv_test_manager import *
 | |
| 
 | |
| def main():
 | |
|     parser = argparse.ArgumentParser(
 | |
|         prog='wifi_cap_to_grafana.py',
 | |
|         formatter_class=argparse.RawTextHelpFormatter,
 | |
|         epilog='''Run Wifi Capacity and record results to Grafana''',
 | |
|         description='''\
 | |
|         wifi_cap_to_grafana.py
 | |
|         ------------------
 | |
|         ./wifi_cap_to_grafana.py
 | |
|             --num_stations
 | |
|             --grafana_token
 | |
|             --influx_host
 | |
|             --influx_org
 | |
|             --influx_token
 | |
|             --influx_bucket
 | |
|             --target_csv
 | |
|             --panel_name
 | |
|             --dut_name'''
 | |
|     )
 | |
| 
 | |
|     cv_add_base_parser(parser)  # see cv_test_manager.py
 | |
| 
 | |
|     parser.add_argument("-b", "--batch_size", type=str, default="",
 | |
|                         help="station increment ex. 1,2,3")
 | |
|     parser.add_argument("-l", "--loop_iter", type=str, default="",
 | |
|                         help="Loop iteration ex. 1")
 | |
|     parser.add_argument("-p", "--protocol", type=str, default="",
 | |
|                         help="Protocol ex.TCP-IPv4")
 | |
|     parser.add_argument("-d", "--duration", type=str, default="",
 | |
|                         help="duration in ms. ex. 5000")
 | |
|     parser.add_argument("--download_rate", type=str, default="1Gbps",
 | |
|                         help="Select requested download rate.  Kbps, Mbps, Gbps units supported.  Default is 1Gbps")
 | |
|     parser.add_argument("--upload_rate", type=str, default="10Mbps",
 | |
|                         help="Select requested upload rate.  Kbps, Mbps, Gbps units supported.  Default is 10Mbps")
 | |
|     parser.add_argument("--sort", type=str, default="interleave",
 | |
|                         help="Select station sorting behaviour:  none | interleave | linear  Default is interleave.")
 | |
|     parser.add_argument('--number_template', help='Start the station numbering with a particular number. Default is 0000',
 | |
|                         default=0000)
 | |
|     parser.add_argument('--mode', help='Used to force mode of stations')
 | |
|     parser.add_argument('--ap', help='Used to force a connection to a particular AP')
 | |
|     parser.add_argument("--num_stations", default=2)
 | |
|     parser.add_argument("--mgr_port", default=8080)
 | |
|     parser.add_argument("--upstream_port", default="1.1.eth1")
 | |
|     parser.add_argument("--scenario", help="", default=None)
 | |
|     args = parser.parse_args()
 | |
| 
 | |
| 
 | |
|     cv_base_adjust_parser(args)
 | |
| 
 | |
|     # Run WiFi Capacity Test
 | |
|     wifi_capacity = WiFiCapacityTest(lfclient_host=args.mgr,
 | |
|                                      lf_port=args.mgr_port,
 | |
|                                      lf_user=args.lf_user,
 | |
|                                      lf_password=args.lf_password,
 | |
|                                      instance_name=args.instance_name,
 | |
|                                      config_name=args.config_name,
 | |
|                                      upstream=args.upstream_port,
 | |
|                                      batch_size=args.batch_size,
 | |
|                                      loop_iter=args.loop_iter,
 | |
|                                      protocol=args.protocol,
 | |
|                                      duration=args.duration,
 | |
|                                      pull_report=args.pull_report,
 | |
|                                      load_old_cfg=args.load_old_cfg,
 | |
|                                      download_rate=args.download_rate,
 | |
|                                      upload_rate=args.upload_rate,
 | |
|                                      sort=args.sort,
 | |
|                                      enables=args.enable,
 | |
|                                      disables=args.disable,
 | |
|                                      raw_lines=args.raw_line,
 | |
|                                      raw_lines_file=args.raw_lines_file,
 | |
|                                      sets=args.set)
 | |
|     wifi_capacity.apply_cv_scenario(args.scenario)
 | |
|     wifi_capacity.build_cv_scenario()
 | |
|     wifi_capacity.setup()
 | |
|     wifi_capacity.run()
 | |
|     wifi_capacity.check_influx_kpi(args)
 | |
| 
 | |
| if __name__ == "__main__":
 | |
|     main() | 
