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: status.py

import time

from uaclient import actions, config, event_logger, messages, status, util
from uaclient.cli.commands import ProArgument, ProArgumentGroup, ProCommand
from uaclient.cli.parser import HelpCategory

event = event_logger.get_event_logger()

def action_status(args, *, cfg: config.UAConfig, **kwargs):
    if not cfg:
        cfg = config.UAConfig()
    show_all = args.all if args else False
    token = args.simulate_with_token if args else None
    active_value = status.UserFacingConfigStatus.ACTIVE.value
    status_dict, ret = actions.status(
        cfg, simulate_with_token=token, show_all=show_all
    )
    config_active = bool(status_dict["execution_status"] == active_value)

if args and args.wait and config_active:
        while status_dict["execution_status"] == active_value:
            event.info(".", end="")
            time.sleep(1)
            status_dict, ret = actions.status(
                cfg,
                simulate_with_token=token,
                show_all=show_all,
            )
        event.info("")

event.set_output_content(status_dict)
    output = status.format_tabular(status_dict, show_all=show_all)
    event.info(util.handle_unicode_characters(output))
    event.process_events()
    return ret

status_command = ProCommand(
    "status",
    help=messages.CLI_ROOT_STATUS,
    description=messages.CLI_STATUS_DESC,
    action=action_status,
    preserve_description=True,
    help_category=HelpCategory.QUICKSTART,
    help_position=1,
    argument_groups=[
        ProArgumentGroup(
            arguments=[
                ProArgument(
                    "--wait",
                    help=messages.CLI_STATUS_WAIT,
                    action="store_true",
                ),
                ProArgument(
                    "--format",
                    help=messages.CLI_FORMAT_DESC.format(default="tabular"),
                    action="store",
                    choices=["tabular", "json", "yaml"],
                    default="tabular",
                ),
                ProArgument(
                    "--simulate-with-token",
                    help=messages.CLI_STATUS_SIMULATE_WITH_TOKEN,
                    metavar="TOKEN",
                    action="store",
                ),
                ProArgument(
                    "--all", help=messages.CLI_STATUS_ALL, action="store_true"
                ),
            ]
        )
    ],
)