| Current File : //proc/self/root/usr/lib/python3/dist-packages/uaclient/cli/security_status.py |
import json
from uaclient import messages, security_status, util
from uaclient.cli.commands import (
ProArgument,
ProArgumentGroup,
ProArgumentMutuallyExclusiveGroup,
ProCommand,
)
from uaclient.cli.parser import HelpCategory
from uaclient.yaml import safe_dump
def action_security_status(args, *, cfg, **kwargs):
if args.format == "text":
if args.thirdparty:
security_status.list_third_party_packages()
elif args.unavailable:
security_status.list_unavailable_packages()
elif args.esm_infra:
security_status.list_esm_infra_packages(cfg)
elif args.esm_apps:
security_status.list_esm_apps_packages(cfg)
else:
security_status.security_status(cfg)
elif args.format == "json":
print(
json.dumps(
security_status.security_status_dict(cfg),
sort_keys=True,
cls=util.DatetimeAwareJSONEncoder,
)
)
else:
print(
safe_dump(
security_status.security_status_dict(cfg),
default_flow_style=False,
)
)
return 0
security_status_command = ProCommand(
"security-status",
help=messages.CLI_ROOT_SECURITY_STATUS,
description=messages.CLI_SS_DESC,
preserve_description=True,
action=action_security_status,
help_category=HelpCategory.QUICKSTART,
help_position=5,
argument_groups=[
ProArgumentGroup(
arguments=[
ProArgument(
"--format",
help=messages.CLI_FORMAT_DESC.format(default="text"),
choices=("json", "yaml", "text"),
default="text",
)
],
mutually_exclusive_groups=[
ProArgumentMutuallyExclusiveGroup(
arguments=[
ProArgument(
"--thirdparty",
help=messages.CLI_SS_THIRDPARTY,
action="store_true",
),
ProArgument(
"--unavailable",
help=messages.CLI_SS_UNAVAILABLE,
action="store_true",
),
ProArgument(
"--esm-infra",
help=messages.CLI_SS_ESM_INFRA,
action="store_true",
),
ProArgument(
"--esm-apps",
help=messages.CLI_SS_ESM_APPS,
action="store_true",
),
]
)
],
)
],
)