[device/celestica] Implement PSU APIs based on the new platform API (#2874)

device/celestica/x86_64-cel_e1031-r0/sonic_platform/chassis.py

@@ -17,6 +17,7 @@ import json
 try:
     from sonic_platform_base.chassis_base import ChassisBase
     from sonic_platform.fan import Fan
+    from sonic_platform.psu import Psu
 except ImportError as e:
     raise ImportError(str(e) + "- required module not found")
 
@@ -26,6 +27,7 @@ CONFIG_DB_PATH = "/etc/sonic/config_db.json"
 SMC_CPLD_PATH = "/sys/devices/platform/e1031.smc/version"
 MMC_CPLD_PATH = "/sys/devices/platform/e1031.smc/getreg"
 NUM_FAN = 3
+NUM_PSU = 2
 
 
 class Chassis(ChassisBase):
@@ -36,6 +38,9 @@ class Chassis(ChassisBase):
         for index in range(0, NUM_FAN):
             fan = Fan(index)
             self._fan_list.append(fan)
+        for index in range(0, NUM_PSU):
+            psu = Psu(index)
+            self._psu_list.append(psu)
         ChassisBase.__init__(self)
 
     def __get_register_value(self, path, register):

device/celestica/x86_64-cel_e1031-r0/sonic_platform/psu.py

@@ -0,0 +1,61 @@
+#!/usr/bin/env python
+
+#############################################################################
+# Celestica
+#
+# Module contains an implementation of SONiC Platform Base API and
+# provides the PSUs status which are available in the platform
+#
+#############################################################################
+
+import os.path
+import sonic_platform
+
+try:
+    from sonic_platform_base.psu_base import PsuBase
+    from sonic_platform.fan import Fan
+except ImportError as e:
+    raise ImportError(str(e) + "- required module not found")
+
+FAN_E1031_SPEED_PATH = "/sys/class/hwmon/hwmon{}/fan1_input"
+FAN_MAX_RPM = 11000
+
+
+class Psu(PsuBase):
+    """Platform-specific Psu class"""
+
+    def __init__(self, psu_index):
+        PsuBase.__init__(self)
+        self.index = psu_index
+
+    def get_fan(self):
+        """
+        Retrieves object representing the fan module contained in this PSU
+        Returns:
+            An object dervied from FanBase representing the fan module
+            contained in this PSU
+        """
+        fan_speed_path = FAN_E1031_SPEED_PATH.format(
+            str(self.index+3))
+        try:
+            with open(fan_speed_path) as fan_speed_file:
+                fan_speed_rpm = int(fan_speed_file.read())
+        except IOError:
+            fan_speed = 0
+
+        fan_speed = float(fan_speed_rpm)/FAN_MAX_RPM * 100
+        fan = Fan(0)
+        fan.fan_speed = int(fan_speed) if int(fan_speed) <= 100 else 100
+        return fan
+
+    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
+                   Note: Only support green and off
+        Returns:
+            bool: True if status LED state is set successfully, False if not
+        """
+        # Hardware not supported
+        return False

device/celestica/x86_64-cel_seastone-r0/sonic_platform/chassis.py

@@ -17,6 +17,7 @@ import json
 try:
     from sonic_platform_base.chassis_base import ChassisBase
     from sonic_platform.fan import Fan
+    from sonic_platform.psu import Psu
 except ImportError as e:
     raise ImportError(str(e) + "- required module not found")
 
@@ -24,6 +25,7 @@ BIOS_VERSION_PATH = "/sys/class/dmi/id/bios_version"
 GETREG_PATH = "/sys/devices/platform/dx010_cpld/getreg"
 CONFIG_DB_PATH = "/etc/sonic/config_db.json"
 NUM_FAN = 5
+NUM_PSU = 2
 CPLD_ADDR_MAPPING = {
     "CPLD1": "0x100",
     "CPLD2": "0x200",
@@ -41,6 +43,9 @@ class Chassis(ChassisBase):
         for index in range(0, NUM_FAN):
             fan = Fan(index)
             self._fan_list.append(fan)
+        for index in range(0, NUM_PSU):
+            psu = Psu(index)
+            self._psu_list.append(psu)
         ChassisBase.__init__(self)
 
     def __get_register_value(self, path, register):

device/celestica/x86_64-cel_seastone-r0/sonic_platform/psu.py

@@ -0,0 +1,78 @@
+#!/usr/bin/env python
+
+#############################################################################
+# Celestica
+#
+# Module contains an implementation of SONiC Platform Base API and
+# provides the PSUs status which are available in the platform
+#
+#############################################################################
+
+import os.path
+import sonic_platform
+
+try:
+    from sonic_platform_base.psu_base import PsuBase
+    from sonic_platform.fan import Fan
+except ImportError as e:
+    raise ImportError(str(e) + "- required module not found")
+
+FAN_DX010_SPEED_PATH = "/sys/class/hwmon/hwmon{}/fan1_input"
+GREEN_LED_PATH = "/sys/devices/platform/leds_dx010/leds/dx010:green:p-{}/brightness"
+FAN_MAX_RPM = 11000
+
+
+class Psu(PsuBase):
+    """Platform-specific Psu class"""
+
+    def __init__(self, psu_index):
+        PsuBase.__init__(self)
+        self.index = psu_index
+        self.green_led_path = GREEN_LED_PATH.format(self.index+1)
+
+    def get_fan(self):
+        """
+        Retrieves object representing the fan module contained in this PSU
+        Returns:
+            An object dervied from FanBase representing the fan module
+            contained in this PSU
+        """
+
+        fan_speed_path = FAN_DX010_SPEED_PATH.format(
+            str(self.index+8))
+        try:
+            with open(fan_speed_path) as fan_speed_file:
+                fan_speed_rpm = int(fan_speed_file.read())
+        except IOError:
+            fan_speed = 0
+
+        fan_speed = float(fan_speed_rpm)/FAN_MAX_RPM * 100
+        fan = Fan(0)
+        fan.fan_speed = int(fan_speed) if int(fan_speed) <= 100 else 100
+        return fan
+
+    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
+                   Note: Only support green and off
+        Returns:
+            bool: True if status LED state is set successfully, False if not
+        """
+
+        set_status_str = {
+            self.STATUS_LED_COLOR_GREEN: '1',
+            self.STATUS_LED_COLOR_OFF: '0'
+        }.get(color, None)
+
+        if not set_status_str:
+            return False
+
+        try:
+            with open(self.green_led_path, 'w') as file:
+                file.write(set_status_str)
+        except IOError:
+            return False
+
+        return True