99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
|
#
|
||
|
|
# ssd_health
|
||
|
|
#
|
||
|
|
|
||
|
|
from sonic_platform_base.sonic_ssd.ssd_base import SsdBase
|
||
|
|
from subprocess import Popen, PIPE
|
||
|
|
from re import findall
|
||
|
|
from os.path import exists
|
||
|
|
|
||
|
|
INNODISK = "iSmart -d {}"
|
||
|
|
NOT_AVAILABLE = "N/A"
|
||
|
|
|
||
|
|
class SsdUtil(SsdBase):
|
||
|
|
|
||
|
|
def __init__(self, diskdev):
|
||
|
|
"""
|
||
|
|
Constructor
|
||
|
|
Args:
|
||
|
|
diskdev: Linux device name to get parameters for
|
||
|
|
"""
|
||
|
|
if not isinstance(diskdev, str):
|
||
|
|
raise TypeError("disk dev type wrong {}".format(type(diskdev)))
|
||
|
|
|
||
|
|
if not exists(diskdev):
|
||
|
|
raise RuntimeError("disk dev {} not found".format(diskdev))
|
||
|
|
|
||
|
|
self.model = NOT_AVAILABLE
|
||
|
|
self.serial = NOT_AVAILABLE
|
||
|
|
self.firmware = NOT_AVAILABLE
|
||
|
|
self.temperature = NOT_AVAILABLE
|
||
|
|
self.health = NOT_AVAILABLE
|
||
|
|
|
||
|
|
self.ssd_info = self._execute_shell(INNODISK.format(diskdev))
|
||
|
|
|
||
|
|
self.model = self._parse_re(r'Model Name:\s*(.+?)\n', self.ssd_info)
|
||
|
|
self.serial = self._parse_re(r'Serial Number:\s*(.+?)\n', self.ssd_info)
|
||
|
|
self.firmware = self._parse_re(r'FW Version:\s*(.+?)\n', self.ssd_info)
|
||
|
|
self.temperature = self._parse_re(r'Temperature\s*\[\s*(.+?)\]', self.ssd_info)
|
||
|
|
self.health = self._parse_re(r'Health:\s*(.+?)', self.ssd_info)
|
||
|
|
|
||
|
|
def _execute_shell(self, cmd):
|
||
|
|
process = Popen(cmd.split(), universal_newlines=True, stdout=PIPE)
|
||
|
|
output, _ = process.communicate()
|
||
|
|
return output
|
||
|
|
|
||
|
|
def _parse_re(self, pattern, buffer):
|
||
|
|
res_list = findall(pattern, buffer)
|
||
|
|
return res_list[0] if res_list else NOT_AVAILABLE
|
||
|
|
|
||
|
|
def get_health(self):
|
||
|
|
"""
|
||
|
|
Retrieves current disk health in percentages
|
||
|
|
Returns:
|
||
|
|
A float number of current ssd health
|
||
|
|
e.g. 83.5
|
||
|
|
"""
|
||
|
|
return self.health
|
||
|
|
|
||
|
|
def get_temperature(self):
|
||
|
|
"""
|
||
|
|
Retrieves current disk temperature in Celsius
|
||
|
|
Returns:
|
||
|
|
A float number of current temperature in Celsius
|
||
|
|
e.g. 40.1
|
||
|
|
"""
|
||
|
|
return self.temperature
|
||
|
|
|
||
|
|
def get_model(self):
|
||
|
|
"""
|
||
|
|
Retrieves model for the given disk device
|
||
|
|
Returns:
|
||
|
|
A string holding disk model as provided by the manufacturer
|
||
|
|
"""
|
||
|
|
return self.model
|
||
|
|
|
||
|
|
def get_firmware(self):
|
||
|
|
"""
|
||
|
|
Retrieves firmware version for the given disk device
|
||
|
|
Returns:
|
||
|
|
A string holding disk firmware version as provided by the manufacturer
|
||
|
|
"""
|
||
|
|
return self.firmware
|
||
|
|
|
||
|
|
def get_serial(self):
|
||
|
|
"""
|
||
|
|
Retrieves serial number for the given disk device
|
||
|
|
Returns:
|
||
|
|
A string holding disk serial number as provided by the manufacturer
|
||
|
|
"""
|
||
|
|
return self.serial
|
||
|
|
|
||
|
|
def get_vendor_output(self):
|
||
|
|
"""
|
||
|
|
Retrieves vendor specific data for the given disk device
|
||
|
|
Returns:
|
||
|
|
A string holding some vendor specific disk information
|
||
|
|
"""
|
||
|
|
return self.ssd_info
|