286 lines
9.6 KiB
Python
286 lines
9.6 KiB
Python
#!/usr/bin/env python
|
|
|
|
#############################################################################
|
|
# Mellanox
|
|
#
|
|
# Module contains an implementation of SONiC Platform Base API and
|
|
# provides the PSUs status which are available in the platform
|
|
#
|
|
#############################################################################
|
|
|
|
try:
|
|
import os.path
|
|
from sonic_platform_base.psu_base import PsuBase
|
|
from sonic_daemon_base.daemon_base import Logger
|
|
from sonic_platform.fan import Fan
|
|
except ImportError as e:
|
|
raise ImportError (str(e) + "- required module not found")
|
|
|
|
LED_ON = '1'
|
|
LED_OFF = '0'
|
|
|
|
# Global logger class instance
|
|
logger = Logger()
|
|
|
|
psu_list = []
|
|
|
|
PSU_CURRENT = "current"
|
|
PSU_VOLTAGE = "voltage"
|
|
PSU_POWER = "power"
|
|
|
|
LED_PATH = "/var/run/hw-management/led/"
|
|
|
|
# SKUs with unplugable PSUs:
|
|
# 1. don't have psuX_status and should be treated as always present
|
|
# 2. don't have voltage, current and power values
|
|
hwsku_dict_with_unplugable_psu = ['ACS-MSN2010', 'ACS-MSN2100']
|
|
|
|
# in most SKUs the file psuX_curr, psuX_volt and psuX_power contain current, voltage and power data respectively.
|
|
# but there are exceptions which will be handled by the following dictionary
|
|
hwsku_dict_psu = {'ACS-MSN3700': 1, 'ACS-MSN3700C': 1, 'ACS-MSN3800': 1}
|
|
psu_profile_list = [
|
|
# default filename convention
|
|
{
|
|
PSU_CURRENT : "power/psu{}_curr",
|
|
PSU_VOLTAGE : "power/psu{}_volt",
|
|
PSU_POWER : "power/psu{}_power"
|
|
},
|
|
# for 3700, 3700c, 3800
|
|
{
|
|
PSU_CURRENT : "power/psu{}_curr",
|
|
PSU_VOLTAGE : "power/psu{}_volt_out2",
|
|
PSU_POWER : "power/psu{}_power"
|
|
}
|
|
]
|
|
|
|
class Psu(PsuBase):
|
|
"""Platform-specific Psu class"""
|
|
|
|
STATUS_LED_COLOR_ORANGE = "orange"
|
|
|
|
def __init__(self, psu_index, sku):
|
|
global psu_list
|
|
PsuBase.__init__(self)
|
|
# PSU is 1-based on Mellanox platform
|
|
self.index = psu_index + 1
|
|
psu_list.append(self.index)
|
|
self.psu_path = "/var/run/hw-management/"
|
|
psu_oper_status = "thermal/psu{}_pwr_status".format(self.index)
|
|
#psu_oper_status should always be present for all SKUs
|
|
self.psu_oper_status = os.path.join(self.psu_path, psu_oper_status)
|
|
|
|
if sku in hwsku_dict_psu:
|
|
filemap = psu_profile_list[hwsku_dict_psu[sku]]
|
|
else:
|
|
filemap = psu_profile_list[0]
|
|
|
|
if sku in hwsku_dict_with_unplugable_psu:
|
|
self.always_presence = True
|
|
self.psu_voltage = None
|
|
self.psu_current = None
|
|
self.psu_power = None
|
|
self.psu_presence = None
|
|
else:
|
|
self.always_presence = False
|
|
psu_voltage = filemap[PSU_VOLTAGE].format(self.index)
|
|
psu_voltage = os.path.join(self.psu_path, psu_voltage)
|
|
self.psu_voltage = psu_voltage
|
|
|
|
psu_current = filemap[PSU_CURRENT].format(self.index)
|
|
psu_current = os.path.join(self.psu_path, psu_current)
|
|
self.psu_current = psu_current
|
|
|
|
psu_power = filemap[PSU_POWER].format(self.index)
|
|
psu_power = os.path.join(self.psu_path, psu_power)
|
|
self.psu_power = psu_power
|
|
|
|
psu_presence = "thermal/psu{}_status".format(self.index)
|
|
psu_presence = os.path.join(self.psu_path, psu_presence)
|
|
self.psu_presence = psu_presence
|
|
|
|
fan = Fan(sku, psu_index, psu_index, True)
|
|
if fan.get_presence():
|
|
self._fan = fan
|
|
|
|
self.psu_green_led_path = "led_psu_green"
|
|
self.psu_red_led_path = "led_psu_red"
|
|
self.psu_orange_led_path = "led_psu_orange"
|
|
self.psu_led_cap_path = "led_psu_capability"
|
|
|
|
|
|
def _read_generic_file(self, filename, len):
|
|
"""
|
|
Read a generic file, returns the contents of the file
|
|
"""
|
|
result = 0
|
|
try:
|
|
with open(filename, 'r') as fileobj:
|
|
result = int(fileobj.read())
|
|
except Exception as e:
|
|
logger.log_info("Fail to read file {} due to {}".format(filename, repr(e)))
|
|
return result
|
|
|
|
|
|
def get_powergood_status(self):
|
|
"""
|
|
Retrieves the operational status of power supply unit (PSU) defined
|
|
|
|
Returns:
|
|
bool: True if PSU is operating properly, False if not
|
|
"""
|
|
status = self._read_generic_file(os.path.join(self.psu_path, self.psu_oper_status), 0)
|
|
|
|
return status == 1
|
|
|
|
|
|
def get_presence(self):
|
|
"""
|
|
Retrieves the presence status of power supply unit (PSU) defined
|
|
|
|
Returns:
|
|
bool: True if PSU is present, False if not
|
|
"""
|
|
if self.always_presence:
|
|
return self.always_presence
|
|
else:
|
|
status = self._read_generic_file(self.psu_presence, 0)
|
|
return status == 1
|
|
|
|
|
|
def get_voltage(self):
|
|
"""
|
|
Retrieves current PSU voltage output
|
|
|
|
Returns:
|
|
A float number, the output voltage in volts,
|
|
e.g. 12.1
|
|
"""
|
|
if self.psu_voltage is not None and self.get_powergood_status():
|
|
voltage = self._read_generic_file(self.psu_voltage, 0)
|
|
return float(voltage) / 1000
|
|
else:
|
|
return None
|
|
|
|
|
|
def get_current(self):
|
|
"""
|
|
Retrieves present electric current supplied by PSU
|
|
|
|
Returns:
|
|
A float number, the electric current in amperes, e.g 15.4
|
|
"""
|
|
if self.psu_current is not None and self.get_powergood_status():
|
|
amperes = self._read_generic_file(self.psu_current, 0)
|
|
return float(amperes) / 1000
|
|
else:
|
|
return None
|
|
|
|
def get_power(self):
|
|
"""
|
|
Retrieves current energy supplied by PSU
|
|
|
|
Returns:
|
|
A float number, the power in watts, e.g. 302.6
|
|
"""
|
|
if self.psu_power is not None and self.get_powergood_status():
|
|
power = self._read_generic_file(self.psu_power, 0)
|
|
return float(power) / 1000000
|
|
else:
|
|
return None
|
|
|
|
|
|
def _get_led_capability(self):
|
|
cap_list = None
|
|
try:
|
|
with open(os.path.join(LED_PATH, self.psu_led_cap_path), 'r') as psu_led_cap:
|
|
caps = psu_led_cap.read()
|
|
cap_list = caps.split()
|
|
except (ValueError, IOError):
|
|
status = 0
|
|
|
|
return cap_list
|
|
|
|
|
|
def set_status_led(self, color):
|
|
"""
|
|
Sets the state of the PSU status LED
|
|
|
|
Args:
|
|
color: A string representing the color with which to set the
|
|
PSU status LED
|
|
|
|
Returns:
|
|
bool: True if status LED state is set successfully, False if not
|
|
|
|
Notes:
|
|
Only one led for all PSUs.
|
|
"""
|
|
led_cap_list = self._get_led_capability()
|
|
if led_cap_list is None:
|
|
return False
|
|
|
|
status = False
|
|
try:
|
|
if color == self.STATUS_LED_COLOR_GREEN:
|
|
with open(os.path.join(LED_PATH, self.psu_green_led_path), 'w') as psu_led:
|
|
psu_led.write(LED_ON)
|
|
status = True
|
|
elif color == self.STATUS_LED_COLOR_RED:
|
|
# Some fan don't support red led but support orange led, in this case we set led to orange
|
|
if self.STATUS_LED_COLOR_RED in led_cap_list:
|
|
led_path = os.path.join(LED_PATH, self.psu_red_led_path)
|
|
elif self.STATUS_LED_COLOR_ORANGE in led_cap_list:
|
|
led_path = os.path.join(LED_PATH, self.psu_orange_led_path)
|
|
else:
|
|
return False
|
|
with open(led_path, 'w') as psu_led:
|
|
psu_led.write(LED_ON)
|
|
status = True
|
|
elif color == self.STATUS_LED_COLOR_OFF:
|
|
if self.STATUS_LED_COLOR_GREEN in led_cap_list:
|
|
with open(os.path.join(LED_PATH, self.psu_green_led_path), 'w') as psu_led:
|
|
psu_led.write(str(LED_OFF))
|
|
if self.STATUS_LED_COLOR_RED in led_cap_list:
|
|
with open(os.path.join(LED_PATH, self.psu_red_led_path), 'w') as psu_led:
|
|
psu_led.write(str(LED_OFF))
|
|
if self.STATUS_LED_COLOR_ORANGE in led_cap_list:
|
|
with open(os.path.join(LED_PATH, self.psu_orange_led_path), 'w') as psu_led:
|
|
psu_led.write(str(LED_OFF))
|
|
|
|
status = True
|
|
else:
|
|
status = False
|
|
except (ValueError, IOError):
|
|
status = False
|
|
|
|
return status
|
|
|
|
|
|
def get_status_led(self):
|
|
"""
|
|
Gets the state of the PSU status LED
|
|
|
|
Returns:
|
|
A string, one of the predefined STATUS_LED_COLOR_* strings above
|
|
"""
|
|
led_cap_list = self._get_led_capability()
|
|
if led_cap_list is None:
|
|
return self.STATUS_LED_COLOR_OFF
|
|
|
|
try:
|
|
with open(os.path.join(LED_PATH, self.psu_green_led_path), 'r') as psu_led:
|
|
if LED_OFF != psu_led.read().rstrip('\n'):
|
|
return self.STATUS_LED_COLOR_GREEN
|
|
if self.STATUS_LED_COLOR_RED in led_cap_list:
|
|
with open(os.path.join(LED_PATH, self.psu_red_led_path), 'r') as psu_led:
|
|
if LED_OFF != psu_led.read().rstrip('\n'):
|
|
return self.STATUS_LED_COLOR_RED
|
|
if self.STATUS_LED_COLOR_ORANGE in led_cap_list:
|
|
with open(os.path.join(LED_PATH, self.psu_orange_led_path), 'r') as psu_led:
|
|
if LED_OFF != psu_led.read().rstrip('\n'):
|
|
return self.STATUS_LED_COLOR_RED
|
|
except (ValueError, IOError) as e:
|
|
raise RuntimeError("Failed to read led status for psu due to {}".format(repr(e)))
|
|
|
|
return self.STATUS_LED_COLOR_OFF
|