Files
wlan-lanforge-scripts/py-json/lf_attenmod.py
2021-09-09 05:24:20 -06:00

71 lines
2.7 KiB
Python

#!/usr/bin/env python3
from LANforge.lfcli_base import LFCliBase
from LANforge import LFRequest
from LANforge import LFUtils
import time
class ATTENUATORProfile(LFCliBase):
def __init__(self, lfclient_host, lfclient_port, local_realm, debug_=False):
super().__init__(lfclient_host, lfclient_port, debug_)
self.local_realm = local_realm
self.lfclient_host = lfclient_host
self.COMMANDS = ["show_attenuators", "set_attenuator"]
self.atten_serno = ""
self.atten_idx = ""
self.atten_val = ""
self.atten_data = {
"shelf": 1,
"resource": 1,
"serno": None,
"atten_idx": None,
"val": None,
"mode": None,
"pulse_width_us5": None,
"pulse_interval_ms": None,
"pulse_count": None,
"pulse_time_ms": None
}
def set_command_param(self, command_name, param_name, param_value):
# we have to check what the param name is
if (command_name is None) or (command_name == ""):
return
if (param_name is None) or (param_name == ""):
return
if command_name not in self.COMMANDS:
raise ValueError("Command name name [%s] not defined in %s" % (command_name, self.COMMANDS))
if command_name == "set_attenuator":
self.atten_data[param_name] = param_value
def show(self, debug=False):
print("Show Attenuators.........")
response = self.json_get("/attenuators/")
time.sleep(0.01)
if response is None:
print(response)
raise ValueError("Cannot find any endpoints")
else:
attenuator_resp = response["attenuator"]
for key, val in attenuator_resp.items():
if key == "entity id":
serial_num = val.split(".")
print("Serial-num : %s" % serial_num[-1])
print("%s : %s" % (key, val))
print("\n")
def create(self, debug=False):
if len(self.atten_serno) == 0 or len(self.atten_idx) == 0 or len(self.atten_val) == 0:
print("ERROR: Must specify atten_serno, atten_idx, and atten_val when setting attenuator.\n")
print("Setting Attenuator...")
self.set_command_param("set_attenuator", "serno", self.atten_serno)
self.set_command_param("set_attenuator", "atten_idx", self.atten_idx)
self.set_command_param("set_attenuator", "val", self.atten_val)
set_attenuators = LFRequest.LFRequest(self.lfclient_url + "/cli-json/set_attenuator", debug_=debug)
set_attenuators.addPostData(self.atten_data)
time.sleep(0.01)
json_response = set_attenuators.jsonPost(debug)
time.sleep(10)
print("\n")