JSON: script for updating dut entry, WIP: needs realm additions

py-scripts/update_dut.py

@@ -0,0 +1,96 @@
+#!/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
+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.params     = {}
+        self.flags      = 0x0
+        self.flags_mask = 0x0
+
+    def add_flag(self, ):
+        pass
+
+    def build(self):
+        pass
+
+    def start(self, print_pass=False, print_fail=False):
+        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,
+        epilog="Update parameters of a DUT record",
+        description='''{file}
+--------------------
+Generic command layout:
+python ./{file} --create [name] # create new DUT record
+--update [name]   # update existing DUT record
+--entry [key,value] # update/add entry by specifying key and value
+
+Command Line Example: 
+python3 ./{file} --mgr 192.168.100.24 --update Pathfinder --entry MAC1,"00:00:ae:f0:b1:b9" --entry api,"build 2901"
+
+DUT Parameters:
+{params}
+DUT Flags:
+{flags}
+'''.format(file=__file__, params=param_string, flags=flags_string)
+    )
+
+    args = parser.parse_args()
+
+    update_dut = UpdateDUT(args.mgr, lfjson_port, _debug_on=args.debug)
+
+    update_dut.build()
+    update_dut.start()
+
+
+if __name__ == "__main__":
+    main()