171 lines
4.9 KiB
Python
Executable File
171 lines
4.9 KiB
Python
Executable File
#
|
|
# psuutil.py
|
|
# Platform-specific PSU status interface for SONiC
|
|
#
|
|
|
|
|
|
import os.path
|
|
import subprocess
|
|
import logging
|
|
|
|
try:
|
|
from sonic_psu.psu_base import PsuBase
|
|
except ImportError as e:
|
|
raise ImportError(str(e) + "- required module not found")
|
|
|
|
DEBUG = False
|
|
|
|
def show_log(txt):
|
|
if DEBUG == True:
|
|
print("[IX2]"+txt)
|
|
return
|
|
|
|
def exec_cmd(cmd, show):
|
|
logging.info('Run :'+cmd)
|
|
try:
|
|
output = subprocess.check_output(cmd, shell=True)
|
|
show_log (cmd +"output:"+str(output))
|
|
except subprocess.CalledProcessError as e:
|
|
logging.info("Failed :"+cmd)
|
|
if show:
|
|
print("Failed :"+cmd +"returncode = {}, err msg: {}".format(e.returncode, e.output))
|
|
return output
|
|
|
|
def my_log(txt):
|
|
if DEBUG == True:
|
|
print("[QUANTA DBG]: "+txt)
|
|
return
|
|
|
|
def log_os_system(cmd, show):
|
|
logging.info('Run :'+cmd)
|
|
status = 1
|
|
output = ""
|
|
try:
|
|
output = subprocess.check_output(cmd, shell=True)
|
|
my_log (cmd +"output:"+str(output))
|
|
except subprocess.CalledProcessError as e:
|
|
logging.info('Failed :'+cmd)
|
|
if show:
|
|
print("Failed :"+cmd +"returncode = {}, err msg: {}".format(e.returncode, e.output))
|
|
return output
|
|
|
|
def gpio16_exist():
|
|
ls = log_os_system("ls /sys/class/gpio/ | grep gpio16", 0)
|
|
logging.info('mods:'+ls)
|
|
if len(ls) ==0:
|
|
return False
|
|
|
|
def gpio17_exist():
|
|
ls = log_os_system("ls /sys/class/gpio/ | grep gpio17", 0)
|
|
logging.info('mods:'+ls)
|
|
if len(ls) ==0:
|
|
return False
|
|
|
|
def gpio19_exist():
|
|
ls = log_os_system("ls /sys/class/gpio/ | grep gpio19", 0)
|
|
logging.info('mods:'+ls)
|
|
if len(ls) ==0:
|
|
return False
|
|
|
|
def gpio20_exist():
|
|
ls = log_os_system("ls /sys/class/gpio/ | grep gpio20", 0)
|
|
logging.info('mods:'+ls)
|
|
if len(ls) ==0:
|
|
return False
|
|
|
|
class PsuUtil(PsuBase):
|
|
"""Platform-specific PSUutil class"""
|
|
|
|
SYSFS_PSU_PRESENT_DIR = ["/sys/class/gpio/gpio16",
|
|
"/sys/class/gpio/gpio19"]
|
|
|
|
SYSFS_PSU_POWERGOOD_DIR = ["/sys/class/gpio/gpio17",
|
|
"/sys/class/gpio/gpio20"]
|
|
def __init__(self):
|
|
PsuBase.__init__(self)
|
|
|
|
if gpio16_exist() == False:
|
|
output = exec_cmd("echo 16 > /sys/class/gpio/export ", 1)
|
|
output = exec_cmd("echo in > /sys/class/gpio/gpio16/direction ", 1)
|
|
|
|
if gpio17_exist() == False:
|
|
output = exec_cmd("echo 17 > /sys/class/gpio/export ", 1)
|
|
output = exec_cmd("echo in > /sys/class/gpio/gpio17/direction ", 1)
|
|
|
|
if gpio19_exist() == False:
|
|
output = exec_cmd("echo 19 > /sys/class/gpio/export ", 1)
|
|
output = exec_cmd("echo in > /sys/class/gpio/gpio19/direction ", 1)
|
|
|
|
if gpio20_exist() == False:
|
|
output = exec_cmd("echo 20 > /sys/class/gpio/export ", 1)
|
|
output = exec_cmd("echo in > /sys/class/gpio/gpio20/direction ", 1)
|
|
|
|
# Get sysfs attribute
|
|
def get_attr_value(self, attr_path):
|
|
|
|
retval = 'ERR'
|
|
if (not os.path.isfile(attr_path)):
|
|
return retval
|
|
|
|
try:
|
|
with open(attr_path, 'r') as fd:
|
|
retval = fd.read()
|
|
except Exception as error:
|
|
logging.error("Unable to open ", attr_path, " file !")
|
|
|
|
retval = retval.rstrip('\r\n')
|
|
return retval
|
|
|
|
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
|
|
"""
|
|
MAX_PSUS = 2
|
|
return MAX_PSUS
|
|
|
|
def get_psu_status(self, index):
|
|
"""
|
|
Retrieves the oprational status of power supply unit (PSU) defined
|
|
by index <index>
|
|
:param index: An integer, index of the PSU of which to query status
|
|
:return: Boolean, True if PSU is operating properly, False if PSU is\
|
|
faulty
|
|
"""
|
|
status = 0
|
|
attr_file = 'value'
|
|
attr_path = self.SYSFS_PSU_POWERGOOD_DIR[index-1] +'/' + attr_file
|
|
|
|
attr_value = self.get_attr_value(attr_path)
|
|
|
|
if (attr_value != 'ERR'):
|
|
attr_value = int(attr_value, 16)
|
|
# Check for PSU status
|
|
if (attr_value == 1):
|
|
status = 1
|
|
|
|
return status
|
|
|
|
def get_psu_presence(self, index):
|
|
"""
|
|
Retrieves the presence status of power supply unit (PSU) defined
|
|
by index <index>
|
|
:param index: An integer, index of the PSU of which to query status
|
|
:return: Boolean, True if PSU is plugged, False if not
|
|
"""
|
|
status = 0
|
|
psu_absent = 0
|
|
attr_file ='value'
|
|
attr_path = self.SYSFS_PSU_PRESENT_DIR[index-1] +'/' + attr_file
|
|
|
|
attr_value = self.get_attr_value(attr_path)
|
|
|
|
if (attr_value != 'ERR'):
|
|
attr_value = int(attr_value, 16)
|
|
# Check for PSU presence
|
|
if (attr_value == 0):
|
|
status = 1
|
|
|
|
return status
|
|
|