mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-10-31 10:48:02 +00:00
125 lines
3.6 KiB
Python
Executable File
125 lines
3.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
This script updates a Device Under Test (DUT) entry in the LANforge test scenario.
|
|
A common reason to use this would be to update MAC addresses in a DUT when you switch
|
|
between different items of the same make/model of a DUT.
|
|
"""
|
|
import sys
|
|
|
|
if sys.version_info[0] != 3:
|
|
print("This script requires Python 3")
|
|
exit(1)
|
|
|
|
if 'py-json' not in sys.path:
|
|
sys.path.append('../py-json')
|
|
|
|
import argparse
|
|
import pprint
|
|
from pprint import pprint
|
|
from LANforge.lfcli_base import LFCliBase
|
|
from LANforge.LFUtils import *
|
|
from LANforge import add_dut
|
|
from LANforge import LFUtils
|
|
import argparse
|
|
import realm
|
|
import time
|
|
import datetime
|
|
|
|
|
|
class UpdateDUT(LFCliBase):
|
|
def __init__(self, host, port,
|
|
_debug_on=False,
|
|
_exit_on_error=False,
|
|
_exit_on_fail=False):
|
|
super().__init__(host, port, _debug=_debug_on, _halt_on_error=_exit_on_error, _exit_on_fail=_exit_on_fail)
|
|
self.host = host
|
|
self.port = port
|
|
self.notes = "NA"
|
|
self.params = {}
|
|
self.flags = 0x0
|
|
self.flags_mask = 0x0
|
|
self.data = {}
|
|
self.url = "/cli-json/add_dut"
|
|
|
|
def build(self):
|
|
|
|
for param in self.params:
|
|
print("param: %s: %s"%(param, self.params[param]))
|
|
|
|
for flag in self.flags:
|
|
print("flags: %s: %s"%(flag, self.flags[flag]))
|
|
|
|
def start(self, print_pass=False, print_fail=False):
|
|
self.json_post(self.url, self.data)
|
|
self._pass("DUT updated")
|
|
pass
|
|
|
|
def stop(self):
|
|
pass
|
|
|
|
def cleanup(self, sta_list):
|
|
pass
|
|
|
|
def main():
|
|
lfjson_host = "localhost"
|
|
lfjson_port = 8080
|
|
|
|
param_string = " "
|
|
for param in add_dut.dut_params:
|
|
param_string += "%s, "%param.name
|
|
|
|
flags_string = " "
|
|
for flag in add_dut.dut_flags:
|
|
flags_string += "%s, "%flag.name
|
|
|
|
parser = LFCliBase.create_bare_argparse( prog='update_dut.py',
|
|
formatter_class=argparse.RawTextHelpFormatter,
|
|
description='''{file}
|
|
--------------------
|
|
Generic command layout:
|
|
python ./{file} --dut [DUT name] # update existing DUT record
|
|
--entry [key,value] # update/add entry by specifying key and value
|
|
--flag [flag,0|1] # toggle a flag on 1 or off 0
|
|
--notes "going to mars...."
|
|
DUT Parameters:
|
|
{params}
|
|
DUT Flags:
|
|
{flags}
|
|
|
|
Command Line Example:
|
|
python3 {file} --mgr 192.168.100.24 --update Pathfinder \
|
|
--entry MAC1,"00:00:ae:f0:b1:b9" \
|
|
--notes "build 2901" \
|
|
--flag STA_MODE,0
|
|
--flag AP_MODE,1
|
|
|
|
'''.format(file=__file__, params=param_string, flags=flags_string),
|
|
epilog="See",
|
|
)
|
|
parser.add_argument("-d", "--dut", type=str, help="name of DUT record")
|
|
parser.add_argument("-p", "--param", type=str, action="append", help="name,value pair to set parameter")
|
|
parser.add_argument("-f", "--flag", type=str, action="append", help="name,1/0/True/False pair to turn parameter on or off")
|
|
parser.add_argument("--notes", type=str, help="add notes to the DUT")
|
|
args = parser.parse_args()
|
|
|
|
update_dut = UpdateDUT(args.mgr, lfjson_port, _debug_on=args.debug)
|
|
pprint.pprint(args)
|
|
for param in args.param:
|
|
(name,value) = param.split(",")
|
|
update_dut.params[name] = value
|
|
|
|
for flag in args.flags:
|
|
(name,value) = flag.split(",")
|
|
update_dut.flags[name] = (False,True)[value]
|
|
update_dut.flags_mask[name] = True
|
|
|
|
if (args.notes is not None) and (args.notes != ""):
|
|
update_dut.notes = args.notes
|
|
|
|
update_dut.build()
|
|
update_dut.start()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|