From 51733d0a72fdaac02ed0c5fab8f6133aa7d18622 Mon Sep 17 00:00:00 2001
From: Andriy Kokhan <akokhan@barefootnetworks.com>
Date: Mon, 29 Jul 2019 08:56:11 -0700
Subject: [PATCH] [platform]: Implemented psuutil.py for BFN reference boards

Signed-off-by: Andriy Kokhan <akokhan@barefootnetworks.com>
---
 .../plugins/psuutil.py                        | 88 +++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 device/barefoot/x86_64-accton_wedge100bf_32x-r0/plugins/psuutil.py

diff --git a/device/barefoot/x86_64-accton_wedge100bf_32x-r0/plugins/psuutil.py b/device/barefoot/x86_64-accton_wedge100bf_32x-r0/plugins/psuutil.py
new file mode 100644
index 000000000000..734901c2adfa
--- /dev/null
+++ b/device/barefoot/x86_64-accton_wedge100bf_32x-r0/plugins/psuutil.py
@@ -0,0 +1,88 @@
+#!/usr/bin/env python
+
+try:
+    import os
+    import sys
+    import importlib
+
+    sys.path.append(os.path.dirname(__file__))
+    import pltfm_mgr_rpc
+    from pltfm_mgr_rpc.ttypes import *
+
+    from thrift.transport import TSocket
+    from thrift.transport import TTransport
+    from thrift.protocol import TBinaryProtocol
+    from thrift.protocol import TMultiplexedProtocol
+
+    from sonic_psu.psu_base import PsuBase
+except ImportError as e:
+    raise ImportError (str(e) + "- required module not found")
+
+thrift_server = 'localhost'
+transport = None
+pltfm_mgr = None
+
+class PsuUtil(PsuBase):
+    """Platform-specific PSUutil class"""
+
+    def __init__(self):
+        PsuBase.__init__(self)
+
+    def thrift_setup(self):
+        global thrift_server, transport, pltfm_mgr
+        transport = TSocket.TSocket(thrift_server, 9090)
+
+        transport = TTransport.TBufferedTransport(transport)
+        bprotocol = TBinaryProtocol.TBinaryProtocol(transport)
+
+        pltfm_mgr_client_module = importlib.import_module(".".join(["pltfm_mgr_rpc", "pltfm_mgr_rpc"]))
+        pltfm_mgr_protocol = TMultiplexedProtocol.TMultiplexedProtocol(bprotocol, "pltfm_mgr_rpc")
+        pltfm_mgr = pltfm_mgr_client_module.Client(pltfm_mgr_protocol)
+
+        transport.open()
+
+    def thrift_teardown(self):
+        global transport
+        transport.close()
+
+    def get_num_psus(self):
+        """
+        Retrieves the number of PSUs available on the device
+        :return: An integer, the number of PSUs available on the device
+        """
+        return 2
+
+    def get_psu_status(self, index):
+        """
+        Retrieves the oprational status of power supply unit (PSU) defined
+                by 1-based index <index>
+        :param index: An integer, 1-based index of the PSU of which to query status
+        :return: Boolean, True if PSU is operating properly, False if PSU is faulty
+        """
+        if index is None:
+            return False
+
+        global pltfm_mgr
+        self.thrift_setup()
+        psu_info = pltfm_mgr.pltfm_mgr_pwr_supply_info_get(index)
+        self.thrift_teardown()
+
+        return (psu_info.ffault == False)
+
+    def get_psu_presence(self, index):
+        """
+        Retrieves the presence status of power supply unit (PSU) defined
+                by 1-based index <index>
+        :param index: An integer, 1-based index of the PSU of which to query status
+        :return: Boolean, True if PSU is plugged, False if not
+        """
+        if index is None:
+            return False
+
+        global pltfm_mgr
+        self.thrift_setup()
+        status = pltfm_mgr.pltfm_mgr_pwr_supply_present_get(index)
+        self.thrift_teardown()
+
+        return status
+