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/api.py (2460B)

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,
    preserve_description=True,
    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="",
                ),
            ]
        )
    ],
)

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, preserve_description=True, 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="", ), ] ) ], )