added attenuator library in realm.py

py-json/lf_attenmod.py

@@ -0,0 +1,70 @@
+#!/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("Creating 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")
+