OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	10/28/2024 08:41:34 AM	rwxr-xr-x
📄 __init__.py	9.16 KB	08/08/2024 08:21:51 PM	rw-r--r--
📁 __pycache__	-	10/28/2024 08:41:34 AM	rwxr-xr-x
📄 api.py	2.37 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 attach.py	5.48 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 auto_attach.py	958 bytes	08/08/2024 08:21:51 PM	rw-r--r--
📄 cli_util.py	5.93 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 collect_logs.py	1.3 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 commands.py	3.68 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 config.py	9.9 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 detach.py	3.38 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 disable.py	10.62 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 enable.py	17.02 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 fix.py	28.7 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 help.py	1.67 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 parser.py	2.64 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 refresh.py	2.42 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 security_status.py	2.79 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 status.py	2.37 KB	08/08/2024 08:21:51 PM	rw-r--r--
📄 system.py	1.05 KB	08/08/2024 08:21:51 PM	rw-r--r--

Editing: refresh.py

import logging

from uaclient import apt_news, config, contract, exceptions, messages, util
from uaclient.cli import cli_util
from uaclient.cli.commands import ProArgument, ProArgumentGroup, ProCommand
from uaclient.cli.parser import HelpCategory
from uaclient.timer.update_messaging import refresh_motd, update_motd_messages

LOG = logging.getLogger(util.replace_top_level_logger_name(__name__))

def _action_refresh_config(args, cfg: config.UAConfig):
    try:
        cfg.process_config()
    except RuntimeError as exc:
        LOG.exception(exc)
        raise exceptions.RefreshConfigFailure()
    print(messages.REFRESH_CONFIG_SUCCESS)

@cli_util.assert_attached()
def _action_refresh_contract(_args, cfg: config.UAConfig):
    try:
        contract.refresh(cfg)
    except exceptions.ConnectivityError:
        raise exceptions.RefreshContractFailure()
    print(messages.REFRESH_CONTRACT_SUCCESS)

def _action_refresh_messages(_args, cfg: config.UAConfig):
    # Not performing any exception handling here since both of these
    # functions should raise UbuntuProError exceptions, which are
    # covered by the main_error_handler decorator
    try:
        update_motd_messages(cfg)
        refresh_motd()
        if cfg.apt_news:
            apt_news.update_apt_news(cfg)
    except Exception as exc:
        LOG.exception(exc)
        raise exceptions.RefreshMessagesFailure()
    else:
        print(messages.REFRESH_MESSAGES_SUCCESS)

@cli_util.assert_root
@cli_util.assert_lock_file("pro refresh")
def action_refresh(args, *, cfg: config.UAConfig, **kwargs):
    if args.target is None or args.target == "config":
        _action_refresh_config(args, cfg)

if args.target is None or args.target == "contract":
        _action_refresh_contract(args, cfg)

if args.target is None or args.target == "messages":
        _action_refresh_messages(args, cfg)

return 0

refresh_command = ProCommand(
    "refresh",
    help=messages.CLI_ROOT_REFRESH,
    description=messages.CLI_REFRESH_DESC,
    action=action_refresh,
    preserve_description=True,
    help_category=HelpCategory.OTHER,
    argument_groups=[
        ProArgumentGroup(
            arguments=[
                ProArgument(
                    "target",
                    help=messages.CLI_REFRESH_TARGET,
                    nargs="?",
                    choices=["contract", "config", "messages"],
                    default=None,
                )
            ]
        )
    ],
)