mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-10-31 18:58:01 +00:00
78 lines
2.6 KiB
Python
Executable File
78 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import pprint
|
|
import sys
|
|
import os
|
|
|
|
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'))
|
|
|
|
import argparse
|
|
from LANforge.lfcli_base import LFCliBase
|
|
from LANforge import LFUtils
|
|
import realm
|
|
|
|
parser = LFCliBase.create_bare_argparse(
|
|
prog='scenario.py',
|
|
formatter_class=argparse.RawTextHelpFormatter,
|
|
epilog='''Load a database file and control test groups\n''',
|
|
description='''scenario.py
|
|
--------------------
|
|
Generic command example:
|
|
scenario.py --load db1 --action overwrite --clean_dut --clean_chambers
|
|
|
|
scenario.py --start test_group1
|
|
|
|
scenario.py --quiesce test_group1
|
|
|
|
scenario.py --stop test_group1
|
|
''')
|
|
|
|
group = parser.add_mutually_exclusive_group()
|
|
|
|
parser.add_argument('--load', help='name of database to load', default=None)
|
|
|
|
parser.add_argument('--action', help='action to take with database {overwrite | append}', default="overwrite")
|
|
|
|
parser.add_argument('--clean_dut',
|
|
help='use to cleanup DUT will be when overwrite is selected, otherwise they will be kept',
|
|
action="store_true")
|
|
|
|
parser.add_argument('--clean_chambers',
|
|
help='use to cleanup Chambers will be when overwrite is selected, otherwise they will be kept',
|
|
action="store_true")
|
|
|
|
group.add_argument('--start', help='name of test group to start', default=None)
|
|
group.add_argument('--quiesce', help='name of test group to quiesce', default=None)
|
|
group.add_argument('--stop', help='name of test group to stop', default=None)
|
|
args = parser.parse_args()
|
|
|
|
local_realm = realm.Realm(lfclient_host=args.mgr, lfclient_port=args.mgr_port, debug_=args.debug)
|
|
|
|
if args.load is not None:
|
|
data = {
|
|
"name": args.load,
|
|
"action": args.action,
|
|
"clean_dut": "no",
|
|
"clean_chambers": "no"
|
|
}
|
|
if args.clean_dut:
|
|
data['clean_dut'] = "yes"
|
|
if args.clean_chambers:
|
|
data['clean_chambers'] = "yes"
|
|
print("Loading database %s" % args.load)
|
|
local_realm.json_post("/cli-json/load", data)
|
|
|
|
elif args.start is not None:
|
|
print("Starting test group %s..." % args.start)
|
|
local_realm.json_post("/cli-json/start_group", {"name": args.start})
|
|
elif args.stop is not None:
|
|
print("Stopping test group %s..." % args.stop)
|
|
local_realm.json_post("/cli-json/stop_group", {"name": args.stop})
|
|
elif args.quiesce is not None:
|
|
print("Quiescing test group %s..." % args.quiesce)
|
|
local_realm.json_post("/cli-json/quiesce_group", {"name": args.quiesce})
|