Maisam Arif
114 строки
4.6 KiB
Python
Исполняемый файл
114 строки
4.6 KiB
Python
Исполняемый файл
#!/usr/bin/env python3
|
|
# PYTHON_ARGCOMPLETE_OK
|
|
|
|
#
|
|
# Copyright (C) 2023 Advanced Micro Devices. All rights reserved.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
# this software and associated documentation files (the "Software"), to deal in
|
|
# the Software without restriction, including without limitation the rights to
|
|
# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
|
# the Software, and to permit persons to whom the Software is furnished to do so,
|
|
# subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in all
|
|
# copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
|
# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
|
# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
#
|
|
|
|
import logging
|
|
import sys
|
|
|
|
try:
|
|
import argcomplete
|
|
except ImportError:
|
|
logging.debug("argcomplete module not found. Autocomplete will not work.")
|
|
|
|
from amdsmi_commands import AMDSMICommands
|
|
from amdsmi_parser import AMDSMIParser
|
|
from amdsmi_logger import AMDSMILogger
|
|
import amdsmi_cli_exceptions
|
|
from amdsmi import amdsmi_interface
|
|
from amdsmi import amdsmi_exception
|
|
|
|
|
|
def _print_error(e, destination):
|
|
if destination in ['stdout', 'json', 'csv']:
|
|
print(e)
|
|
else:
|
|
f = open(destination, "w")
|
|
f.write(e)
|
|
f.close()
|
|
print("Error occured. Result written to " + str(destination) + " file")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Disable traceback before possible init errors in AMDSMICommands and AMDSMIParser
|
|
if "DEBUG" in sys.argv:
|
|
sys.tracebacklimit = 10
|
|
else:
|
|
sys.tracebacklimit = -1
|
|
|
|
amd_smi_commands = AMDSMICommands()
|
|
amd_smi_parser = AMDSMIParser(amd_smi_commands.version,
|
|
amd_smi_commands.list,
|
|
amd_smi_commands.static,
|
|
amd_smi_commands.firmware,
|
|
amd_smi_commands.bad_pages,
|
|
amd_smi_commands.metric,
|
|
amd_smi_commands.process,
|
|
amd_smi_commands.profile,
|
|
amd_smi_commands.event,
|
|
amd_smi_commands.topology,
|
|
amd_smi_commands.set_value,
|
|
amd_smi_commands.reset,
|
|
amd_smi_commands.monitor,
|
|
amd_smi_commands.rocm_smi)
|
|
try:
|
|
try:
|
|
argcomplete.autocomplete(amd_smi_parser)
|
|
except NameError:
|
|
logging.debug("argcomplete module not found. Autocomplete will not work.")
|
|
|
|
args = amd_smi_parser.parse_args(args=None if sys.argv[1:] else ['--help'])
|
|
|
|
# Handle command modifiers before subcommand execution
|
|
if args.json:
|
|
amd_smi_commands.logger.format = amd_smi_commands.logger.LoggerFormat.json.value
|
|
if args.csv:
|
|
amd_smi_commands.logger.format = amd_smi_commands.logger.LoggerFormat.csv.value
|
|
if args.file:
|
|
amd_smi_commands.logger.destination = args.file
|
|
|
|
# Remove previous log handlers
|
|
for handler in logging.root.handlers[:]:
|
|
logging.root.removeHandler(handler)
|
|
|
|
logging_dict = {'DEBUG' : logging.DEBUG,
|
|
'INFO' : logging.INFO,
|
|
'WARNING': logging.WARNING,
|
|
'ERROR': logging.ERROR,
|
|
'CRITICAL': logging.CRITICAL}
|
|
# To enable debug logs on rocm-smi library set RSMI_LOGGING = 1 in environment
|
|
logging.basicConfig(format='%(levelname)s: %(message)s', level=logging_dict[args.loglevel])
|
|
|
|
# Disable traceback for non-debug log levels
|
|
if args.loglevel == "DEBUG":
|
|
sys.tracebacklimit = 10
|
|
else:
|
|
sys.tracebacklimit = -1
|
|
|
|
# Execute subcommands
|
|
args.func(args)
|
|
except amdsmi_cli_exceptions.AmdSmiException as e:
|
|
_print_error(str(e), amd_smi_commands.logger.destination)
|
|
except amdsmi_exception.AmdSmiLibraryException as e:
|
|
exc = amdsmi_cli_exceptions.AmdSmiAMDSMIErrorException(amd_smi_commands.logger.format, e.get_error_code())
|
|
_print_error(str(exc), amd_smi_commands.logger.destination)
|