17d2d5175e
Remove all usages of six. Change-Id: I6ea8d1cdc3060d5c2a4311c7454b66ba75109b0c
343 lines
11 KiB
Python
343 lines
11 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
"""Database v1 Users action implementations"""
|
|
|
|
from osc_lib.command import command
|
|
from osc_lib import exceptions
|
|
from osc_lib import utils
|
|
|
|
from troveclient.i18n import _
|
|
|
|
|
|
class CreateDatabaseUser(command.Command):
|
|
|
|
_description = _("Creates a user on an instance.")
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(CreateDatabaseUser, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'instance',
|
|
metavar='<instance>',
|
|
help=_('ID or name of the instance.')
|
|
)
|
|
parser.add_argument(
|
|
'name',
|
|
metavar='<name>',
|
|
help=_('Name of user.')
|
|
)
|
|
parser.add_argument(
|
|
'password',
|
|
metavar='<password>',
|
|
help=_('Password of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--host',
|
|
metavar='<host>',
|
|
help=_('Optional host of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--databases',
|
|
metavar='<databases>',
|
|
nargs='+',
|
|
default=[],
|
|
help=_('Optional list of databases.')
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
manager = self.app.client_manager.database
|
|
users = manager.users
|
|
instance = utils.find_resource(manager.instances,
|
|
parsed_args.instance)
|
|
databases = [{'name': value} for value in parsed_args.databases]
|
|
user = {'name': parsed_args.name, 'password': parsed_args.password,
|
|
'databases': databases}
|
|
if parsed_args.host:
|
|
user['host'] = parsed_args.host
|
|
users.create(instance, [user])
|
|
|
|
|
|
class ListDatabaseUsers(command.Lister):
|
|
|
|
_description = _("Lists the users for an instance.")
|
|
columns = ['Name', 'Host', 'Databases']
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(ListDatabaseUsers, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
dest='instance',
|
|
metavar='<instance>',
|
|
help=_('ID or name of the instance.')
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
manager = self.app.client_manager.database
|
|
db_users = manager.users
|
|
instance = utils.find_resource(manager.instances,
|
|
parsed_args.instance)
|
|
items = db_users.list(instance)
|
|
users = items
|
|
while (items.next):
|
|
items = db_users.list(parsed_args.instance, marker=items.next)
|
|
users += items
|
|
for user in users:
|
|
db_names = [db['name'] for db in user.databases]
|
|
user.databases = ', '.join(db_names)
|
|
users = [utils.get_item_properties(u, self.columns) for u in users]
|
|
return self.columns, users
|
|
|
|
|
|
class ShowDatabaseUser(command.ShowOne):
|
|
|
|
_description = _("Shows details of a database user of an instance.")
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(ShowDatabaseUser, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'instance',
|
|
metavar='<instance>',
|
|
help=_('ID or name of the instance.'),
|
|
)
|
|
parser.add_argument(
|
|
'name',
|
|
metavar='<name>',
|
|
help=_('Name of user.'),
|
|
)
|
|
parser.add_argument(
|
|
"--host",
|
|
metavar="<host>",
|
|
help=_("Optional host of user."),
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
manager = self.app.client_manager.database
|
|
db_users = manager.users
|
|
instance = utils.find_resource(manager.instances,
|
|
parsed_args.instance)
|
|
user = db_users.get(instance, parsed_args.name,
|
|
hostname=parsed_args.host)
|
|
return zip(*sorted(user._info.items()))
|
|
|
|
|
|
class DeleteDatabaseUser(command.Command):
|
|
|
|
_description = _("Deletes a user from an instance.")
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(DeleteDatabaseUser, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'instance',
|
|
metavar='<instance>',
|
|
help=_('ID or name of the instance.')
|
|
)
|
|
parser.add_argument(
|
|
'name',
|
|
metavar='<name>',
|
|
help=_('Name of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--host',
|
|
metavar='<host>',
|
|
help=_('Optional host of user.')
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
manager = self.app.client_manager.database
|
|
users = manager.users
|
|
try:
|
|
instance = utils.find_resource(manager.instances,
|
|
parsed_args.instance)
|
|
users.delete(instance, parsed_args.name, parsed_args.host)
|
|
except Exception as e:
|
|
msg = (_("Failed to delete user %(user)s: %(e)s")
|
|
% {'user': parsed_args.name, 'e': e})
|
|
raise exceptions.CommandError(msg)
|
|
|
|
|
|
class GrantDatabaseUserAccess(command.Command):
|
|
|
|
_description = _("Grants access to a database(s) for a user.")
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(GrantDatabaseUserAccess, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'instance',
|
|
metavar='<instance>',
|
|
help=_('ID or name of the instance.')
|
|
)
|
|
parser.add_argument(
|
|
'name',
|
|
metavar='<name>',
|
|
help=_('Name of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--host',
|
|
metavar='<host>',
|
|
default=None,
|
|
help=_('Optional host of user.')
|
|
)
|
|
parser.add_argument(
|
|
'databases',
|
|
metavar='<databases>',
|
|
nargs="+",
|
|
default=[],
|
|
help=_('List of databases.')
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
manager = self.app.client_manager.database
|
|
users = manager.users
|
|
instance = utils.find_resource(manager.instances,
|
|
parsed_args.instance)
|
|
users.grant(instance, parsed_args.name,
|
|
parsed_args.databases, hostname=parsed_args.host)
|
|
|
|
|
|
class RevokeDatabaseUserAccess(command.Command):
|
|
|
|
_description = _("Revokes access to a database for a user.")
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(RevokeDatabaseUserAccess, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'instance',
|
|
metavar='<instance>',
|
|
help=_('ID or name of the instance.')
|
|
)
|
|
parser.add_argument(
|
|
'name',
|
|
metavar='<name>',
|
|
help=_('Name of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--host',
|
|
metavar='<host>',
|
|
default=None,
|
|
help=_('Optional host of user.')
|
|
)
|
|
parser.add_argument(
|
|
'databases',
|
|
metavar='<databases>',
|
|
help=_('A single database.')
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
manager = self.app.client_manager.database
|
|
users = manager.users
|
|
instance = utils.find_resource(manager.instances,
|
|
parsed_args.instance)
|
|
users.revoke(instance, parsed_args.name,
|
|
parsed_args.databases, hostname=parsed_args.host)
|
|
|
|
|
|
class ShowDatabaseUserAccess(command.Lister):
|
|
|
|
_description = _("Shows access details of a user of an instance.")
|
|
columns = ['Name']
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(ShowDatabaseUserAccess, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'instance',
|
|
metavar='<instance>',
|
|
help=_('ID or name of the instance.')
|
|
)
|
|
parser.add_argument(
|
|
'name',
|
|
metavar='<name>',
|
|
help=_('Name of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--host',
|
|
metavar='<host>',
|
|
default=None,
|
|
help=_('Optional host of user.')
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
manager = self.app.client_manager.database
|
|
users = manager.users
|
|
instance = utils.find_resource(manager.instances,
|
|
parsed_args.instance)
|
|
names = users.list_access(instance, parsed_args.name,
|
|
hostname=parsed_args.host)
|
|
access = [utils.get_item_properties(n, self.columns) for n in names]
|
|
return self.columns, access
|
|
|
|
|
|
class UpdateDatabaseUserAttributes(command.Command):
|
|
|
|
_description = _("Updates a user's attributes on an instance."
|
|
"At least one optional argument must be provided.")
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(UpdateDatabaseUserAttributes,
|
|
self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'instance',
|
|
metavar='<instance>',
|
|
help=_('ID or name of the instance.')
|
|
)
|
|
parser.add_argument(
|
|
'name',
|
|
metavar='<name>',
|
|
help=_('Name of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--host',
|
|
metavar='<host>',
|
|
default=None,
|
|
help=_('Optional host of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--new_name',
|
|
metavar='<new_name>',
|
|
default=None,
|
|
help=_('Optional new name of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--new_password',
|
|
metavar='<new_password>',
|
|
default=None,
|
|
help=_('Optional new password of user.')
|
|
)
|
|
parser.add_argument(
|
|
'--new_host',
|
|
metavar='<new_host>',
|
|
default=None,
|
|
help=_('Optional new host of user.')
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
manager = self.app.client_manager.database
|
|
users = manager.users
|
|
instance = utils.find_resource(manager.instances,
|
|
parsed_args.instance)
|
|
new_attrs = {}
|
|
if parsed_args.new_name:
|
|
new_attrs['name'] = parsed_args.new_name
|
|
if parsed_args.new_password:
|
|
new_attrs['password'] = parsed_args.new_password
|
|
if parsed_args.new_host:
|
|
new_attrs['host'] = parsed_args.new_host
|
|
users.update_attributes(instance, parsed_args.name,
|
|
newuserattr=new_attrs,
|
|
hostname=parsed_args.host)
|