import subprocess

try:
    from sonic_psu.psu_base import PsuBase
except ImportError as e:
    raise ImportError(str(e) + "- required module not found")


class PsuUtil(PsuBase):
    """Platform-specific PSUutil class"""

    def __init__(self):
        PsuBase.__init__(self)

        self.psu_path = "/sys/bus/i2c/devices/{}-0058/"
        self.psu_oper_status = "in1_input"
        self.psu_presence = ["i2cget", "-y", "", "0x50", "0x00"]

    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):
        if index is None:
            return False
        Base_bus_number = 39
        status = 0
        # index from 1, psu attribute bus from 40
        try:
            with open(self.psu_path.format(index + Base_bus_number) + self.psu_oper_status, 'r') as power_status:
                if int(power_status.read()) == 0:
                    return False
                else:
                    status = 1
        except IOError:
            return False
        return status == 1

    def get_psu_presence(self, index):
        if index is None:
            return False
        Base_bus_number = 39
        status = 0
        self.psu_presence[2] = str(index + Base_bus_number)
        try:
            p = subprocess.Popen(self.psu_presence, stdout=subprocess.PIPE, universal_newlines=True)
            if p.stdout.readline() != None:
                status = 1
            p.close()
        except IOError:
            return False
        return status == 1