mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-02 11:48:03 +00:00
lf_radio_info.py: module unit test built
Signed-off-by: Scott Wedge <scott.wedge@candelatech.com>
This commit is contained in:
Scott Wedge
317
py-scripts/lf_radio_info.py
Executable file
317
py-scripts/lf_radio_info.py
Executable file
@@ -0,0 +1,317 @@
|
||||
#!/usr/bin/env python3
|
||||
'''
|
||||
NAME: lf_radio_info.py
|
||||
|
||||
PURPOSE:
|
||||
|
||||
This script will gather all wiphy radio information, and then parse and
|
||||
return specific wiphy radio data that is requested per script function.
|
||||
|
||||
EXAMPLE:
|
||||
|
||||
This will run through the module test:
|
||||
./lf_radio_info.py --mgr 192.168.30.12
|
||||
|
||||
'''
|
||||
|
||||
import argparse
|
||||
import sys
|
||||
import os
|
||||
import logging
|
||||
import importlib
|
||||
import json
|
||||
import requests
|
||||
import pandas as pd
|
||||
|
||||
if sys.version_info[0] != 3:
|
||||
print("This script requires Python 3")
|
||||
exit(1)
|
||||
|
||||
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
|
||||
|
||||
LFUtils = importlib.import_module("py-json.LANforge.LFUtils")
|
||||
removeCX = LFUtils.removeCX
|
||||
removeEndps = LFUtils.removeEndps
|
||||
realm = importlib.import_module("py-json.realm")
|
||||
Realm = realm.Realm
|
||||
lf_report = importlib.import_module("py-scripts.lf_report")
|
||||
lf_graph = importlib.import_module("py-scripts.lf_graph")
|
||||
logger = logging.getLogger(__name__)
|
||||
lf_logger_config = importlib.import_module("py-scripts.lf_logger_config")
|
||||
|
||||
OPEN = "open"
|
||||
WEP = "wep"
|
||||
WPA = "wpa"
|
||||
WPA2 = "wpa2"
|
||||
WPA3 = "wpa3"
|
||||
MODE_AUTO = 0
|
||||
|
||||
class RadioInfo(Realm):
|
||||
def __init__(self, host,
|
||||
_resource="1",
|
||||
debug_=False, _exit_on_fail=False):
|
||||
super().__init__(host, debug_=debug_, _exit_on_fail=_exit_on_fail)
|
||||
self.host = host
|
||||
self.resource = _resource
|
||||
self.lf_mgr_ip = host
|
||||
self.lf_mgr_port = "8080"
|
||||
self.lf_mgr_user = "lanforge"
|
||||
self.lf_mgr_pass = "lanforge"
|
||||
self.lf_radio_df = ""
|
||||
self.debug = debug_
|
||||
|
||||
# returns all wihpy radio data as a pandas.df:
|
||||
def get_lanforge_radio_information(self):
|
||||
# https://docs.python-requests.org/en/latest/
|
||||
# https://stackoverflow.com/questions/26000336/execute-curl-command-within-a-python-script - use requests
|
||||
# curl --user "lanforge:lanforge" -H 'Accept: application/json'
|
||||
# http://192.168.100.116:8080/radiostatus/all | json_pp , where --user
|
||||
# "USERNAME:PASSWORD"
|
||||
request_command = 'http://{lfmgr}:{port}/radiostatus/all'.format(lfmgr=self.lf_mgr_ip, port=self.lf_mgr_port)
|
||||
request = requests.get(request_command, auth=(self.lf_mgr_user, self.lf_mgr_pass))
|
||||
|
||||
logger.info("radio request command: {request_command}".format(request_command=request_command))
|
||||
logger.info("radio request status_code {status}".format(status=request.status_code))
|
||||
lanforge_radio_json = request.json()
|
||||
|
||||
lanforge_radio_formatted_str = json.dumps(lanforge_radio_json, indent=2)
|
||||
logger.info("lanforge_radio_json: {lanforge_radio_json}".format(lanforge_radio_json=lanforge_radio_formatted_str))
|
||||
# note put into the meta data
|
||||
# self.radio_dataframe
|
||||
self.lf_radio_df = pd.DataFrame(
|
||||
columns=[
|
||||
'Radio',
|
||||
'WIFI-Radio Driver',
|
||||
'Radio Capabilities',
|
||||
'Firmware Version',
|
||||
'max_sta',
|
||||
'max_vap',
|
||||
'max_vifs'])
|
||||
for key in lanforge_radio_json:
|
||||
if 'wiphy' in key:
|
||||
driver = lanforge_radio_json[key]['driver'].split(
|
||||
'Driver:', maxsplit=1)[-1].split(maxsplit=1)[0]
|
||||
try:
|
||||
firmware_version = lanforge_radio_json[key]['firmware version']
|
||||
except BaseException:
|
||||
logger.info("5.4.5 radio fw version not in /radiostatus/all ")
|
||||
firmware_version = "5.4.5 N/A"
|
||||
self.lf_radio_df = self.lf_radio_df.append(
|
||||
{'Radio': lanforge_radio_json[key]['entity id'],
|
||||
'WIFI-Radio Driver': driver,
|
||||
'Radio Capabilities': lanforge_radio_json[key]['capabilities'],
|
||||
'Firmware Version': firmware_version,
|
||||
'max_sta': lanforge_radio_json[key]['max_sta'],
|
||||
'max_vap': lanforge_radio_json[key]['max_vap'],
|
||||
'max_vifs': lanforge_radio_json[key]['max_vifs']}, ignore_index=True)
|
||||
|
||||
return self.lf_radio_df
|
||||
|
||||
# returns all radios and maximum stations:
|
||||
def get_max_stations_all(self):
|
||||
max_station = self.lf_radio_df.get(["Radio", "max_sta"])
|
||||
# logger.info(max_station)
|
||||
|
||||
return max_station
|
||||
|
||||
# returns specific radio maximum stations:
|
||||
def get_radio_max_station(self, wiphy_radio):
|
||||
max_station = self.lf_radio_df.get(["Radio", "max_sta"])
|
||||
# logger.info(max_station)
|
||||
|
||||
# process specific elements from dataframe:
|
||||
for row in max_station.index:
|
||||
if wiphy_radio == "1.1.wiphy0":
|
||||
if max_station.loc[row, "Radio"] == "1." + self.resource + ".wiphy0":
|
||||
max_sta = max_station.loc[row, "max_sta"]
|
||||
# logger.info(max_sta)
|
||||
elif wiphy_radio == "1.1.wiphy1":
|
||||
if max_station.loc[row, "Radio"] == "1." + self.resource + ".wiphy1":
|
||||
max_sta = max_station.loc[row, "max_sta"]
|
||||
# logger.info(max_sta)
|
||||
|
||||
return max_sta
|
||||
|
||||
|
||||
# returns specific radio maximum virtual access points:
|
||||
def get_max_vap(self, wiphy_radio):
|
||||
max_vaps = self.lf_radio_df.get(["Radio", "max_vap"])
|
||||
# logger.info(max_vaps)
|
||||
|
||||
# process specific elements from dataframe:
|
||||
for row in max_vaps.index:
|
||||
if wiphy_radio == "1.1.wiphy0":
|
||||
if max_vaps.loc[row, "Radio"] == "1." + self.resource + ".wiphy0":
|
||||
wiphy_max_vaps = max_vaps.loc[row, "max_vap"]
|
||||
# logger.info(max_sta)
|
||||
elif wiphy_radio == "1.1.wiphy1":
|
||||
if max_vaps.loc[row, "Radio"] == "1." + self.resource + ".wiphy1":
|
||||
wiphy_max_vaps = max_vaps.loc[row, "max_vap"]
|
||||
# logger.info(max_sta)
|
||||
|
||||
return wiphy_max_vaps
|
||||
|
||||
# returns specific radio maximum supported virtual clients:
|
||||
def get_max_vifs(self, wiphy_radio):
|
||||
max_vif = self.lf_radio_df.get(["Radio", "max_vifs"])
|
||||
# logger.info(max_vif)
|
||||
|
||||
# process specific elements from dataframe:
|
||||
for row in max_vif.index:
|
||||
if wiphy_radio == "1.1.wiphy0":
|
||||
if max_vif.loc[row, "Radio"] == "1." + self.resource + ".wiphy0":
|
||||
wiphy_max_vif = max_vif.loc[row, "max_vifs"]
|
||||
# logger.info(max_sta)
|
||||
elif wiphy_radio == "1.1.wiphy1":
|
||||
if max_vif.loc[row, "Radio"] == "1." + self.resource + ".wiphy1":
|
||||
wiphy_max_vif = max_vif.loc[row, "max_vifs"]
|
||||
# logger.info(max_sta)
|
||||
|
||||
return wiphy_max_vif
|
||||
|
||||
# returns specific radio driver information:
|
||||
def get_radio_driver(self, wiphy_radio):
|
||||
radio_driver = self.lf_radio_df.get(["Radio", "WIFI-Radio Driver"])
|
||||
# logger.info(radio_driver)
|
||||
|
||||
# process specific elements from dataframe:
|
||||
for row in radio_driver.index:
|
||||
if wiphy_radio == "1.1.wiphy0":
|
||||
# rad_name = "1." + self.resource + ".wiphy0"
|
||||
if radio_driver.loc[row, "Radio"] == "1." + self.resource + ".wiphy0":
|
||||
wiphy_driver = radio_driver.loc[row, "WIFI-Radio Driver"]
|
||||
# logger.info(max_sta)
|
||||
elif wiphy_radio == "1.1.wiphy1":
|
||||
if radio_driver.loc[row, "Radio"] == "1." + self.resource + ".wiphy1":
|
||||
wiphy_driver = radio_driver.loc[row, "WIFI-Radio Driver"]
|
||||
# logger.info(max_sta)
|
||||
|
||||
return wiphy_driver
|
||||
|
||||
# returns specific radio 802.11 wifi capabilities:
|
||||
def get_radio_capabilities(self, wiphy_radio):
|
||||
radio_caps = self.lf_radio_df.get(["Radio", "Radio Capabilities"])
|
||||
# logger.info(radio_capabilities)
|
||||
|
||||
# process specific elements from dataframe:
|
||||
for row in radio_caps.index:
|
||||
if wiphy_radio == "1.1.wiphy0":
|
||||
if radio_caps.loc[row, "Radio"] == "1." + self.resource + ".wiphy0":
|
||||
wiphy_capability = radio_caps.loc[row, "Radio Capabilities"]
|
||||
# logger.info(max_sta)
|
||||
elif wiphy_radio == "1.1.wiphy1":
|
||||
if radio_caps.loc[row, "Radio"] == "1." + self.resource + ".wiphy1":
|
||||
wiphy_capability = radio_caps.loc[row, "Radio Capabilities"]
|
||||
# logger.info(max_sta)
|
||||
|
||||
return wiphy_capability
|
||||
|
||||
|
||||
|
||||
# ~class
|
||||
|
||||
|
||||
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(
|
||||
prog="lf_radio_info.py",
|
||||
formatter_class=argparse.RawTextHelpFormatter,
|
||||
description="""
|
||||
---------------------------
|
||||
LANforge Unit Test: Retreive Wiphy Radio Information - lf_radio_info.py
|
||||
---------------------------
|
||||
Summary:
|
||||
This script will gather all wiphy radio information to be used by other test scripts.
|
||||
---------------------------
|
||||
CLI Example:
|
||||
|
||||
./lf_radio_info.py --mgr 192.168.30.12
|
||||
|
||||
---------------------------
|
||||
""")
|
||||
parser.add_argument("-m", "--mgr", type=str, help="address of the LANforge GUI machine (localhost is default)",
|
||||
default='localhost')
|
||||
parser.add_argument("--debug", help="enable debugging", action="store_true")
|
||||
parser.add_argument("--resource", type=str, help="LANforge Station resource ID to use, default is 1", default="1")
|
||||
#parser.add_argument('--radio_list', default=None, help="Specify a file to send debug output to")
|
||||
#parser.add_argument('--index_list', default=None, help="Specify a file to send debug output to")
|
||||
parser.add_argument('--debug_log', default=None, help="Specify a file to send debug output to")
|
||||
# logging configuration:
|
||||
parser.add_argument("--lf_logger_config_json",
|
||||
help="--lf_logger_config_json <json file> , json configuration of logger")
|
||||
|
||||
# TODO: Try to start with: --radio 'radio==wiphy4,stations==1,ssid==asus11ax-2,ssid_pw==hello123,security==wpa2':
|
||||
parser.add_argument(
|
||||
'-r', '--radio',
|
||||
action='append',
|
||||
nargs=1,
|
||||
help=(' --radio'
|
||||
' "radio==<number_of_wiphy stations=<=number of stations>'
|
||||
' ssid==<ssid> ssid_pw==<ssid password> security==<security> '
|
||||
' wifi_settings==True wifi_mode==<wifi_mode>'
|
||||
' enable_flags==<enable_flags> '
|
||||
' reset_port_enable==True reset_port_time_min==<min>s'
|
||||
' reset_port_time_max==<max>s" ')
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
# set up logger
|
||||
logger_config = lf_logger_config.lf_logger_config()
|
||||
if args.lf_logger_config_json:
|
||||
# logger_config.lf_logger_config_json = "lf_logger_config.json"
|
||||
logger_config.lf_logger_config_json = args.lf_logger_config_json
|
||||
logger_config.load_lf_logger_config()
|
||||
|
||||
if args.debug:
|
||||
logger_config.set_level("debug")
|
||||
|
||||
|
||||
radioInfo = RadioInfo(args.mgr,
|
||||
_resource=args.resource,
|
||||
debug_=args.debug,
|
||||
_exit_on_fail=True)
|
||||
|
||||
# basic unit module test:
|
||||
|
||||
radio = "1.1.wiphy1"
|
||||
|
||||
# get all system radio information:
|
||||
all_radio_info = radioInfo.get_lanforge_radio_information()
|
||||
logger.info(all_radio_info)
|
||||
|
||||
# get max stations per each install wiphy radio:
|
||||
all_max_sta = radioInfo.get_max_stations_all()
|
||||
print()
|
||||
logger.info(all_max_sta)
|
||||
|
||||
# get max stations for specified wiphy radio:
|
||||
radio_max_sta = radioInfo.get_radio_max_station(radio)
|
||||
logger.info(radio_max_sta)
|
||||
|
||||
# get max vap that can be created per spicified wiphy radio:
|
||||
max_vap = radioInfo.get_max_vap(radio)
|
||||
logger.info(max_vap)
|
||||
|
||||
# get max vif that can be created per specified wiphy radio:
|
||||
max_vifs = radioInfo.get_max_vifs(radio)
|
||||
logger.info(max_vifs)
|
||||
|
||||
# get radio driver for specified wiphy radio:
|
||||
radio_driver = radioInfo.get_radio_driver(radio)
|
||||
logger.info(radio_driver)
|
||||
|
||||
# get 802.11 capabilities for specified radio:
|
||||
radio_capabilities = radioInfo.get_radio_capabilities(radio)
|
||||
logger.info(radio_capabilities)
|
||||
|
||||
|
||||
|
||||
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user