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

import json
import sys
from collections import OrderedDict
from typing import Any, Optional  # noqa: F401

from uaclient import exceptions, messages
from uaclient.api import AbstractProgress
from uaclient.api.api import call_api
from uaclient.cli.commands import ProArgument, ProArgumentGroup, ProCommand
from uaclient.cli.parser import HelpCategory

class CLIAPIProgress(AbstractProgress):
    def progress(
        self,
        *,
        total_steps: int,
        done_steps: int,
        previous_step_message: Optional[str],
        current_step_message: Optional[str]
    ):
        d = OrderedDict()  # type: OrderedDict[str, Any]
        d["total_steps"] = total_steps
        d["done_steps"] = done_steps
        d["previous_step_message"] = previous_step_message
        d["current_step_message"] = current_step_message
        print(json.dumps(d))

def action_api(args, *, cfg, **kwargs):
    if args.options and args.data:
        raise exceptions.CLIAPIOptionsXORData()

if args.data and args.data == "-":
        if not sys.stdin.isatty():
            args.data = sys.stdin.read()

if args.show_progress:
        progress = CLIAPIProgress()
    else:
        progress = None

result = call_api(
        args.endpoint_path, args.options, args.data, cfg, progress
    )
    print(result.to_json())
    return 0 if result.result == "success" else 1

api_command = ProCommand(
    "api",
    help=messages.CLI_ROOT_API,
    description=messages.CLI_API_DESC,
    action=action_api,
    help_category=HelpCategory.OTHER,
    argument_groups=[
        ProArgumentGroup(
            arguments=[
                ProArgument(
                    "endpoint_path",
                    help=messages.CLI_API_ENDPOINT,
                    metavar="endpoint",
                ),
                ProArgument(
                    "--show-progress",
                    help=messages.CLI_API_SHOW_PROGRESS,
                    action="store_true",
                ),
                ProArgument(
                    "--args",
                    help=messages.CLI_API_ARGS,
                    dest="options",
                    default=[],
                    nargs="*",
                ),
                ProArgument(
                    "--data",
                    help=messages.CLI_API_DATA,
                    dest="data",
                    default="",
                ),
            ]
        )
    ],
)