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

example_security_connection autopep8

Browse Source 
Signed-off-by: Matthew Stidham <stidmatt@gmail.com>
This commit is contained in:
Matthew Stidham
2021-11-10 14:58:09 -08:00
parent 37872d77d7
commit 772f2425ed
1 changed files with 32 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
py-scripts/example_security_connection.py
View File

@@ -8,7 +8,7 @@ if sys.version_info[0] != 3:
print("This script requires Python 3") print("This script requires Python 3")
exit(1) exit(1)
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../"))) sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
lfcli_base = importlib.import_module("py-json.LANforge.lfcli_base") lfcli_base = importlib.import_module("py-json.LANforge.lfcli_base")
@@ -19,7 +19,7 @@ Realm = realm.Realm
class IPv4Test(LFCliBase): class IPv4Test(LFCliBase):
def __init__(self, ssid, security, password, sta_list=None, ap=None, mode = 0, number_template="00000", host="localhost", port=8080,radio = "wiphy0",_debug_on=False, def __init__(self, ssid, security, password, sta_list=None, ap=None, mode=0, number_template="00000", host="localhost", port=8080, radio="wiphy0", _debug_on=False,
_exit_on_error=False, _exit_on_error=False,
_exit_on_fail=False): _exit_on_fail=False):
super().__init__(host, port, _debug=_debug_on, _exit_on_fail=_exit_on_fail) super().__init__(host, port, _debug=_debug_on, _exit_on_fail=_exit_on_fail)
@@ -35,43 +35,49 @@ class IPv4Test(LFCliBase):
self.timeout = 120 self.timeout = 120
self.number_template = number_template self.number_template = number_template
self.debug = _debug_on self.debug = _debug_on
self.local_realm = realm.Realm(lfclient_host=self.host, lfclient_port=self.port) self.local_realm = realm.Realm(
lfclient_host=self.host, lfclient_port=self.port)
self.station_profile = self.local_realm.new_station_profile() self.station_profile = self.local_realm.new_station_profile()
self.station_profile.lfclient_url = self.lfclient_url self.station_profile.lfclient_url = self.lfclient_url
self.station_profile.ssid = self.ssid self.station_profile.ssid = self.ssid
self.station_profile.ssid_pass = self.password self.station_profile.ssid_pass = self.password
self.station_profile.mode =self.mode self.station_profile.mode = self.mode
self.station_profile.security = self.security self.station_profile.security = self.security
self.station_profile.number_template_ = self.number_template self.station_profile.number_template_ = self.number_template
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)
def build(self): def build(self):
# Build stations # Build stations
#print("We've gotten into the build stations function") #print("We've gotten into the build stations function")
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.station_profile.admin_up() self.station_profile.admin_up()
if self.local_realm.wait_for_ip(station_list=self.sta_list, debug=self.debug, timeout_sec=30): if self.local_realm.wait_for_ip(station_list=self.sta_list, debug=self.debug, timeout_sec=30):
self._pass("Station build finished") self._pass("Station build finished")
self.exit_success() self.exit_success()
else: else:
self._fail("Stations not able to acquire IP. Please check network input.") self._fail(
"Stations not able to acquire IP. Please check network input.")
self.exit_fail() self.exit_fail()
def cleanup(self, sta_list): def cleanup(self, sta_list):
self.station_profile.cleanup(sta_list) self.station_profile.cleanup(sta_list)
LFUtils.wait_until_ports_disappear(base_url=self.lfclient_url, port_list=sta_list, LFUtils.wait_until_ports_disappear(base_url=self.lfclient_url, port_list=sta_list,
debug=self.debug) debug=self.debug)
def main(): def main():
parser = LFCliBase.create_basic_argparse( parser = LFCliBase.create_basic_argparse(
@@ -97,18 +103,19 @@ def main():
--passwd admin123-wpa3 --passwd admin123-wpa3
--debug --debug
''') ''')
required=None required = None
for agroup in parser._action_groups: for agroup in parser._action_groups:
if agroup.title == "required arguments": if agroup.title == "required arguments":
required = agroup required = agroup
#if required is not None: # if required is not None:
optional = None optional = None
for agroup in parser._action_groups: for agroup in parser._action_groups:
if agroup.title == "optional arguments": if agroup.title == "optional arguments":
optional = agroup optional = agroup
if optional is not None: if optional is not None:
optional.add_argument('--mode',help=LFCliBase.Help_Mode) optional.add_argument('--mode', help=LFCliBase.Help_Mode)
optional.add_argument('--ap',help='Add BSSID of access point to connect to') optional.add_argument(
'--ap', help='Add BSSID of access point to connect to')
args = parser.parse_args() args = parser.parse_args()
num_sta = 2 num_sta = 2
@@ -117,18 +124,19 @@ def main():
num_sta = num_stations_converted num_sta = num_stations_converted
station_list = LFUtils.portNameSeries(prefix_="sta", station_list = LFUtils.portNameSeries(prefix_="sta",
start_id_=0, start_id_=0,
end_id_=num_sta-1, end_id_=num_sta-1,
padding_number_=10000, padding_number_=10000,
radio=args.radio) radio=args.radio)
ip_test = IPv4Test(host=args.mgr, port=args.mgr_port, ip_test = IPv4Test(host=args.mgr, port=args.mgr_port,
ssid=args.ssid, password=args.passwd, ssid=args.ssid, password=args.passwd,
radio=args.radio, mode= args.mode, radio=args.radio, mode=args.mode,
security=args.security, sta_list=station_list, security=args.security, sta_list=station_list,
ap=args.ap) ap=args.ap)
ip_test.cleanup(station_list) ip_test.cleanup(station_list)
ip_test.timeout = 60 ip_test.timeout = 60
ip_test.build() ip_test.build()
if __name__ == "__main__": if __name__ == "__main__":
main() main()