You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
175 lines
4.8 KiB
175 lines
4.8 KiB
#! /usr/bin/env python
|
|
import subprocess
|
|
import sys
|
|
import os
|
|
import time
|
|
|
|
ld_file = "/tmp/megacli-logical.out"
|
|
pd_file = "/tmp/megacli-physical.out"
|
|
ld_cmd = "cat /opt/py/megacli-logical.out"
|
|
pd_cmd = "cat /opt/py/megacli-physical.out"
|
|
|
|
pd_dict = dict(slot='Slot Number', port_stat='Port status', size='Raw Size', data='Inquiry Data',
|
|
temp='Drive Temperature ', smart_err='Drive has flagged a S.M.A.R.T alert',
|
|
manufacter='Inquiry Data', model='Inquiry Data', serial='Inquiry Data')
|
|
|
|
ld_dict = dict(num='Virtual Drive:', raid_lvl='RAID Level ', size='Size', state='State', pd_count='Number Of Drives')
|
|
|
|
|
|
def f_ld_discover(cli_out):
|
|
drives = []
|
|
linesplit = ''
|
|
for line in cli_out:
|
|
if 'Virtual Drive:' in line:
|
|
linesplit = line.split(' ')
|
|
drives.append(str(linesplit[2]))
|
|
return drives
|
|
|
|
|
|
def f_pd_discover(cli_out):
|
|
drives = []
|
|
linesplit = ''
|
|
for line in cli_out:
|
|
if 'Slot Number:' in line:
|
|
linesplit = line.split(' ')
|
|
drives.append(str(linesplit[2]))
|
|
return drives
|
|
|
|
|
|
def f_json_print(inp, name):
|
|
first = True
|
|
print("{")
|
|
print("\t\"data\":[")
|
|
for key in inp:
|
|
if not first:
|
|
sys.stdout.write(",")
|
|
else:
|
|
first = False
|
|
sys.stdout.write("\n\t\t{\"{#" + name + "}\":\"" + key + "\"}")
|
|
print("\n\t]")
|
|
print("}")
|
|
|
|
|
|
def f_read_file(path):
|
|
cli_out = []
|
|
try:
|
|
f = open(path, 'r')
|
|
except Exception:
|
|
print(Exception)
|
|
return []
|
|
else:
|
|
for line in f.readlines():
|
|
cli_out.append(line.strip())
|
|
f.close()
|
|
return cli_out
|
|
|
|
|
|
def f_read_cli(cmd, outfile):
|
|
outdata, stderr = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True).communicate()
|
|
try:
|
|
f = open(outfile, 'w')
|
|
except Exception:
|
|
return 1
|
|
else:
|
|
f.write(outdata)
|
|
f.close()
|
|
return 0
|
|
|
|
|
|
def f_get_pd_val(slot, val, source):
|
|
data = f_read_file(source)
|
|
k = 0
|
|
out = ""
|
|
for line in data:
|
|
if pd_dict['slot'] in line:
|
|
# print(line.strip().split(':')[1].strip())
|
|
if line.strip().split(':')[1].strip() == slot:
|
|
k = 1
|
|
if k == 0:
|
|
continue
|
|
if k == 1:
|
|
if pd_dict[val] in line:
|
|
out = line.strip().split(':')[1].strip().split()
|
|
if out != "":
|
|
break
|
|
if len(out) == 1 and out[0].isdigit():
|
|
return out[0]
|
|
elif val == 'smart_err':
|
|
if out[0] == 'No':
|
|
return '0'
|
|
else:
|
|
return '1'
|
|
elif val == "manufacter":
|
|
return out[0].strip()
|
|
elif val == "model":
|
|
return out[1].strip()
|
|
elif val == "serial":
|
|
return out[2].strip()
|
|
elif val == 'size':
|
|
return out[0]
|
|
elif val == 'port_stat':
|
|
if out[0] == 'Active':
|
|
return '1'
|
|
else:
|
|
return '0'
|
|
elif val == 'temp':
|
|
return out[0][:-1]
|
|
|
|
|
|
def f_get_ld_val(num, val, source):
|
|
data = f_read_file(source)
|
|
k = 0
|
|
out = ""
|
|
for line in data:
|
|
if ld_dict['num'] in line:
|
|
if line.split(':')[1].strip().split()[0].strip() == num:
|
|
k = 1
|
|
if k == 0:
|
|
continue
|
|
if k == 1:
|
|
if ld_dict[val] in line:
|
|
out = line.split(':')[1].strip()
|
|
if out != "":
|
|
break
|
|
if val == 'raid_lvl':
|
|
return out
|
|
elif val == 'size':
|
|
return out.split()[0]
|
|
elif val == 'state':
|
|
if out.strip() == 'Optimal':
|
|
return '0'
|
|
else:
|
|
return '1'
|
|
elif val == 'pd_count':
|
|
return out.strip()
|
|
return out
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) == 1 or len(sys.argv) == 3 or len(sys.argv) > 4:
|
|
print("Wrong args. Cache will be updated if necessary")
|
|
if not os.path.exists(pd_file):
|
|
f_read_cli(pd_cmd, pd_file)
|
|
else:
|
|
stat_pd = os.stat(pd_file)
|
|
if (float(time.time()) - float(stat_pd.st_ctime)) > 50:
|
|
f_read_cli(pd_cmd, pd_file)
|
|
if not os.path.exists(ld_file):
|
|
f_read_cli(ld_cmd, ld_file)
|
|
else:
|
|
stat_ld = os.stat(ld_file)
|
|
if (float(time.time()) - float(stat_ld.st_ctime)) > 50:
|
|
f_read_cli(ld_cmd, ld_file)
|
|
|
|
if len(sys.argv) == 2:
|
|
if sys.argv[1] == "ld":
|
|
f_json_print(f_ld_discover(f_read_file(ld_file)), "LD")
|
|
if sys.argv[1] == "pd":
|
|
f_json_print(f_pd_discover(f_read_file(pd_file)), "PD")
|
|
elif len(sys.argv) == 4:
|
|
if sys.argv[1] == 'pd':
|
|
if sys.argv[3] in pd_dict.keys():
|
|
print(f_get_pd_val(sys.argv[2], sys.argv[3], pd_file))
|
|
if sys.argv[1] == 'ld':
|
|
if sys.argv[3] in ld_dict.keys():
|
|
print(f_get_ld_val(sys.argv[2], sys.argv[3], ld_file))
|