Platform Driver Development Framework (PDDF): Adding PDDF CLI utils (#…

…624)

This change is related to Platform Driver Development Framework (PDDF) which is being added to sonic-buildimage repo. More details can be found here, sonic-net/SONiC#406

PDDF supports its own CLI utilities, which use generic PDDF component plugins. I added these PDDF CLI utilities.

data/etc/bash_completion.d/pddf_fanutil

@@ -0,0 +1,8 @@
+_pddf_fanutil_completion() {
+ COMPREPLY=( $( env COMP_WORDS="${COMP_WORDS[*]}" \
+ COMP_CWORD=$COMP_CWORD \
+ _PDDF_FANUTIL_COMPLETE=complete $1 ) )
+ return 0
+}
+
+complete -F _pddf_fanutil_completion -o default pddf_fanutil;

data/etc/bash_completion.d/pddf_ledutil

@@ -0,0 +1,8 @@
+_pddf_ledutil_completion() {
+ COMPREPLY=( $( env COMP_WORDS="${COMP_WORDS[*]}" \
+ COMP_CWORD=$COMP_CWORD \
+ _PDDF_LEDUTIL_COMPLETE=complete $1 ) )
+ return 0
+}
+
+complete -F _pddf_ledutil_completion -o default pddf_ledutil;

data/etc/bash_completion.d/pddf_psuutil

@@ -0,0 +1,8 @@
+_pddf_psuutil_completion() {
+ COMPREPLY=( $( env COMP_WORDS="${COMP_WORDS[*]}" \
+ COMP_CWORD=$COMP_CWORD \
+ _PDDF_PSUUTIL_COMPLETE=complete $1 ) )
+ return 0
+}
+
+complete -F _pddf_psuutil_completion -o default pddf_psuutil;

data/etc/bash_completion.d/pddf_thermalutil

@@ -0,0 +1,8 @@
+_pddf_thermalutil_completion() {
+ COMPREPLY=( $( env COMP_WORDS="${COMP_WORDS[*]}" \
+ COMP_CWORD=$COMP_CWORD \
+ _PDDF_THERMALUTIL_COMPLETE=complete $1 ) )
+ return 0
+}
+
+complete -F _pddf_thermalutil_completion -o default pddf_thermalutil;

pddf_fanutil/main.py

@@ -0,0 +1,193 @@
+#!/usr/bin/env python
+#
+# main.py
+#
+# Command-line utility for interacting with FAN Controller in PDDF mode in SONiC
+#
+
+try:
+ import sys
+ import os
+ import subprocess
+ import click
+ import imp
+ import syslog
+ import types
+ import traceback
+ from tabulate import tabulate
+ from utilities_common import util_base
+ from utilities_common.util_base import UtilLogger
+ from utilities_common.util_base import UtilHelper
+except ImportError as e:
+ raise ImportError("%s - required module not found" % str(e))
+
+VERSION = '1.0'
+
+SYSLOG_IDENTIFIER = "fanutil"
+PLATFORM_SPECIFIC_MODULE_NAME = "fanutil"
+PLATFORM_SPECIFIC_CLASS_NAME = "FanUtil"
+
+# Global platform-specific fanutil class instance
+platform_fanutil = None
+
+#logger = UtilLogger(SYSLOG_IDENTIFIER)
+
+# This is our main entrypoint - the main 'fanutil' command
+@click.group()
+def cli():
+ """pddf_fanutil - Command line utility for providing FAN information"""
+
+ if os.geteuid() != 0:
+ click.echo("Root privileges are required for this operation")
+ sys.exit(1)
+
+ # Load the helper class
+ helper = UtilHelper()
+
+ if not helper.check_pddf_mode():
+ click.echo("PDDF mode should be supported and enabled for this platform for this operation")
+ sys.exit(1)
+
+ # Load platform-specific fanutil class
+ global platform_fanutil
+ try:
+ platform_fanutil = helper.load_platform_util(PLATFORM_SPECIFIC_MODULE_NAME, PLATFORM_SPECIFIC_CLASS_NAME)
+ except Exception as e:
+ click.echo("Failed to load {}: {}".format(PLATFORM_SPECIFIC_MODULE_NAME, str(e)))
+ sys.exit(2)
+
+# 'version' subcommand
+@cli.command()
+def version():
+ """Display version info"""
+ click.echo("PDDF fanutil version {0}".format(VERSION))
+
+# 'numfans' subcommand
+@cli.command()
+def numfans():
+ """Display number of FANs installed on device"""
+ click.echo(str(platform_fanutil.get_num_fans()))
+
+# 'status' subcommand
+@cli.command()
+@click.option('-i', '--index', default=-1, type=int, help="the index of FAN")
+def status(index):
+ """Display FAN status"""
+ supported_fan = range(1, platform_fanutil.get_num_fans() + 1)
+ fan_ids = []
+ if (index < 0):
+ fan_ids = supported_fan
+ else:
+ fan_ids = [index]
+
+ header = ['FAN', 'Status']
+ status_table = []
+
+ for fan in fan_ids:
+ msg = ""
+ fan_name = "FAN {}".format(fan)
+ if fan not in supported_fan:
+ click.echo("Error! The {} is not available on the platform.\n" \
+ "Number of supported FAN - {}.".format(fan_name, platform_fanutil.get_num_fans()))
+ continue
+ presence = platform_fanutil.get_presence(fan)
+ if presence:
+ oper_status = platform_fanutil.get_status(fan)
+ msg = 'OK' if oper_status else "NOT OK"
+ else:
+ msg = 'NOT PRESENT'
+ status_table.append([fan_name, msg])
+
+ if status_table:
+ click.echo(tabulate(status_table, header, tablefmt="simple"))
+
+# 'direction' subcommand
+@cli.command()
+@click.option('-i', '--index', default=-1, type=int, help="the index of FAN")
+def direction(index):
+ """Display FAN airflow direction"""
+ supported_fan = range(1, platform_fanutil.get_num_fans() + 1)
+ fan_ids = []
+ if (index < 0):
+ fan_ids = supported_fan
+ else:
+ fan_ids = [index]
+
+ header = ['FAN', 'Direction']
+ status_table = []
+
+ for fan in fan_ids:
+ msg = ""
+ fan_name = "FAN {}".format(fan)
+ if fan not in supported_fan:
+ click.echo("Error! The {} is not available on the platform.\n" \
+ "Number of supported FAN - {}.".format(fan_name, platform_fanutil.get_num_fans()))
+ continue
+ direction = platform_fanutil.get_direction(fan)
+ status_table.append([fan_name, direction])
+
+ if status_table:
+ click.echo(tabulate(status_table, header, tablefmt="simple"))
+
+# 'speed' subcommand
+@cli.command()
+@click.option('-i', '--index', default=-1, type=int, help="the index of FAN")
+def getspeed(index):
+ """Display FAN speed in RPM"""
+ supported_fan = range(1, platform_fanutil.get_num_fans() + 1)
+ fan_ids = []
+ if (index < 0):
+ fan_ids = supported_fan
+ else:
+ fan_ids = [index]
+
+ header = ['FAN', 'Front Fan RPM', 'Rear Fan RPM']
+ status_table = []
+
+ for fan in fan_ids:
+ msg = ""
+ fan_name = "FAN {}".format(fan)
+ if fan not in supported_fan:
+ click.echo("Error! The {} is not available on the platform.\n" \
+ "Number of supported FAN - {}.".format(fan_name, platform_fanutil.get_num_fans()))
+ continue
+ front = platform_fanutil.get_speed(fan)
+ rear = platform_fanutil.get_speed_rear(fan)
+ status_table.append([fan_name, front, rear])
+
+ if status_table:
+ click.echo(tabulate(status_table, header, tablefmt="simple"))
+
+# 'setspeed' subcommand
+@cli.command()
+@click.argument('speed', type=int)
+def setspeed(speed):
+ """Set FAN speed in percentage"""
+ if speed is None:
+ click.echo("speed value is required")
+ raise click.Abort()
+
+ status = platform_fanutil.set_speed(speed)
+ if status:
+ click.echo("Successful")
+ else:
+ click.echo("Failed")
+
+@cli.group()
+def debug():
+ """pddf_fanutil debug commands"""
+ pass
+
+@debug.command()
+def dump_sysfs():
+ """Dump all Fan related SysFS paths"""
+ status = platform_fanutil.dump_sysfs()
+
+ if status:
+ for i in status:
+ click.echo(i)
+
+
+
+if __name__ == '__main__':
+ cli()

pddf_ledutil/main.py

@@ -0,0 +1,96 @@
+#!/usr/bin/env python
+#
+# main.py
+#
+# Command-line utility for interacting with FAN Controller in PDDF mode in SONiC
+#
+
+try:
+ import sys
+ import os
+ import subprocess
+ import click
+ import imp
+ import syslog
+ import types
+ import traceback
+ from tabulate import tabulate
+ from utilities_common import util_base
+ from utilities_common.util_base import UtilLogger
+ from utilities_common.util_base import UtilHelper
+except ImportError as e:
+ raise ImportError("%s - required module not found" % str(e))
+
+VERSION = '1.0'
+
+SYSLOG_IDENTIFIER = "ledutil"
+PLATFORM_SPECIFIC_MODULE_NAME = "ledutil"
+PLATFORM_SPECIFIC_CLASS_NAME = "LedUtil"
+
+# Global platform-specific ledutil class instance
+platform_ledutil = None
+
+#logger = UtilLogger(SYSLOG_IDENTIFIER)
+
+# ==================== CLI commands and groups ====================
+
+
+# This is our main entrypoint - the main 'ledutil' command
+@click.group()
+def cli():
+ """pddf_ledutil - Command line utility for providing FAN information"""
+
+ if os.geteuid() != 0:
+ click.echo("Root privileges are required for this operation")
+ sys.exit(1)
+
+ # Load the helper class
+ helper = UtilHelper()
+
+ if not helper.check_pddf_mode():
+ click.echo("PDDF mode should be supported and enabled for this platform for this operation")
+ sys.exit(1)
+
+ # Load platform-specific fanutil class
+ global platform_ledutil
+ try:
+ platform_ledutil = helper.load_platform_util(PLATFORM_SPECIFIC_MODULE_NAME, PLATFORM_SPECIFIC_CLASS_NAME)
+ except Exception as e:
+ click.echo("Failed to load {}: {}".format(PLATFORM_SPECIFIC_MODULE_NAME, str(e)))
+ sys.exit(2)
+
+# 'version' subcommand
+@cli.command()
+def version():
+ """Display version info"""
+ click.echo("PDDF ledutil version {0}".format(VERSION))
+
+# 'getstatusled' subcommand
+@cli.command()
+@click.argument('device_name', type=click.STRING)
+@click.argument('index', type=click.STRING)
+def getstatusled(device_name, index):
+ if device_name is None:
+ click.echo("device_name is required")
+ raise click.Abort()
+
+ outputs = platform_ledutil.get_status_led(device_name, index)
+ click.echo(outputs)
+
+
+# 'setstatusled' subcommand
+@cli.command()
+@click.argument('device_name', type=click.STRING)
+@click.argument('index', type=click.STRING)
+@click.argument('color', type=click.STRING)
+@click.argument('color_state', type=click.STRING)
+def setstatusled(device_name, index, color, color_state):
+ if device_name is None:
+ click.echo("device_name is required")
+ raise click.Abort()
+
+ outputs = platform_ledutil.set_status_led(device_name, index, color, color_state)
+ click.echo(outputs)
+
+if __name__ == '__main__':
+ cli()