Templates Datacom DmOS

datacom/datacom_dmos/dmos_hardware_monitor_mib_psu.py

@@ -0,0 +1,86 @@
+#!/usr/bin/env python3
+
+""" Script to discovery DmOS PSU objects in Zabbix """
+
+from subprocess import run
+from json import dumps
+from sys import argv
+from re import search, I
+
+
+def snmp_discovery_info(host, community, oid):
+    """ Return SNMP discovery info """
+
+    cmd = f"snmpbulkwalk -v 2c -c {community} {host} {oid}"
+    process = run(cmd, shell=True, capture_output=True, text=True, timeout=30)
+
+    snmp_info = []
+    snmp_error = ''
+    while True:
+        return_code = process.returncode
+        if return_code is not None:
+            for output in process.stdout.rstrip().split('\n'):
+                if search(r'no such', output, I):
+                    snmp_error = output.strip()
+                    break
+                else:
+                    snmp_error = process.stderr.strip()
+                snmp_info.append(output.strip())
+            break
+
+    return snmp_info, snmp_error
+
+def convert_snmp_value_string_to_decimal(snmp_value):
+    """ Return object size and dotted decimal for use in SNMP get """
+
+    snmp_value_dec = ''
+
+    snmp_value_size = len(snmp_value)
+
+    for index in range(snmp_value_size):
+        char_decimal = int(''.join(str(ord(snmp_value[index]))))
+
+        if index == snmp_value_size-1:
+            snmp_value_dec += str(char_decimal)
+        else:
+            snmp_value_dec += str(char_decimal) + '.'
+
+    return snmp_value_size, snmp_value_dec
+
+def main():
+    """ Main function """
+
+    host = argv[1]
+    community = argv[2]
+
+    oid = 'DMOS-HW-MONITOR-MIB::psuStatus'
+    snmp_info = snmp_discovery_info(host, community, oid)
+
+    number_snmp_info = len(snmp_info[0])
+    json_data = []
+
+    ret_error = snmp_info[1]
+    snmp_info = snmp_info[0]
+
+    if ret_error != '':
+        print(ret_error)
+        exit(1)
+
+    for i in range(number_snmp_info):
+        object = {}
+
+        regex = r'%s\.\'\.([a-zA-Z0-9\/]{1,50})\'\s.*\s(.*)$' % oid
+        output_regex = search(regex, snmp_info[i])
+
+        object['{#PSU_SLOT_NAME}'] = output_regex.group(1)
+
+        snmp_value_decimal = convert_snmp_value_string_to_decimal(output_regex.group(1))
+        object['{#PSU_SLOT_NAME_LEN}'] = snmp_value_decimal[0]
+        object['{#PSU_SLOT_NAME_DEC}'] = snmp_value_decimal[1]
+
+        json_data.append(object)
+
+    print(dumps({"data": json_data}, indent=4))
+
+
+main()