Viewing file:  collect_logs.py (1.6 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
import logging
import sys
import tarfile
import tempfile
from uaclient import messages
from uaclient.actions import collect_logs
from uaclient.cli.commands import ProArgument, ProArgumentGroup, ProCommand
from uaclient.cli.parser import HelpCategory
from uaclient.util import replace_top_level_logger_name
PRO_COLLECT_LOGS_FILE = "pro_logs.tar.gz"
LOG = logging.getLogger(replace_top_level_logger_name(__name__))
def action_collect_logs(args, *, cfg, **kwargs):
output_file = args.output or PRO_COLLECT_LOGS_FILE
with tempfile.TemporaryDirectory() as output_dir:
collect_logs(cfg, output_dir)
try:
with tarfile.open(output_file, "x:gz") as results:
results.add(output_dir, arcname="logs/")
except PermissionError as e:
LOG.error(e)
return 1
except FileExistsError as e:
LOG.error(e)
print(
messages.E_FILE_ALREADY_EXISTS.format(
filename=output_file,
),
file=sys.stderr,
)
return 1
return 0
collect_logs_command = ProCommand(
"collect-logs",
help=messages.CLI_ROOT_COLLECT_LOGS,
description=messages.CLI_COLLECT_LOGS_DESC,
preserve_description=True,
action=action_collect_logs,
help_category=HelpCategory.TROUBLESHOOT,
argument_groups=[
ProArgumentGroup(
arguments=[
ProArgument(
"--output",
short_name="-o",
help=messages.CLI_COLLECT_LOGS_OUTPUT,
)
]
)
],
)
|