[Mellanox] Add a trigger to set LED to blink (#8995)
Depends on Mellanox hw-mgmt 7.0010.3300 Why I did it Adjust LED logical according to hw-mgmt change. How I did it Add a trigger to set LED to blink. How to verify it Manual test
This commit is contained in:
parent
fa63c056d1
commit
2a3738ead5
@ -1,4 +1,6 @@
|
|||||||
import os
|
import os
|
||||||
|
import time
|
||||||
|
from . import utils
|
||||||
|
|
||||||
|
|
||||||
class Led(object):
|
class Led(object):
|
||||||
@ -10,7 +12,7 @@ class Led(object):
|
|||||||
STATUS_LED_COLOR_ORANGE_BLINK = 'orange_blink'
|
STATUS_LED_COLOR_ORANGE_BLINK = 'orange_blink'
|
||||||
STATUS_LED_COLOR_OFF = 'off'
|
STATUS_LED_COLOR_OFF = 'off'
|
||||||
|
|
||||||
LED_ON = '1'
|
LED_ON = '255'
|
||||||
LED_OFF = '0'
|
LED_OFF = '0'
|
||||||
LED_BLINK = '50'
|
LED_BLINK = '50'
|
||||||
|
|
||||||
@ -26,12 +28,11 @@ class Led(object):
|
|||||||
self._stop_blink(led_cap_list)
|
self._stop_blink(led_cap_list)
|
||||||
blink_pos = color.find('blink')
|
blink_pos = color.find('blink')
|
||||||
if blink_pos != -1:
|
if blink_pos != -1:
|
||||||
return self._set_status_blink(color, blink_pos, led_cap_list)
|
return self._set_status_blink(color, led_cap_list)
|
||||||
|
|
||||||
if color == Led.STATUS_LED_COLOR_GREEN:
|
if color == Led.STATUS_LED_COLOR_GREEN:
|
||||||
with open(self.get_green_led_path(), 'w') as led:
|
utils.write_file(self.get_green_led_path(), Led.LED_ON)
|
||||||
led.write(Led.LED_ON)
|
status = True
|
||||||
status = True
|
|
||||||
elif color == Led.STATUS_LED_COLOR_RED:
|
elif color == Led.STATUS_LED_COLOR_RED:
|
||||||
# Some led don't support red led but support orange led, in this case we set led to orange
|
# Some led don't support red led but support orange led, in this case we set led to orange
|
||||||
if Led.STATUS_LED_COLOR_RED in led_cap_list:
|
if Led.STATUS_LED_COLOR_RED in led_cap_list:
|
||||||
@ -41,19 +42,15 @@ class Led(object):
|
|||||||
else:
|
else:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
with open(led_path, 'w') as led:
|
utils.write_file(led_path, Led.LED_ON)
|
||||||
led.write(Led.LED_ON)
|
status = True
|
||||||
status = True
|
|
||||||
elif color == Led.STATUS_LED_COLOR_OFF:
|
elif color == Led.STATUS_LED_COLOR_OFF:
|
||||||
if Led.STATUS_LED_COLOR_GREEN in led_cap_list:
|
if Led.STATUS_LED_COLOR_GREEN in led_cap_list:
|
||||||
with open(self.get_green_led_path(), 'w') as led:
|
utils.write_file(self.get_green_led_path(), Led.LED_OFF)
|
||||||
led.write(Led.LED_OFF)
|
|
||||||
if Led.STATUS_LED_COLOR_RED in led_cap_list:
|
if Led.STATUS_LED_COLOR_RED in led_cap_list:
|
||||||
with open(self.get_red_led_path(), 'w') as led:
|
utils.write_file(self.get_red_led_path(), Led.LED_OFF)
|
||||||
led.write(Led.LED_OFF)
|
|
||||||
if Led.STATUS_LED_COLOR_ORANGE in led_cap_list:
|
if Led.STATUS_LED_COLOR_ORANGE in led_cap_list:
|
||||||
with open(self.get_orange_led_path(), 'w') as led:
|
utils.write_file(self.get_orange_led_path(), Led.LED_OFF)
|
||||||
led.write(Led.LED_OFF)
|
|
||||||
|
|
||||||
status = True
|
status = True
|
||||||
else:
|
else:
|
||||||
@ -63,7 +60,7 @@ class Led(object):
|
|||||||
|
|
||||||
return status
|
return status
|
||||||
|
|
||||||
def _set_status_blink(self, color, blink_pos, led_cap_list):
|
def _set_status_blink(self, color, led_cap_list):
|
||||||
if color not in led_cap_list:
|
if color not in led_cap_list:
|
||||||
if color == Led.STATUS_LED_COLOR_RED_BLINK and Led.STATUS_LED_COLOR_ORANGE_BLINK in led_cap_list:
|
if color == Led.STATUS_LED_COLOR_RED_BLINK and Led.STATUS_LED_COLOR_ORANGE_BLINK in led_cap_list:
|
||||||
color = Led.STATUS_LED_COLOR_ORANGE_BLINK
|
color = Led.STATUS_LED_COLOR_ORANGE_BLINK
|
||||||
@ -73,32 +70,60 @@ class Led(object):
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
if Led.STATUS_LED_COLOR_GREEN_BLINK == color:
|
if Led.STATUS_LED_COLOR_GREEN_BLINK == color:
|
||||||
self._set_led_blink_status(self.get_green_led_delay_on_path(), self.get_green_led_delay_off_path(), Led.LED_BLINK)
|
self._trigger_blink(self.get_green_led_trigger())
|
||||||
|
return self._set_led_blink_status(self.get_green_led_delay_on_path(), self.get_green_led_delay_off_path(), Led.LED_BLINK)
|
||||||
elif Led.STATUS_LED_COLOR_RED_BLINK == color:
|
elif Led.STATUS_LED_COLOR_RED_BLINK == color:
|
||||||
self._set_led_blink_status(self.get_red_led_delay_on_path(), self.get_red_led_delay_off_path(), Led.LED_BLINK)
|
self._trigger_blink(self.get_red_led_trigger())
|
||||||
|
return self._set_led_blink_status(self.get_red_led_delay_on_path(), self.get_red_led_delay_off_path(), Led.LED_BLINK)
|
||||||
elif Led.STATUS_LED_COLOR_ORANGE_BLINK == color:
|
elif Led.STATUS_LED_COLOR_ORANGE_BLINK == color:
|
||||||
self._set_led_blink_status(self.get_orange_led_delay_on_path(), self.get_orange_led_delay_off_path(), Led.LED_BLINK)
|
self._trigger_blink(self.get_orange_led_trigger())
|
||||||
|
return self._set_led_blink_status(self.get_orange_led_delay_on_path(), self.get_orange_led_delay_off_path(), Led.LED_BLINK)
|
||||||
else:
|
else:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
return True
|
|
||||||
|
|
||||||
def _stop_blink(self, led_cap_list):
|
def _stop_blink(self, led_cap_list):
|
||||||
try:
|
try:
|
||||||
if Led.STATUS_LED_COLOR_GREEN_BLINK in led_cap_list:
|
if Led.STATUS_LED_COLOR_GREEN_BLINK in led_cap_list:
|
||||||
self._set_led_blink_status(self.get_green_led_delay_on_path(), self.get_green_led_delay_off_path(), Led.LED_OFF)
|
self._untrigger_blink(self.get_green_led_trigger())
|
||||||
if Led.STATUS_LED_COLOR_RED_BLINK in led_cap_list:
|
if Led.STATUS_LED_COLOR_RED_BLINK in led_cap_list:
|
||||||
self._set_led_blink_status(self.get_red_led_delay_on_path(), self.get_red_led_delay_off_path(), Led.LED_OFF)
|
self._untrigger_blink(self.get_red_led_trigger())
|
||||||
if Led.STATUS_LED_COLOR_ORANGE_BLINK in led_cap_list:
|
if Led.STATUS_LED_COLOR_ORANGE_BLINK in led_cap_list:
|
||||||
self._set_led_blink_status(self.get_orange_led_delay_on_path(), self.get_orange_led_delay_off_path(), Led.LED_OFF)
|
self._untrigger_blink(self.get_orange_led_trigger())
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
def _trigger_blink(self, blink_trigger_file):
|
||||||
|
utils.write_file(blink_trigger_file, 'timer')
|
||||||
|
|
||||||
|
def _untrigger_blink(self, blink_trigger_file):
|
||||||
|
utils.write_file(blink_trigger_file, 'none')
|
||||||
|
|
||||||
def _set_led_blink_status(self, delay_on_file, delay_off_file, value):
|
def _set_led_blink_status(self, delay_on_file, delay_off_file, value):
|
||||||
with open(delay_on_file, 'w') as led:
|
if not self._wait_files_ready((delay_on_file, delay_off_file)):
|
||||||
led.write(value)
|
return False
|
||||||
with open(delay_off_file, 'w') as led:
|
|
||||||
led.write(value)
|
utils.write_file(delay_on_file, value)
|
||||||
|
utils.write_file(delay_off_file, value)
|
||||||
|
return True
|
||||||
|
|
||||||
|
def _wait_files_ready(self, file_list):
|
||||||
|
"""delay_off and delay_on sysfs will be available only if _trigger_blink is called. And once
|
||||||
|
_trigger_blink is called, driver might need time to prepare delay_off and delay_on. So,
|
||||||
|
need wait a few seconds here to make sure the sysfs is ready
|
||||||
|
|
||||||
|
Args:
|
||||||
|
file_list (list of str): files to be checked
|
||||||
|
"""
|
||||||
|
wait_time = 5.0
|
||||||
|
initial_sleep = 0.01
|
||||||
|
while wait_time > 0:
|
||||||
|
if all([os.path.exists(x) for x in file_list]):
|
||||||
|
return True
|
||||||
|
time.sleep(initial_sleep)
|
||||||
|
wait_time -= initial_sleep
|
||||||
|
initial_sleep = initial_sleep * 2
|
||||||
|
|
||||||
|
return False
|
||||||
|
|
||||||
def get_status(self):
|
def get_status(self):
|
||||||
led_cap_list = self.get_capability()
|
led_cap_list = self.get_capability()
|
||||||
@ -110,18 +135,15 @@ class Led(object):
|
|||||||
if blink_status is not None:
|
if blink_status is not None:
|
||||||
return blink_status
|
return blink_status
|
||||||
|
|
||||||
with open(self.get_green_led_path(), 'r') as led:
|
if utils.read_str_from_file(self.get_green_led_path()) != Led.LED_OFF:
|
||||||
if Led.LED_OFF != led.read().rstrip('\n'):
|
return Led.STATUS_LED_COLOR_GREEN
|
||||||
return Led.STATUS_LED_COLOR_GREEN
|
|
||||||
|
|
||||||
if Led.STATUS_LED_COLOR_RED in led_cap_list:
|
if Led.STATUS_LED_COLOR_RED in led_cap_list:
|
||||||
with open(self.get_red_led_path(), 'r') as led:
|
if utils.read_str_from_file(self.get_red_led_path()) != Led.LED_OFF:
|
||||||
if Led.LED_OFF != led.read().rstrip('\n'):
|
return Led.STATUS_LED_COLOR_RED
|
||||||
return Led.STATUS_LED_COLOR_RED
|
|
||||||
if Led.STATUS_LED_COLOR_ORANGE in led_cap_list:
|
if Led.STATUS_LED_COLOR_ORANGE in led_cap_list:
|
||||||
with open(self.get_orange_led_path(), 'r') as led:
|
if utils.read_str_from_file(self.get_orange_led_path()) != Led.LED_OFF:
|
||||||
if Led.LED_OFF != led.read().rstrip('\n'):
|
return Led.STATUS_LED_COLOR_RED
|
||||||
return Led.STATUS_LED_COLOR_RED
|
|
||||||
except (ValueError, IOError) as e:
|
except (ValueError, IOError) as e:
|
||||||
raise RuntimeError("Failed to read led status due to {}".format(repr(e)))
|
raise RuntimeError("Failed to read led status due to {}".format(repr(e)))
|
||||||
|
|
||||||
@ -132,6 +154,7 @@ class Led(object):
|
|||||||
if Led.STATUS_LED_COLOR_GREEN_BLINK in led_cap_list:
|
if Led.STATUS_LED_COLOR_GREEN_BLINK in led_cap_list:
|
||||||
if self._is_led_blinking(self.get_green_led_delay_on_path(), self.get_green_led_delay_off_path()):
|
if self._is_led_blinking(self.get_green_led_delay_on_path(), self.get_green_led_delay_off_path()):
|
||||||
return Led.STATUS_LED_COLOR_GREEN_BLINK
|
return Led.STATUS_LED_COLOR_GREEN_BLINK
|
||||||
|
|
||||||
if Led.STATUS_LED_COLOR_RED_BLINK in led_cap_list:
|
if Led.STATUS_LED_COLOR_RED_BLINK in led_cap_list:
|
||||||
if self._is_led_blinking(self.get_red_led_delay_on_path(), self.get_red_led_delay_off_path()):
|
if self._is_led_blinking(self.get_red_led_delay_on_path(), self.get_red_led_delay_off_path()):
|
||||||
return Led.STATUS_LED_COLOR_RED_BLINK
|
return Led.STATUS_LED_COLOR_RED_BLINK
|
||||||
@ -144,126 +167,73 @@ class Led(object):
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
def _is_led_blinking(self, delay_on_file, delay_off_file):
|
def _is_led_blinking(self, delay_on_file, delay_off_file):
|
||||||
with open(delay_on_file, 'r') as led:
|
delay_on = utils.read_str_from_file(delay_on_file, default=Led.LED_OFF)
|
||||||
delay_on = led.read().rstrip('\n')
|
delay_off = utils.read_str_from_file(delay_off_file, default=Led.LED_OFF)
|
||||||
with open(delay_off_file, 'r') as led:
|
|
||||||
delay_off = led.read().rstrip('\n')
|
|
||||||
return delay_on != Led.LED_OFF and delay_off != Led.LED_OFF
|
return delay_on != Led.LED_OFF and delay_off != Led.LED_OFF
|
||||||
|
|
||||||
def get_capability(self):
|
def get_capability(self):
|
||||||
cap_list = None
|
caps = utils.read_str_from_file(self.get_led_cap_path())
|
||||||
try:
|
return set(caps.split())
|
||||||
with open(self.get_led_cap_path(), 'r') as led_cap:
|
|
||||||
caps = led_cap.read()
|
|
||||||
cap_list = set(caps.split())
|
|
||||||
except (ValueError, IOError):
|
|
||||||
pass
|
|
||||||
|
|
||||||
return cap_list
|
|
||||||
|
|
||||||
def get_green_led_path(self):
|
def get_green_led_path(self):
|
||||||
pass
|
return os.path.join(Led.LED_PATH, 'led_{}_green'.format(self._led_id))
|
||||||
|
|
||||||
def get_green_led_delay_off_path(self):
|
def get_green_led_delay_off_path(self):
|
||||||
return '{}_delay_off'.format(self.get_green_led_path())
|
return os.path.join(Led.LED_PATH, 'led_{}_green_delay_off'.format(self._led_id))
|
||||||
|
|
||||||
def get_green_led_delay_on_path(self):
|
def get_green_led_delay_on_path(self):
|
||||||
return '{}_delay_on'.format(self.get_green_led_path())
|
return os.path.join(Led.LED_PATH, 'led_{}_green_delay_on'.format(self._led_id))
|
||||||
|
|
||||||
|
def get_green_led_trigger(self):
|
||||||
|
return os.path.join(Led.LED_PATH, 'led_{}_green_trigger'.format(self._led_id))
|
||||||
|
|
||||||
def get_red_led_path(self):
|
def get_red_led_path(self):
|
||||||
pass
|
return os.path.join(Led.LED_PATH, 'led_{}_red'.format(self._led_id))
|
||||||
|
|
||||||
def get_red_led_delay_off_path(self):
|
def get_red_led_delay_off_path(self):
|
||||||
return '{}_delay_off'.format(self.get_red_led_path())
|
return os.path.join(Led.LED_PATH, 'led_{}_red_delay_off'.format(self._led_id))
|
||||||
|
|
||||||
def get_red_led_delay_on_path(self):
|
def get_red_led_delay_on_path(self):
|
||||||
return '{}_delay_on'.format(self.get_red_led_path())
|
return os.path.join(Led.LED_PATH, 'led_{}_red_delay_on'.format(self._led_id))
|
||||||
|
|
||||||
|
def get_red_led_trigger(self):
|
||||||
|
return os.path.join(Led.LED_PATH, 'led_{}_red_trigger'.format(self._led_id))
|
||||||
|
|
||||||
def get_orange_led_path(self):
|
def get_orange_led_path(self):
|
||||||
pass
|
return os.path.join(Led.LED_PATH, 'led_{}_orange'.format(self._led_id))
|
||||||
|
|
||||||
def get_orange_led_delay_off_path(self):
|
def get_orange_led_delay_off_path(self):
|
||||||
return '{}_delay_off'.format(self.get_orange_led_path())
|
return os.path.join(Led.LED_PATH, 'led_{}_orange_delay_off'.format(self._led_id))
|
||||||
|
|
||||||
def get_orange_led_delay_on_path(self):
|
def get_orange_led_delay_on_path(self):
|
||||||
return '{}_delay_on'.format(self.get_orange_led_path())
|
return os.path.join(Led.LED_PATH, 'led_{}_orange_delay_on'.format(self._led_id))
|
||||||
|
|
||||||
|
def get_orange_led_trigger(self):
|
||||||
|
return os.path.join(Led.LED_PATH, 'led_{}_orange_trigger'.format(self._led_id))
|
||||||
|
|
||||||
def get_led_cap_path(self):
|
def get_led_cap_path(self):
|
||||||
pass
|
return os.path.join(Led.LED_PATH, 'led_{}_capability'.format(self._led_id))
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class FanLed(Led):
|
class FanLed(Led):
|
||||||
LED_PATH = "/var/run/hw-management/led/"
|
|
||||||
|
|
||||||
def __init__(self, index):
|
def __init__(self, index):
|
||||||
if index is not None:
|
if index is not None:
|
||||||
self._green_led_path = os.path.join(Led.LED_PATH, "led_fan{}_green".format(index))
|
self._led_id = 'fan{}'.format(index)
|
||||||
self._red_led_path = os.path.join(Led.LED_PATH, "led_fan{}_red".format(index))
|
|
||||||
self._orange_led_path = os.path.join(Led.LED_PATH, "led_fan{}_orange".format(index))
|
|
||||||
self._led_cap_path = os.path.join(Led.LED_PATH, "led_fan{}_capability".format(index))
|
|
||||||
else:
|
else:
|
||||||
self._green_led_path = os.path.join(Led.LED_PATH, "led_fan_green")
|
self._led_id = 'fan'
|
||||||
self._red_led_path = os.path.join(Led.LED_PATH, "led_fan_red")
|
|
||||||
self._orange_led_path = os.path.join(Led.LED_PATH, "led_fan_orange")
|
|
||||||
self._led_cap_path = os.path.join(Led.LED_PATH, "led_fan_capability")
|
|
||||||
|
|
||||||
def get_green_led_path(self):
|
|
||||||
return self._green_led_path
|
|
||||||
|
|
||||||
def get_red_led_path(self):
|
|
||||||
return self._red_led_path
|
|
||||||
|
|
||||||
def get_orange_led_path(self):
|
|
||||||
return self._orange_led_path
|
|
||||||
|
|
||||||
def get_led_cap_path(self):
|
|
||||||
return self._led_cap_path
|
|
||||||
|
|
||||||
|
|
||||||
class PsuLed(Led):
|
class PsuLed(Led):
|
||||||
def __init__(self, index):
|
def __init__(self, index):
|
||||||
if index is not None:
|
if index is not None:
|
||||||
self._green_led_path = os.path.join(Led.LED_PATH, "led_psu{}_green".format(index))
|
self._led_id = 'psu{}'.format(index)
|
||||||
self._red_led_path = os.path.join(Led.LED_PATH, "led_psu{}_red".format(index))
|
|
||||||
self._orange_led_path = os.path.join(Led.LED_PATH, "led_psu{}_orange".format(index))
|
|
||||||
self._led_cap_path = os.path.join(Led.LED_PATH, "led_psu{}_capability".format(index))
|
|
||||||
else:
|
else:
|
||||||
self._green_led_path = os.path.join(Led.LED_PATH, "led_psu_green")
|
self._led_id = 'psu'
|
||||||
self._red_led_path = os.path.join(Led.LED_PATH, "led_psu_red")
|
|
||||||
self._orange_led_path = os.path.join(Led.LED_PATH, "led_psu_orange")
|
|
||||||
self._led_cap_path = os.path.join(Led.LED_PATH, "led_psu_capability")
|
|
||||||
|
|
||||||
def get_green_led_path(self):
|
|
||||||
return self._green_led_path
|
|
||||||
|
|
||||||
def get_red_led_path(self):
|
|
||||||
return self._red_led_path
|
|
||||||
|
|
||||||
def get_orange_led_path(self):
|
|
||||||
return self._orange_led_path
|
|
||||||
|
|
||||||
def get_led_cap_path(self):
|
|
||||||
return self._led_cap_path
|
|
||||||
|
|
||||||
|
|
||||||
class SystemLed(Led):
|
class SystemLed(Led):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self._green_led_path = os.path.join(Led.LED_PATH, "led_status_green")
|
self._led_id = 'status'
|
||||||
self._red_led_path = os.path.join(Led.LED_PATH, "led_status_red")
|
|
||||||
self._orange_led_path = os.path.join(Led.LED_PATH, "led_status_orange")
|
|
||||||
self._led_cap_path = os.path.join(Led.LED_PATH, "led_status_capability")
|
|
||||||
|
|
||||||
def get_green_led_path(self):
|
|
||||||
return self._green_led_path
|
|
||||||
|
|
||||||
def get_red_led_path(self):
|
|
||||||
return self._red_led_path
|
|
||||||
|
|
||||||
def get_orange_led_path(self):
|
|
||||||
return self._orange_led_path
|
|
||||||
|
|
||||||
def get_led_cap_path(self):
|
|
||||||
return self._led_cap_path
|
|
||||||
|
|
||||||
|
|
||||||
class SharedLed(object):
|
class SharedLed(object):
|
||||||
|
Reference in New Issue
Block a user