wangzihao 17d2d5175e Remove six
Remove all usages of six.

Change-Id: I6ea8d1cdc3060d5c2a4311c7454b66ba75109b0c
2020-10-13 11:02:06 +08:00

229 lines
7.7 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.
from osc_lib.command import command
from osc_lib import utils as osc_utils
from troveclient import exceptions
from troveclient.i18n import _
class ListDatabaseLogs(command.Lister):
_description = _("Lists the log files available for instance.")
columns = ['Name', 'Type', 'Status', 'Published', 'Pending',
'Container', 'Prefix']
def get_parser(self, prog_name):
parser = super(ListDatabaseLogs, self).get_parser(prog_name)
parser.add_argument(
'instance',
metavar='<instance>',
help=_('ID or name of the instance.')
)
return parser
def take_action(self, parsed_args):
database_instances = self.app.client_manager.database.instances
instance = osc_utils.find_resource(database_instances,
parsed_args.instance)
log_list = database_instances.log_list(instance)
logs = [osc_utils.get_item_properties(l, self.columns)
for l in log_list]
return self.columns, logs
class SetDatabaseInstanceLog(command.ShowOne):
_description = _("Instructs Trove guest to operate logs.")
def get_parser(self, prog_name):
parser = super(SetDatabaseInstanceLog, self).get_parser(prog_name)
parser.add_argument(
'instance',
metavar='<instance>',
type=str,
help=_('Id or Name of the instance.')
)
parser.add_argument(
'log_name',
metavar='<log_name>',
type=str,
help=_('Name of log to operate.')
)
parser.add_argument(
'--enable',
action='store_true',
help="Whether or not to enable log collection.",
)
parser.add_argument(
'--disable',
action='store_true',
help="Whether or not to disable log collection.",
)
parser.add_argument(
'--publish',
action='store_true',
help="Whether or not to publish log files to the backend storage "
"for logs(Swift by default).",
)
parser.add_argument(
'--discard',
action='store_true',
help="Whether or not to discard the existing logs before publish.",
)
return parser
def take_action(self, parsed_args):
db_instances = self.app.client_manager.database.instances
instance = osc_utils.find_resource(db_instances,
parsed_args.instance)
log_info = db_instances.log_action(
instance, parsed_args.log_name,
enable=parsed_args.enable,
disable=parsed_args.disable,
discard=parsed_args.discard,
publish=parsed_args.publish
)
result = log_info._info
return zip(*sorted(result.items()))
class ShowDatabaseInstanceLog(command.ShowOne):
_description = _("Show information of given log name for the database "
"instance.")
def get_parser(self, prog_name):
parser = super(ShowDatabaseInstanceLog, self).get_parser(prog_name)
parser.add_argument(
'instance',
metavar='<instance>',
type=str,
help=_('Id or Name of the instance.')
)
parser.add_argument(
'log_name',
metavar='<log_name>',
type=str,
help=_('Name of log to operate.')
)
return parser
def take_action(self, parsed_args):
db_instances = self.app.client_manager.database.instances
instance = osc_utils.find_resource(db_instances,
parsed_args.instance)
log_info = db_instances.log_show(instance, parsed_args.log_name)
result = log_info._info
return zip(*sorted(result.items()))
class ShowDatabaseInstanceLogContents(command.Command):
_description = _("Show the content of log file.")
def get_parser(self, prog_name):
parser = super(ShowDatabaseInstanceLogContents, self).get_parser(
prog_name)
parser.add_argument(
'instance',
metavar='<instance>',
type=str,
help=_('Id or Name of the instance.')
)
parser.add_argument(
'log_name',
metavar='<log_name>',
type=str,
help=_('Name of log to operate.')
)
parser.add_argument(
'--lines', default=50, type=int,
help="The number of log lines can be shown in batch.",
)
return parser
def take_action(self, parsed_args):
db_instances = self.app.client_manager.database.instances
instance = osc_utils.find_resource(db_instances,
parsed_args.instance)
try:
log_gen = db_instances.log_generator(instance,
parsed_args.log_name,
lines=parsed_args.lines)
for log_part in log_gen():
print(log_part, end="")
except exceptions.GuestLogNotFoundError:
print(
"ERROR: No published '%(log_name)s' log was found for "
"%(instance)s" % {'log_name': parsed_args.log_name,
'instance': instance}
)
except Exception as ex:
error_msg = ex.message.split('\n')
print(error_msg[0])
class SaveDatabaseInstanceLog(command.Command):
_description = _("Save the log file.")
def get_parser(self, prog_name):
parser = super(SaveDatabaseInstanceLog, self).get_parser(prog_name)
parser.add_argument(
'instance',
metavar='<instance>',
type=str,
help=_('Id or Name of the instance.')
)
parser.add_argument(
'log_name',
metavar='<log_name>',
type=str,
help=_('Name of log to operate.')
)
parser.add_argument(
'--file',
help="Path of file to save log to for instance.",
)
return parser
def take_action(self, parsed_args):
db_instances = self.app.client_manager.database.instances
instance = osc_utils.find_resource(db_instances,
parsed_args.instance)
try:
filepath = db_instances.log_save(instance,
parsed_args.log_name,
filename=parsed_args.file)
print(_('Log "%(log_name)s" written to %(file_name)s')
% {'log_name': parsed_args.log_name,
'file_name': filepath})
except exceptions.GuestLogNotFoundError:
print(
"ERROR: No published '%(log_name)s' log was found for "
"%(instance)s" % {'log_name': parsed_args.log_name,
'instance': instance}
)
except Exception as ex:
error_msg = ex.message.split('\n')
print(error_msg[0])