azure-cli/src/azure-cli/azure/cli/command_modules/resource/_completers.py at 7125fda124eea93d2d0cdc73c703141ed467f3d0 · mentat9/azure-cli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# --------------------------------------------------------------------------------------------

from azure.cli.core.decorators import Completer

from azure.cli.command_modules.resource._client_factory import (
   #_resource_policy_client_factory,
   _resource_client_factory)


# @Completer
# def get_policy_completion_list(cmd, prefix, namespace, **kwargs):  # pylint: disable=unused-argument
#     policy_client = _resource_policy_client_factory(cmd.cli_ctx)
#     result = policy_client.policy_definitions.list()
#     return [i.name for i in result]


# @Completer
# def get_policy_set_completion_list(cmd, prefix, namespace, **kwargs):  # pylint: disable=unused-argument
#     policy_client = _resource_policy_client_factory(cmd.cli_ctx)
#     result = policy_client.policy_set_definitions.list()
#     return [i.name for i in result]


# @Completer
# def get_policy_assignment_completion_list(cmd, prefix, namespace, **kwargs):  # pylint: disable=unused-argument
#     policy_client = _resource_policy_client_factory(cmd.cli_ctx)
#     result = policy_client.policy_assignments.list()
#     return [i.name for i in result]


# @Completer
# def get_policy_exemption_completion_list(cmd, prefix, namespace, **kwargs):  # pylint: disable=unused-argument
#     policy_client = _resource_policy_client_factory(cmd.cli_ctx)
#     result = policy_client.policy_exemptions.list()
#     return [i.name for i in result]


@Completer
def get_providers_completion_list(cmd, prefix, namespace, **kwargs):  # pylint: disable=unused-argument
    rcf = _resource_client_factory(cmd.cli_ctx)
    result = rcf.providers.list()
    return [r.namespace for r in result]


@Completer
def get_resource_types_completion_list(cmd, prefix, namespace, **kwargs):  # pylint: disable=unused-argument
    rcf = _resource_client_factory(cmd.cli_ctx)
    result = rcf.providers.list()
    types = []
    for p in list(result):
        for r in p.resource_types:
            types.append(p.namespace + '/' + r.resource_type)
    return types