util_base.py

#!/usr/bin/env python2

try:
    import imp
    import subprocess
    import os
    import syslog
except ImportError as e:
    raise ImportError (str(e) + " - required module not found")

#
# Constants ====================================================================
#
# Platform root directory inside docker
PLATFORM_ROOT_DOCKER = '/usr/share/sonic/platform'
SONIC_CFGGEN_PATH = '/usr/local/bin/sonic-cfggen'
HWSKU_KEY = 'DEVICE_METADATA.localhost.hwsku'
PLATFORM_KEY = 'DEVICE_METADATA.localhost.platform'
PDDF_FILE_PATH = '/usr/share/sonic/platform/pddf_support'

# Port config information
PORT_CONFIG = 'port_config.ini'
PORTMAP = 'portmap.ini'


EEPROM_MODULE_NAME = 'eeprom'
EEPROM_CLASS_NAME = 'board'

class UtilLogger(object):
    def __init__(self, syslog_identifier):
        self.syslog = syslog
        self.syslog.openlog(ident=syslog_identifier, logoption=self.syslog.LOG_NDELAY, facility=self.syslog.LOG_DAEMON)

    def __del__(self):
        self.syslog.closelog()

    def log_error(self, msg, print_to_console=False):
        self.syslog.syslog(self.syslog.LOG_ERR, msg)

        if print_to_console:
            print msg

    def log_warning(self, msg, print_to_console=False):
        self.syslog.syslog(self.syslog.LOG_WARNING, msg)

        if print_to_console:
            print msg

    def log_notice(self, msg, print_to_console=False):
        self.syslog.syslog(self.syslog.LOG_NOTICE, msg)

        if print_to_console:
            print msg

    def log_info(self, msg, print_to_console=False):
        self.syslog.syslog(self.syslog.LOG_INFO, msg)

        if print_to_console:
            print msg

    def log_debug(self, msg, print_to_console=False):
        self.syslog.syslog(self.syslog.LOG_DEBUG, msg)

        if print_to_console:
            print msg


class UtilHelper(object):
    def __init__(self):
        pass

    # Returns platform and hwsku
    def get_platform_and_hwsku(self):
        try:
            proc = subprocess.Popen([SONIC_CFGGEN_PATH, '-H', '-v', PLATFORM_KEY],
                                    stdout=subprocess.PIPE,
                                    shell=False,
                                    stderr=subprocess.STDOUT)
            stdout = proc.communicate()[0]
            proc.wait()
            platform = stdout.rstrip('\n')

            proc = subprocess.Popen([SONIC_CFGGEN_PATH, '-d', '-v', HWSKU_KEY],
                                    stdout=subprocess.PIPE,
                                    shell=False,
                                    stderr=subprocess.STDOUT)
            stdout = proc.communicate()[0]
            proc.wait()
            hwsku = stdout.rstrip('\n')
        except OSError as e:
            raise OSError("Failed to detect platform: %s" % (str(e)))

        return (platform, hwsku)

    # Returns path to platform and hwsku
    def get_path_to_platform_and_hwsku(self):
        # Get platform and hwsku
        (platform, hwsku) = self.get_platform_and_hwsku()

        # Load platform module from source
        platform_path = PLATFORM_ROOT_DOCKER
        hwsku_path = "/".join([platform_path, hwsku])

        return (platform_path, hwsku_path)

    # Returns path to port config file
    def get_path_to_port_config_file(self):
        # Get platform and hwsku path
        (platform_path, hwsku_path) = self.get_path_to_platform_and_hwsku()

        # First check for the presence of the new 'port_config.ini' file
        port_config_file_path = "/".join([hwsku_path, PORT_CONFIG])
        if not os.path.isfile(port_config_file_path):
            # port_config.ini doesn't exist. Try loading the legacy 'portmap.ini' file
            port_config_file_path = "/".join([hwsku_path, PORTMAP])
            if not os.path.isfile(port_config_file_path):
                raise IOError("Failed to detect port config file: %s" % (port_config_file_path))

        return port_config_file_path

    # Loads platform specific psuutil module from source
    def load_platform_util(self, module_name, class_name):
        platform_util = None

        # Get path to platform and hwsku
        (platform_path, hwsku_path) = self.get_path_to_platform_and_hwsku()

        try:
            module_file = "/".join([platform_path, "plugins", module_name + ".py"])
            module = imp.load_source(module_name, module_file)
        except IOError as e:
            raise IOError("Failed to load platform module '%s': %s" % (module_name, str(e)))

        try:
            platform_util_class = getattr(module, class_name)
            # board class of eeprom requires 4 paramerters, need special treatment here.
            if module_name == EEPROM_MODULE_NAME and class_name == EEPROM_CLASS_NAME:
                platform_util = platform_util_class('','','','')
            else:
                platform_util = platform_util_class()
        except AttributeError as e:
            raise AttributeError("Failed to instantiate '%s' class: %s" % (class_name, str(e)))

        return platform_util

    def check_pddf_mode(self):
        if os.path.exists(PDDF_FILE_PATH):
            return True
        else:
            return False