File manager

File manager - Edit - /usr/lib/python3.6/site-packages/syspurpose/cli.py

Back
# -- coding: utf-8 -- from __future__ import print_function, division, absolute_import # # Copyright (c) 2018 Red Hat, Inc. # # This software is licensed to you under the GNU General Public License, # version 2 (GPLv2). There is NO WARRANTY for this software, express or # implied, including the implied warranties of MERCHANTABILITY or FITNESS # FOR A PARTICULAR PURPOSE. You should have received a copy of GPLv2 # along with this software; if not, see # http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt. # # Red Hat trademarks are not licensed under GPLv2. No permission is # granted to use or replicate Red Hat trademarks that are incorporated # in this software or its documentation. import argparse import os import sys import six from syspurpose.files import SyncedStore from syspurpose.utils import in_container, make_utf8 from syspurpose.i18n import ugettext as _ import json SP_CONFLICT_MESSAGE = _("Warning: A {attr} of \"{download_value}\" was recently set for this system " "by the entitlement server administrator.\n{advice}") SP_ADVICE = _("If you'd like to overwrite the server side change please run: {command}") def add_command(args, syspurposestore): """ Uses the syspurposestore to add one or more values to a particular property. :param args: The parsed args from argparse, expected attributes are: prop_name: the string name of the property to add to values: A list of the values to add to the given property (could be anything json-serializable) :param syspurposestore: An SyspurposeStore object to manipulate :return: None """ any_prop_added = False for value in args.values: if syspurposestore.add(args.prop_name, value): any_prop_added = True print(_("Added \"{value}\" to {prop_name}.").format( value=make_utf8(value), prop_name=make_utf8(args.prop_name))) else: print(_("Not adding value \"{value}\" to {prop_name}; it already exists.").format( value=make_utf8(value), prop_name=make_utf8(args.prop_name))) if any_prop_added is False: return success_msg = _("{attr} updated.").format(attr=make_utf8(args.prop_name)) to_add = "".join(args.values) command = "syspurpose add-{name} ".format(name=args.prop_name) + to_add check_result( syspurposestore, expectation=lambda res: all(x in res.get(args.prop_name, []) for x in args.values), success_msg=success_msg, command=command, attr=args.prop_name ) def remove_command(args, syspurposestore): """ Uses the syspurposestore to remove one or more values from a particular property. :param args: The parsed args from argparse, expected attributes are: prop_name: the string name of the property to add to values: A list of the values to remove from the given property (could be anything json-serializable) :param syspurposestore: An SyspurposeStore object to manipulate :return: None """ for value in args.values: if syspurposestore.remove(args.prop_name, value): print(_("Removed \"{val}\" from {name}.").format(val=make_utf8(value), name=make_utf8(args.prop_name))) else: print(_("Not removing value \"{val}\" from {name}; it was not there.").format( val=make_utf8(value), name=make_utf8(args.prop_name))) return success_msg = _("{attr} updated.").format(attr=make_utf8(args.prop_name)) to_remove = "".join(args.values) command = "syspurpose remove-{name} ".format(name=args.prop_name) + to_remove check_result( syspurposestore, expectation=lambda res: all(x not in res.get('addons', []) for x in args.values), success_msg=success_msg, command=command, attr=args.prop_name ) def set_command(args, syspurposestore): """ Uses the syspurposestore to set the prop_name to value. :param args: The parsed args from argparse, expected attributes are: prop_name: the string name of the property to set value: An object to set the property to (could be anything json-serializable) :param syspurposestore: An SyspurposeStore object to manipulate :return: None """ syspurposestore.set(args.prop_name, args.value) success_msg = _("{attr} set to \"{val}\".").format(attr=make_utf8(args.prop_name), val=make_utf8(args.value)) check_result( syspurposestore, expectation=lambda res: res.get(args.prop_name) == args.value, success_msg=success_msg, command="syspurpose set {name} \"{val}\"".format( name=args.prop_name, val=args.value), attr=args.prop_name ) def unset_command(args, syspurposestore): """ Uses the syspurposestore to unset (clear entirely) the prop_name. :param args: The parsed args from argparse, expected attributes are: prop_name: the string name of the property to unset (clear) :param syspurposestore: An SyspurposeStore object to manipulate :return: None """ syspurposestore.unset(args.prop_name) success_msg = _("{attr} unset.").format(attr=make_utf8(args.prop_name)) check_result( syspurposestore, expectation=lambda res: res.get(args.prop_name) in ["", None, []], success_msg=success_msg, command="syspurpose unset {name}".format(name=args.prop_name), attr=args.prop_name ) def show_contents(args, syspurposestore): """ :param args: :param syspurposestore: :return: """ sync_result = syspurposestore.sync() contents = sync_result.result contents = {key: contents[key] for key in contents if contents[key]} print(json.dumps(contents, indent=2, ensure_ascii=False, sort_keys=True)) return sync_result def setup_arg_parser(): """ Sets up argument parsing for the syspurpose tool. :return: An argparse.ArgumentParser ready to use to parse_args """ parser = argparse.ArgumentParser(prog="syspurpose", description=_("System Syspurpose Management Tool"), epilog=_("The 'syspurpose' command is deprecated and will be removed in a future major release." " Please use the 'subscription-manager syspurpose' command going forward.")) subparsers = parser.add_subparsers(help="sub-command help") # Arguments shared by subcommands add_options = argparse.ArgumentParser(add_help=False) add_options.add_argument("values", help=_("The value(s) to add"), nargs='+') add_options.set_defaults(func=add_command, requires_sync=True) remove_options = argparse.ArgumentParser(add_help=False) remove_options.add_argument("values", help=_("The value(s) to remove"), nargs='+') remove_options.set_defaults(func=remove_command, requires_sync=True) set_options = argparse.ArgumentParser(add_help=False) set_options.add_argument("value", help=_("The value to set"), action="store") set_options.set_defaults(func=set_command, requires_sync=True) unset_options = argparse.ArgumentParser(add_help=False) unset_options.set_defaults(func=unset_command, requires_sync=True) # Generic assignments # Set ################ generic_set_parser = subparsers.add_parser( "set", help=_("Sets the value for the given property") ) generic_set_parser.add_argument( "prop_name", metavar="property", help=_("The name of the property to set/update"), action="store" ) generic_set_parser.add_argument( "value", help=_("The value to set"), action="store" ) generic_set_parser.set_defaults(func=set_command, requires_sync=True) # Unset ############## generic_unset_parser = subparsers.add_parser( "unset", help=_("Unsets (clears) the value for the given property"), parents=[unset_options] ) generic_unset_parser.add_argument( "prop_name", metavar="property", help=_("The name of the property to set/update"), action="store" ) # Add ################ generic_add_parser = subparsers.add_parser( "add", help=_("Adds the value(s) to the given property") ) generic_add_parser.add_argument( "prop_name", metavar="property", help=_("The name of the property to update"), action="store" ) generic_add_parser.add_argument( "values", help=_("The value(s) to add"), action="store", nargs="+" ) generic_add_parser.set_defaults(func=add_command, requires_sync=True) # Remove ############# generic_remove_parser = subparsers.add_parser( "remove", help=_("Removes the value(s) from the given property") ) generic_remove_parser.add_argument( "prop_name", metavar="property", help=_("The name of the property to update"), action="store" ) generic_remove_parser.add_argument( "values", help=_("The value(s) to remove"), action="store", nargs="+" ) generic_remove_parser.set_defaults(func=remove_command, requires_sync=True) # Targeted commands # Roles ########## set_role_parser = subparsers.add_parser( "set-role", help=_("Set the system role to the system syspurpose"), parents=[set_options] ) # TODO: Set prop_name from schema file set_role_parser.set_defaults(prop_name="role") unset_role_parser = subparsers.add_parser( "unset-role", help=_("Clear set role"), parents=[unset_options] ) unset_role_parser.set_defaults(prop_name="role") # ADDONS ############# add_addons_parser = subparsers.add_parser( "add-addons", help=_("Add addons to the system syspurpose"), parents=[add_options] ) # TODO: Set prop_name from schema file add_addons_parser.set_defaults(prop_name="addons") remove_addons_parser = subparsers.add_parser( "remove-addons", help=_("Remove addons from the system syspurpose"), parents=[remove_options] ) remove_addons_parser.set_defaults(prop_name="addons") unset_role_parser = subparsers.add_parser( "unset-addons", help=_("Clear set addons"), parents=[unset_options] ) unset_role_parser.set_defaults(prop_name="addons") # SLA ################ set_sla_parser = subparsers.add_parser( "set-sla", help=_("Set the system sla"), parents=[set_options] ) set_sla_parser.set_defaults(prop_name="service_level_agreement") unset_sla_parser = subparsers.add_parser( "unset-sla", help=_("Clear set sla"), parents=[unset_options] ) unset_sla_parser.set_defaults(prop_name="service_level_agreement") # USAGE ############## set_usage_parser = subparsers.add_parser( "set-usage", help=_("Set the system usage"), parents=[set_options] ) set_usage_parser.set_defaults(prop_name="usage") unset_usage_parser = subparsers.add_parser( "unset-usage", help=_("Clear set usage"), parents=[unset_options] ) unset_usage_parser.set_defaults(prop_name="usage") # Pretty Print Json contents of default syspurpose file show_parser = subparsers.add_parser( "show", help=_("Show the current system syspurpose") ) show_parser.set_defaults(func=show_contents, requires_sync=False) return parser def main(): """ Run the cli (Do the syspurpose tool thing!!) :return: 0 """ parser = setup_arg_parser() args = parser.parse_args() # Syspurpose is not intended to be used in containers for the time being (could change later). if in_container(): print(_("WARNING: Setting syspurpose in containers has no effect.\n" "Please run syspurpose on the host.\n")) print(_("The 'syspurpose' command is deprecated and will be removed in a future major release." " Please use the 'subscription-manager syspurpose' command going forward."), file=sys.stderr) try: from subscription_manager.identity import Identity from subscription_manager.cp_provider import CPProvider identity = Identity() uuid = identity.uuid uep = CPProvider().get_consumer_auth_cp() except ImportError: uuid = None uep = None print(_("Warning: Unable to sync system purpose with subscription management server:" " subscription_manager module is not available.")) syspurposestore = SyncedStore(uep=uep, consumer_uuid=uuid, use_valid_fields=True) if getattr(args, 'func', None) is not None: args.func(args, syspurposestore) else: parser.print_help() return 0 return 0 def system_exit(code, msgs=None): """ Exit with a code and optional message(s). Saved a few lines of code. Note: this method was copied from subscription_manager, because syspurpose should be independent on subscription_manager module as much as possible. """ if msgs: if type(msgs) not in [type([]), type(())]: msgs = (msgs, ) for msg in msgs: if isinstance(msg, Exception): msg = "%s" % msg if isinstance(msg, six.text_type) and six.PY2: sys.stderr.write(msg.encode("utf8")) sys.stderr.write("\n") else: sys.stderr.write(msg) sys.stderr.write("\n") try: sys.stdout.flush() sys.stderr.flush() except IOError: pass sys.exit(code) def check_result(syspurposestore, expectation, success_msg, command, attr): if syspurposestore: syspurposestore.sync() result = syspurposestore.get_cached_contents() else: result = {} if result and not expectation(result): advice = SP_ADVICE.format(command=command) value = result[attr] system_exit( os.EX_SOFTWARE, msgs=_(SP_CONFLICT_MESSAGE.format( attr=attr, download_value=value, advice=advice )) ) else: print(success_msg)

| ver. 1.4 | Github | . | PHP 8.0.30 | Generation time: 1.4 | proxy | phpinfo | Settings