class AddDefaultInfo(OrganizationAction): description = _("add default custom info") def __init__(self): super(AddDefaultInfo, self).__init__() self.default_info_api = OrganizationDefaultInfoAPI() def setup_parser(self, parser): parser.add_option('--name', dest='name', help=_("organization name eg: foo.example.com (required)")) parser.add_option('--type', dest='type', help=_("'system' (required)")) parser.add_option('--keyname', dest='keyname', help=_("name of the default custom info (required)")) def check_options(self, validator): validator.require(('name', 'keyname', 'type')) def run(self): org_name = self.get_option('name') keyname = self.get_option('keyname') informable_type = self.get_option('type').lower() response = self.default_info_api.create(org_name, informable_type, keyname) output_hash = {'keyname': keyname, 'org_name': org_name, 'katello_obj': informable_type.capitalize()} if response: print _("Successfully added [ %(katello_obj)s ] " \ + "default custom info [ %(keyname)s ] to Org [ %(org_name)s ]") \ % output_hash else: print _("Could not add [ %(katello_obj)s ] " \ + "default custom info [ %(keyname)s ] to Org [ %(org_name)s ]") \ % output_hash