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

create_l4 autopep8

Browse Source 
Signed-off-by: Matthew Stidham <stidmatt@gmail.com>
This commit is contained in:
Matthew Stidham
2021-11-10 12:15:16 -08:00
parent c94839f0d5
commit 9c42b1ca18
1 changed files with 64 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
py-scripts/create_l4.py
View File

@@ -24,15 +24,28 @@ TestGroupProfile = realm.TestGroupProfile
class CreateL4(Realm): class CreateL4(Realm):
def __init__(self, def __init__(
ssid, security, password, sta_list, name_prefix, upstream, radio, self,
host="localhost", port=8080, mode=0, ap=None, ssid,
side_a_min_rate=56, side_a_max_rate=0, security,
side_b_min_rate=56, side_b_max_rate=0, password,
number_template="00000", use_ht160=False, sta_list,
_debug_on=False, name_prefix,
_exit_on_error=False, upstream,
_exit_on_fail=False): radio,
host="localhost",
port=8080,
mode=0,
ap=None,
side_a_min_rate=56,
side_a_max_rate=0,
side_b_min_rate=56,
side_b_max_rate=0,
number_template="00000",
use_ht160=False,
_debug_on=False,
_exit_on_error=False,
_exit_on_fail=False):
super().__init__(host, port) super().__init__(host, port)
self.upstream = upstream self.upstream = upstream
self.host = host self.host = host
@@ -61,7 +74,9 @@ class CreateL4(Realm):
self.station_profile.mode = mode self.station_profile.mode = mode
if self.ap is not None: if self.ap is not None:
self.station_profile.set_command_param("add_sta", "ap", self.ap) self.station_profile.set_command_param("add_sta", "ap", self.ap)
# self.station_list= LFUtils.portNameSeries(prefix_="sta", start_id_=0, end_id_=2, padding_number_=10000, radio='wiphy0') #Make radio a user defined variable from terminal. # self.station_list= LFUtils.portNameSeries(prefix_="sta", start_id_=0,
# end_id_=2, padding_number_=10000, radio='wiphy0') #Make radio a user
# defined variable from terminal.
self.cx_profile.host = self.host self.cx_profile.host = self.host
self.cx_profile.port = self.port self.cx_profile.port = self.port
@@ -74,23 +89,33 @@ class CreateL4(Realm):
def cleanup(self): def cleanup(self):
self.cx_profile.cleanup() self.cx_profile.cleanup()
self.station_profile.cleanup() self.station_profile.cleanup()
LFUtils.wait_until_ports_disappear(base_url=self.lfclient_url, LFUtils.wait_until_ports_disappear(
port_list=self.station_profile.station_names, base_url=self.lfclient_url,
debug=self.debug) port_list=self.station_profile.station_names,
debug=self.debug)
def build(self): def build(self):
# Build stations # Build stations
self.station_profile.use_security(self.security, self.ssid, self.password) self.station_profile.use_security(
self.security, self.ssid, self.password)
self.station_profile.set_number_template(self.number_template) self.station_profile.set_number_template(self.number_template)
print("Creating stations") print("Creating stations")
self.station_profile.set_command_flag("add_sta", "create_admin_down", 1) self.station_profile.set_command_flag(
self.station_profile.set_command_param("set_port", "report_timer", 1500) "add_sta", "create_admin_down", 1)
self.station_profile.set_command_param(
"set_port", "report_timer", 1500)
self.station_profile.set_command_flag("set_port", "rpt_timer", 1) self.station_profile.set_command_flag("set_port", "rpt_timer", 1)
self.station_profile.create(radio=self.radio, sta_names_=self.sta_list, debug=self.debug) self.station_profile.create(
radio=self.radio,
sta_names_=self.sta_list,
debug=self.debug)
self._pass("PASS: Station build finished") self._pass("PASS: Station build finished")
self.cx_profile.create(ports=self.station_profile.station_names, sleep_time=.5, debug_=self.debug, self.cx_profile.create(
suppress_related_commands_=True) ports=self.station_profile.station_names,
sleep_time=.5,
debug_=self.debug,
suppress_related_commands_=True)
def main(): def main():
@@ -140,8 +165,14 @@ python3 ./layer4.py
required_args = group required_args = group
break break
if required_args is not None: if required_args is not None:
required_args.add_argument('--a_min', help='--a_min bps rate minimum for side_a', default=256000) required_args.add_argument(
required_args.add_argument('--b_min', help='--b_min bps rate minimum for side_b', default=256000) '--a_min',
help='--a_min bps rate minimum for side_a',
default=256000)
required_args.add_argument(
'--b_min',
help='--b_min bps rate minimum for side_b',
default=256000)
optional_args = None optional_args = None
for group in parser._action_groups: for group in parser._action_groups:
@@ -149,16 +180,24 @@ python3 ./layer4.py
optional_args = group optional_args = group
break break
if optional_args is not None: if optional_args is not None:
optional_args.add_argument('--mode', help='Used to force mode of stations', default=0) optional_args.add_argument(
optional_args.add_argument('--ap', help='Used to force a connection to a particular AP') '--mode',
help='Used to force mode of stations',
default=0)
optional_args.add_argument(
'--ap', help='Used to force a connection to a particular AP')
args = parser.parse_args() args = parser.parse_args()
num_sta = 2 num_sta = 2
if (args.num_stations is not None) and (int(args.num_stations) > 0): if (args.num_stations is not None) and (int(args.num_stations) > 0):
num_sta = int(args.num_stations) num_sta = int(args.num_stations)
station_list = LFUtils.portNameSeries(prefix_="sta", start_id_=0, end_id_=num_sta - 1, padding_number_=10000, station_list = LFUtils.portNameSeries(
radio=args.radio) prefix_="sta",
start_id_=0,
end_id_=num_sta - 1,
padding_number_=10000,
radio=args.radio)
ip_var_test = CreateL4(host=args.mgr, ip_var_test = CreateL4(host=args.mgr,
port=args.mgr_port, port=args.mgr_port,
number_template="0000", number_template="0000",