ae2cfd52e8
1. In the raise statement of the check_positive() function, the error message conflicts with the if condition. 2. If the value of val is 0, it passes the if condition but raises an error message stating [0 has to be >= 0]. 3. This PR removes the = from the raise statement, so when val is 0, it will correctly prompt that the value of val [has to be > 0]. Signed-off-by: dufuhang <dufuhang@kylinos.cn>
142 lines
6.7 KiB
Python
Executable file
142 lines
6.7 KiB
Python
Executable file
#!/usr/bin/python3 -Es
|
|
#
|
|
# tuned: daemon for monitoring and adaptive tuning of system devices
|
|
#
|
|
# Copyright (C) 2008-2013 Red Hat, Inc.
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
|
|
from __future__ import print_function
|
|
import argparse
|
|
import sys
|
|
import traceback
|
|
import tuned.admin
|
|
import tuned.consts as consts
|
|
import tuned.version as ver
|
|
from tuned.utils.global_config import GlobalConfig
|
|
|
|
def check_positive(value):
|
|
try:
|
|
val = int(value)
|
|
except ValueError:
|
|
val = -1
|
|
if val <= 0:
|
|
raise argparse.ArgumentTypeError("%s has to be > 0" % value)
|
|
return val
|
|
|
|
def check_log_level(value):
|
|
try:
|
|
return consts.CAPTURE_LOG_LEVELS[value.lower()]
|
|
except KeyError:
|
|
levels = ", ".join(consts.CAPTURE_LOG_LEVELS.keys())
|
|
raise argparse.ArgumentTypeError(
|
|
"Invalid log level: %s. Valid log levels: %s."
|
|
% (value, levels))
|
|
|
|
if __name__ == "__main__":
|
|
config = GlobalConfig()
|
|
parser = argparse.ArgumentParser(description="Manage tuned daemon.")
|
|
parser.add_argument('--version', "-v", action = "version", version = "%%(prog)s %s.%s.%s" % (ver.TUNED_VERSION_MAJOR, ver.TUNED_VERSION_MINOR, ver.TUNED_VERSION_PATCH))
|
|
parser.add_argument("--debug", "-d", action="store_true", help="show debug messages")
|
|
parser.add_argument("--async", "-a", action="store_true", help="with dbus do not wait on commands completion and return immediately")
|
|
parser.add_argument("--timeout", "-t", default = consts.ADMIN_TIMEOUT, type = check_positive, help="with sync operation use specific timeout instead of the default %d second(s)" % consts.ADMIN_TIMEOUT)
|
|
levels = ", ".join(consts.CAPTURE_LOG_LEVELS.keys())
|
|
help = "level of log messages to capture (one of %s). Default: %s" \
|
|
% (levels, consts.CAPTURE_LOG_LEVEL)
|
|
parser.add_argument("--loglevel", "-l",
|
|
default = consts.CAPTURE_LOG_LEVEL,
|
|
type = check_log_level,
|
|
help = help)
|
|
|
|
subparsers = parser.add_subparsers()
|
|
|
|
parser_list = subparsers.add_parser("list", help="list available profiles or plugins (by default profiles)")
|
|
parser_list.set_defaults(action="list")
|
|
|
|
parser_list.add_argument("list_choice", nargs="?",default="profiles", choices=["plugins","profiles"], help="choose what to list", metavar="{plugins|profiles}")
|
|
parser_list.add_argument("--verbose", "-v", action="store_true", help="show plugin's configuration parameters and their meaning")
|
|
|
|
parser_active = subparsers.add_parser("active", help="show active profile")
|
|
parser_active.set_defaults(action="active")
|
|
|
|
parser_off = subparsers.add_parser("off", help="switch off all tunings")
|
|
parser_off.set_defaults(action="off")
|
|
|
|
parser_profile = subparsers.add_parser("profile", help="switch to a given profile, or list available profiles if no profile is given")
|
|
parser_profile.set_defaults(action="profile")
|
|
parser_profile.add_argument("profiles", metavar="profile", type=str, nargs="*", help="profile name")
|
|
|
|
parser_profile_info = subparsers.add_parser("profile_info", help="show information/description of given profile or current profile if no profile is specified")
|
|
parser_profile_info.set_defaults(action="profile_info")
|
|
parser_profile_info.add_argument("profile", metavar="profile", type=str, nargs="?", default="", help="profile name, current profile if not specified")
|
|
|
|
if config.get(consts.CFG_RECOMMEND_COMMAND, consts.CFG_DEF_RECOMMEND_COMMAND):
|
|
parser_off = subparsers.add_parser("recommend", help="recommend profile")
|
|
parser_off.set_defaults(action="recommend_profile")
|
|
|
|
parser_verify = subparsers.add_parser("verify", help="verify profile")
|
|
parser_verify.set_defaults(action="verify_profile")
|
|
parser_verify.add_argument("--ignore-missing", "-i", action="store_true", help="do not treat missing/non-supported tunings as errors")
|
|
|
|
parser_auto_profile = subparsers.add_parser("auto_profile", help="enable automatic profile selection mode, switch to the recommended profile")
|
|
parser_auto_profile.set_defaults(action="auto_profile")
|
|
|
|
parser_profile_mode = subparsers.add_parser("profile_mode", help="show current profile selection mode")
|
|
parser_profile_mode.set_defaults(action="profile_mode")
|
|
|
|
parser_instance_acquire_devices = subparsers.add_parser("instance_acquire_devices", help="acquire devices from other instances and assign them to the given instance")
|
|
parser_instance_acquire_devices.set_defaults(action="instance_acquire_devices")
|
|
parser_instance_acquire_devices.add_argument("devices", metavar="devices", type=str, help="comma-separated list of device names; may use the cpulist syntax if prefixed with 'cpulist:'")
|
|
parser_instance_acquire_devices.add_argument("instance", metavar="instance", type=str, help="name of the plugin instance which should acquire the devices")
|
|
|
|
parser_get_instances = subparsers.add_parser("get_instances", help="list active instances of a given plugin or all active instances if no plugin is specified")
|
|
parser_get_instances.set_defaults(action="get_instances")
|
|
parser_get_instances.add_argument("plugin_name", metavar="plugin_name", type=str, nargs="?", default="", help="name of the plugin to restrict the list of instances to")
|
|
|
|
parser_instance_get_devices = subparsers.add_parser("instance_get_devices", help="list devices assigned to a given instance")
|
|
parser_instance_get_devices.set_defaults(action="instance_get_devices")
|
|
parser_instance_get_devices.add_argument("instance", metavar="instance", type=str, help="name of the plugin instance")
|
|
|
|
args = parser.parse_args(sys.argv[1:])
|
|
|
|
options = vars(args)
|
|
debug = options.pop("debug")
|
|
asynco = options.pop("async")
|
|
timeout = options.pop("timeout")
|
|
try:
|
|
action_name = options.pop("action")
|
|
except KeyError:
|
|
parser.print_usage(file = sys.stderr)
|
|
sys.exit(1)
|
|
log_level = options.pop("loglevel")
|
|
result = False
|
|
|
|
profile_dirs = config.get_list(consts.CFG_PROFILE_DIRS, consts.CFG_DEF_PROFILE_DIRS)
|
|
daemon = config.get_bool(consts.CFG_DAEMON, consts.CFG_DEF_DAEMON)
|
|
dbus = daemon and config.get_bool(consts.CFG_ENABLE_DBUS, consts.CFG_DEF_ENABLE_DBUS)
|
|
|
|
try:
|
|
admin = tuned.admin.Admin(profile_dirs, dbus, debug, asynco, timeout, log_level)
|
|
|
|
result = admin.action(action_name, **options)
|
|
except:
|
|
traceback.print_exc()
|
|
sys.exit(3)
|
|
|
|
if result == False:
|
|
sys.exit(1)
|
|
else:
|
|
sys.exit(0)
|