mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-01 19:28:00 +00:00
- integrated cv_commands.pywith cv_test_manager.py
- create_chamberview_dut.py to create a dut - cv_dut_profile.py as library to create a dut - cv test manager can be used to implement chamber view also Signed-off-by: SushantBawiskar <sushant.bawiskar@candelatech.com>
This commit is contained in:
SushantBawiskar
@@ -46,7 +46,7 @@ if sys.version_info[0] != 3:
|
||||
if 'py-json' not in sys.path:
|
||||
sys.path.append(os.path.join(os.path.abspath('..'), 'py-json'))
|
||||
|
||||
from cv_commands import chamberview as cv
|
||||
from cv_test_manager import cv_test as cv
|
||||
|
||||
class CreateChamberview(cv):
|
||||
def __init__(self,
|
||||
@@ -126,7 +126,7 @@ class CreateChamberview(cv):
|
||||
else:
|
||||
continue
|
||||
|
||||
self.manage_cv_scenario(scenario_name,
|
||||
self.add_text_blob_line(scenario_name,
|
||||
Resource,
|
||||
Profile,
|
||||
Amount,
|
||||
@@ -151,7 +151,16 @@ class CreateChamberview(cv):
|
||||
self.show_text_blob(None, None, False) # Show changes on GUI
|
||||
self.apply_cv_scenario(scenario_name) # Apply scenario
|
||||
self.build_cv_scenario() # build scenario
|
||||
while self.is_cv_build() == "NO": continue # wait for cv to build
|
||||
tries = 0
|
||||
while (True):
|
||||
if not self.get_cv_is_built():
|
||||
print("Waiting %i/60 for Chamber-View to be built." % (tries))
|
||||
tries += 1
|
||||
if (tries > 60):
|
||||
break
|
||||
time.sleep(1)
|
||||
else:
|
||||
break
|
||||
print("completed building %s scenario" %scenario_name)
|
||||
|
||||
|
||||
|
||||
141
py-scripts/create_chamberview_dut.py
Normal file
141
py-scripts/create_chamberview_dut.py
Normal file
@@ -0,0 +1,141 @@
|
||||
import sys
|
||||
import os
|
||||
import argparse
|
||||
import time
|
||||
import re
|
||||
|
||||
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'))
|
||||
|
||||
from cv_dut_profile import cv_dut as dut
|
||||
from cv_test_manager import cv_test as cvtest
|
||||
|
||||
class DUT(dut):
|
||||
def __init__(self,
|
||||
lfmgr="localhost",
|
||||
port="8080",
|
||||
dut_name="DUT",
|
||||
ssid=[],
|
||||
password=[],
|
||||
bssid=[],
|
||||
security=[]
|
||||
):
|
||||
super().__init__(
|
||||
lfclient_host=lfmgr,
|
||||
lfclient_port=port,
|
||||
)
|
||||
self.cv = cvtest(lfmgr, port)
|
||||
self.dut_name = dut_name
|
||||
self.ssid = ssid
|
||||
self.password = password
|
||||
self.bssid = bssid
|
||||
self.security = security
|
||||
|
||||
def setup(self):
|
||||
self.create_dut(dut_name=self.dut_name)
|
||||
|
||||
def add_ssids(self):
|
||||
if self.ssid:
|
||||
|
||||
for i in range(len(self.ssid)):
|
||||
|
||||
if " " in self.ssid[i][0]:
|
||||
self.ssid[i][0] = (re.split(' ', self.ssid[i][0]))
|
||||
elif "," in self.ssid[i][0]:
|
||||
self.ssid[i][0] = (re.split(',', self.ssid[i][0]))
|
||||
elif ", " in self.ssid[i][0]:
|
||||
self.ssid[i][0] = (re.split(',', self.ssid[i][0]))
|
||||
elif " ," in self.ssid[i][0]:
|
||||
self.ssid[i][0] = (re.split(',', self.ssid[i][0]))
|
||||
else:
|
||||
print("Wrong arguments entered !")
|
||||
exit(1)
|
||||
|
||||
ssid_idx = 0
|
||||
ssid = "[BLANK]"
|
||||
passwd = "[BLANK]"
|
||||
bssid = "00:00:00:00:00:00"
|
||||
flag = 0x0
|
||||
|
||||
for j in range(len(self.ssid[i][0])):
|
||||
self.ssid[i][0][j] = self.ssid[i][0][j].split("=")
|
||||
for k in range(len(self.ssid[i][0][j])):
|
||||
name = self.ssid[i][0][j][k]
|
||||
if str(name) == "SSID" or str(name) == "ssid" or str(name) == "s":
|
||||
ssid = self.ssid[i][0][j][k + 1]
|
||||
elif str(name) == "PASSWORD" or str(name) == "password" or str(name) == "pass":
|
||||
passwd = self.ssid[i][0][j][k + 1]
|
||||
elif str(name) == "ssid_idx" or str(name) == "no" or str(name) == "N":
|
||||
ssid_idx = self.ssid[i][0][j][k + 1]
|
||||
elif str(name) == "security" or str(name) == "sec":
|
||||
if self.ssid[i][0][j][k + 1]:
|
||||
all_flags = self.ssid[i][0][j][k + 1].split("|")
|
||||
for flags in all_flags:
|
||||
if flags == "WEP" or flags == "wep":
|
||||
flag += 0x8
|
||||
if flags == "WPA" or flags == "wpa":
|
||||
flag += 0x10
|
||||
if flags == "WPA2" or flags == "wpa2":
|
||||
flag += 0x20
|
||||
if flags == "WPA3" or flags == "wpa3":
|
||||
flag += 0x100
|
||||
if flags == "11r":
|
||||
flag += 0x200
|
||||
if flags == "EAP-TTLS":
|
||||
flag += 0x400
|
||||
if flags == "EAP-PEAP":
|
||||
flag += 0x800
|
||||
elif str(name) == "BSSID" or str(name) == "bssid" or str(name) == "B":
|
||||
bssid = self.ssid[i][0][j][k + 1]
|
||||
else:
|
||||
continue
|
||||
|
||||
self.add_ssid(dut_name=self.dut_name,
|
||||
ssid_idx=ssid_idx,
|
||||
ssid=ssid,
|
||||
passwd=passwd,
|
||||
bssid=bssid,
|
||||
ssid_flags=flag,
|
||||
ssid_flags_mask=0xFFFFFFFF
|
||||
)
|
||||
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(
|
||||
description="""
|
||||
./create_chamberview_dut -m "localhost" -o "8080" -d "dut_name"
|
||||
-s "ssid_idx=0 ssid=NET1 security=WPA|WEP|11r|EAP-PEAP bssid=78:d2:94:bf:16:41"
|
||||
-s "ssid_idx=1 ssid=NET1 security=WPA password=test bssid=78:d2:94:bf:16:40"
|
||||
""")
|
||||
parser.add_argument("-m", "--lfmgr", type=str, default="localhost",
|
||||
help="address of the LANforge GUI machine (localhost is default)")
|
||||
parser.add_argument("-o", "--port", type=str, default="8080",
|
||||
help="IP Port the LANforge GUI is listening on (8080 is default)")
|
||||
parser.add_argument("-d", "--dut_name", type=str, default="DUT",
|
||||
help="set dut name")
|
||||
parser.add_argument("-s", "--ssid", action='append', nargs=1,
|
||||
help="SSID", default=[])
|
||||
|
||||
|
||||
args = parser.parse_args()
|
||||
new_dut = DUT(args.lfmgr,
|
||||
args.port,
|
||||
args.dut_name,
|
||||
args.ssid,
|
||||
)
|
||||
|
||||
new_dut.setup()
|
||||
new_dut.add_ssids()
|
||||
cv =cvtest(args.lfmgr,
|
||||
args.port)
|
||||
cv.show_text_blob(None, None, False) # Show changes on GUI
|
||||
cv.sync_cv()
|
||||
time.sleep(2)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user