FM / cli

/usr/lib/python3/dist-packages/uaclient/cli/

/usr/lib/python3/dist-packages/uaclient/cli

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
api.py	2460	0644	edit dl rm
attach.py	5611	0644	edit dl rm
auto_attach.py	958	0644	edit dl rm
cli_util.py	8273	0644	edit dl rm
collect_logs.py	1774	0644	edit dl rm
commands.py	3765	0644	edit dl rm
config.py	10982	0644	edit dl rm
cve.py	6891	0644	edit dl rm
cves.py	3869	0644	edit dl rm
detach.py	3464	0644	edit dl rm
disable.py	10906	0644	edit dl rm
enable.py	19265	0644	edit dl rm
fix.py	29315	0644	edit dl rm
formatter.py	8955	0644	edit dl rm
help.py	1707	0644	edit dl rm
parser.py	2782	0644	edit dl rm
refresh.py	2480	0644	edit dl rm
security_status.py	2860	0644	edit dl rm
status.py	2424	0644	edit dl rm
system.py	1074	0644	edit dl rm
__init__.py	9501	0644	edit dl rm

Edit: /usr/lib/python3/dist-packages/uaclient/cli/collect_logs.py (1774B)

import logging
import os
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:
            old_umask = os.umask(0o177)
            try:
                with tarfile.open(output_file, "x:gz") as results:
                    results.add(output_dir, arcname="logs/")
            finally:
                os.umask(old_umask)
        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,
                )
            ]
        )
    ],
)

import logging import os 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: old_umask = os.umask(0o177) try: with tarfile.open(output_file, "x:gz") as results: results.add(output_dir, arcname="logs/") finally: os.umask(old_umask) 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, ) ] ) ], )