def load_arguments(self, _):
with self.argument_context('graph query') as c:
c.argument('graph_query',
options_list=['--graph-query', '--q', '-q'],
required=True,
completer=get_generic_completion_list(_QUERY_EXAMPLES),
help='Resource Graph query to execute.')
c.argument(
'first',
options_list=['--first'],
required=False,
type=int,
default=100,
help=
'The maximum number of objects to return. Accepted range: 1-5000.')
c.argument(
'skip',
options_list=['--skip'],
required=False,
type=int,
default=0,
help=
'Ignores the first N objects and then gets the remaining objects.')
c.argument(
'subscriptions',
options_list=['--subscriptions', '-s'],
nargs='*',
required=False,
default=None,
help=
'List of subscriptions to run query against. By default all accessible subscriptions are queried.'
)
def load_arguments(self, _):
with self.argument_context('graph query') as c:
c.argument('graph_query', options_list=['--graph-query', '--q', '-q'], required=True,
completer=get_generic_completion_list(_QUERY_EXAMPLES), help='Resource Graph query to execute.')
c.argument('first', options_list=['--first'], required=False, type=int, default=None,
help='The maximum number of objects to return. Accepted range: 1-1000. Default value is 100.')
c.argument('skip', options_list=['--skip'], required=False, type=int, default=None,
help='Ignores the first N objects and then gets the remaining objects. Default value is 0.')
c.argument('subscriptions', options_list=['--subscriptions', '-s'], nargs='*', required=False, default=None,
help='List of subscriptions to run query against. By default all accessible subscriptions are queried.')
c.argument('management_groups', options_list=['--management-groups', '-m'], nargs='*', required=False, default=None,
help='List of management groups to run query against.')
c.argument('skip_token', options_list=['--skip-token'], type=str, required=False, default=None,
help='Skip token to get the next page of the query if applicable.')
c.argument('allow_partial_scopes', options_list=['--allow-partial-scopes', '-a'],
arg_type=get_three_state_flag(), required=False, default=False,
help='Indicates if query should succeed when only partial number of subscription underneath can be processed by server.')
def load_arguments(self, _):
from ._completers import storage_account_completion_list, storage_account_key_completion_list
from knack.arguments import CLIArgumentType
node_size_type = CLIArgumentType(
arg_group='Node',
help=
'The size of the node. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-hadoop-provision-linux-clusters#configure-cluster-size'
)
with self.argument_context('hdinsight') as c:
c.argument('cluster_name',
arg_type=name_type,
completer=get_resource_name_completion_list(
'Microsoft.HDInsight/clusters'),
help='The name of the cluster.')
c.argument('tags', tags_type)
c.argument(
'cluster_version',
options_list=['--version', '-v'],
arg_group='Cluster',
help=
'The HDInsight cluster version. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-component-versioning#supported-hdinsight-versions'
)
c.argument(
'cluster_type',
options_list=['--type', '-t'],
arg_group='Cluster',
completer=get_generic_completion_list(known_cluster_types),
help='Type of HDInsight cluster, like: {}. '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight-'
'hadoop-provision-linux-clusters#cluster-types'.format(
', '.join(known_cluster_types)))
c.argument(
'component_version',
arg_group='Cluster',
nargs='*',
validator=validate_component_version,
help='The versions of various Hadoop components, in space-'
'separated versions in \'component=version\' format. Example: '
'Spark=2.0 Hadoop=2.7.3 '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight'
'-component-versioning#hadoop-components-available-with-different-'
'hdinsight-versions')
c.argument('cluster_configurations',
arg_group='Cluster',
help='Extra configurations of various components, in JSON.')
c.argument('cluster_tier',
arg_type=get_enum_type(['standard', 'premium']),
arg_group='Cluster',
help='The tier of the cluster: standard or premium.')
c.argument('http_username',
options_list=['--http-user', '-u'],
arg_group='HTTP',
help='HTTP username for the cluster. Default: admin.')
c.argument('http_password',
options_list=['--http-password', '-p'],
arg_group='HTTP',
help='HTTP password for the cluster.')
c.argument('ssh_username',
options_list=['--ssh-user', '-U'],
arg_group='SSH',
help='SSH username for the cluster nodes.')
c.argument(
'ssh_password',
options_list=['--ssh-password', '-P'],
arg_group='SSH',
help=
'SSH password for the cluster nodes. If none specified, uses the HTTP password.'
)
c.argument('ssh_public_key',
options_list=['--ssh-public-key', '-K'],
arg_group='SSH',
help='SSH public key for the cluster nodes.')
c.argument('headnode_size', arg_type=node_size_type)
c.argument('workernode_size', arg_type=node_size_type)
c.argument('workernode_data_disks_per_node',
arg_group='Node',
help='The number of data disks to use per worker node.')
c.argument(
'workernode_data_disk_storage_account_type',
arg_group='Node',
arg_type=get_enum_type(['standard_lrs', 'premium_lrs']),
help=
'The type of storage account that will be used for the data disks: standard_lrs or premium_lrs'
)
c.argument('workernode_data_disk_size',
arg_group='Node',
help='The size of the data disk in GB, e.g. 1023.')
c.argument('zookeepernode_size', arg_type=node_size_type)
c.argument('edgenode_size', arg_type=node_size_type)
c.argument('workernode_count',
options_list=['--size', '-s'],
arg_group='Cluster',
help='The number of worker nodes in the cluster.')
c.argument(
'storage_account',
arg_group='Storage',
completer=storage_account_completion_list,
help='The storage account, e.g. "<name>.blob.core.windows.net".')
c.argument(
'storage_account_key',
arg_group='Storage',
completer=storage_account_key_completion_list,
help='The storage account key. A key can be retrieved automatically '
'if the user has access to the storage account.')
c.argument('storage_default_container',
arg_group='Storage',
help='The storage container the cluster will use. '
'Uses the cluster name if none was specified. (WASB only)')
c.argument(
'storage_default_filesystem',
arg_group='Storage',
help='The storage filesystem the cluster will use. (DFS only)')
c.argument(
'virtual_network',
arg_group='Network',
help=
'The virtual network resource ID of an existing virtual network.')
c.argument(
'subnet_name',
arg_group='Network',
help='The name of the subnet in the specified virtual network.')
def load_arguments(self, _):
from ._completers import subnet_completion_list, cluster_admin_account_completion_list, \
cluster_user_group_completion_list, get_resource_name_completion_list_under_subscription
from knack.arguments import CLIArgumentType
from azure.mgmt.hdinsight.models import Tier, JsonWebKeyEncryptionAlgorithm
from argcomplete.completers import FilesCompleter
node_size_type = CLIArgumentType(
arg_group='Node',
help='The size of the node. See also: https://docs.microsoft.com/azure/'
'hdinsight/hdinsight-hadoop-provision-linux-clusters#configure-cluster-size'
)
# cluster
with self.argument_context('hdinsight') as c:
# Cluster
c.argument('cluster_name',
arg_type=name_type,
completer=get_resource_name_completion_list(
'Microsoft.HDInsight/clusters'),
help='The name of the cluster.')
c.argument('tags', tags_type)
c.argument(
'no_validation_timeout',
action='store_true',
help=
'Permit timeout error during argument validation phase. If omitted, '
'validation timeout error will be permitted.')
c.argument(
'cluster_version',
options_list=['--version', '-v'],
arg_group='Cluster',
help=
'The HDInsight cluster version. See also: https://docs.microsoft.com/azure/'
'hdinsight/hdinsight-component-versioning#supported-hdinsight-versions'
)
c.argument(
'cluster_type',
options_list=['--type', '-t'],
arg_group='Cluster',
completer=get_generic_completion_list(known_cluster_types),
help='Type of HDInsight cluster, like: {}. '
'See also: https://docs.microsoft.com/azure/hdinsight/hdinsight-'
'hadoop-provision-linux-clusters#cluster-types'.format(
', '.join(known_cluster_types)))
c.argument(
'component_version',
arg_group='Cluster',
nargs='*',
validator=validate_component_version,
help='The versions of various Hadoop components, in space-'
'separated versions in \'component=version\' format. Example: '
'Spark=2.0 Hadoop=2.7.3 '
'See also: https://docs.microsoft.com/azure/hdinsight/hdinsight'
'-component-versioning#hadoop-components-available-with-different-'
'hdinsight-versions')
c.argument(
'cluster_configurations',
arg_group='Cluster',
type=shell_safe_json_parse,
completer=FilesCompleter(),
help='Extra configurations of various components. '
'Configurations may be supplied from a file using the `@{path}` syntax or a JSON string. '
'See also: https://docs.microsoft.com/azure/hdinsight/'
'hdinsight-hadoop-customize-cluster-bootstrap')
c.argument('cluster_tier',
arg_type=get_enum_type(Tier),
arg_group='Cluster',
help='The tier of the cluster')
c.argument(
'esp',
arg_group='Cluster',
action='store_true',
help=
'Specify to create cluster with Enterprise Security Package. If omitted, '
'creating cluster with Enterprise Security Package will not not allowed.'
)
c.argument('minimal_tls_version',
arg_type=get_enum_type(['1.0', '1.1', '1.2']),
arg_group='Cluster',
help='The minimal supported TLS version.')
# HTTP
c.argument('http_username',
options_list=['--http-user', '-u'],
arg_group='HTTP',
help='HTTP username for the cluster. Default: admin.')
c.argument(
'http_password',
options_list=['--http-password', '-p'],
arg_group='HTTP',
help='HTTP password for the cluster. Will prompt if not given.')
# SSH
c.argument('ssh_username',
options_list=['--ssh-user', '-U'],
arg_group='SSH',
help='SSH username for the cluster nodes.')
c.argument(
'ssh_password',
options_list=['--ssh-password', '-P'],
arg_group='SSH',
help=
'SSH password for the cluster nodes. If none specified, uses the HTTP password.'
)
c.argument('ssh_public_key',
options_list=['--ssh-public-key', '-K'],
arg_group='SSH',
help='SSH public key for the cluster nodes.')
# Node
c.argument('headnode_size', arg_type=node_size_type)
c.argument('workernode_size', arg_type=node_size_type)
c.argument('workernode_data_disks_per_node',
arg_group='Node',
help='The number of data disks to use per worker node.')
c.argument(
'workernode_data_disk_storage_account_type',
arg_group='Node',
arg_type=get_enum_type(['standard_lrs', 'premium_lrs']),
help=
'The type of storage account that will be used for the data disks: standard_lrs or premium_lrs'
)
c.argument('workernode_data_disk_size',
arg_group='Node',
help='The size of the data disk in GB, e.g. 1023.')
c.argument('zookeepernode_size', arg_type=node_size_type)
c.argument('edgenode_size', arg_type=node_size_type)
c.argument('kafka_management_node_size', arg_type=node_size_type)
c.argument('workernode_count',
options_list=['--workernode-count', '-c'],
arg_group='Node',
help='The number of worker nodes in the cluster.')
c.argument('kafka_management_node_count',
arg_group='Node',
help='The number of kafka management node in the cluster')
# Storage
c.argument(
'storage_account',
arg_group='Storage',
validator=validate_storage_account,
completer=get_resource_name_completion_list_under_subscription(
'Microsoft.Storage/storageAccounts'),
help='The name or ID of the storage account.')
c.argument(
'storage_account_key',
arg_group='Storage',
help='The storage account key. A key can be retrieved automatically '
'if the user has access to the storage account.')
c.argument('storage_default_container',
arg_group='Storage',
options_list=['--storage-container'],
help='The storage container the cluster will use. '
'Uses the cluster name if none was specified. (WASB only)')
c.argument('storage_default_filesystem',
arg_group='Storage',
options_list=['--storage-filesystem'],
help='The storage filesystem the cluster will use. '
'Uses the cluster name if none was specified. (DFS only)')
c.argument(
'storage_account_managed_identity',
arg_group='Storage',
validator=validate_storage_msi,
completer=get_resource_name_completion_list_under_subscription(
'Microsoft.ManagedIdentity/userAssignedIdentities'),
help=
'User-assigned managed identity with access to the storage account filesystem. '
'Only required when storage account type is Azure Data Lake Storage Gen2.'
)
# Network
c.argument(
'vnet_name',
arg_group='Network',
validator=validate_subnet,
completer=get_resource_name_completion_list_under_subscription(
'Microsoft.Network/virtualNetworks'),
help='The name of a virtual network.')
c.argument(
'subnet',
arg_group='Network',
completer=subnet_completion_list,
help=
'The name or ID of subnet. If name is supplied, `--vnet-name` must be supplied.'
)
# Script Action
c.argument('script_action_name',
arg_group='Script Action',
help='The name of the script action.')
c.argument('script_uri',
arg_group='Script Action',
help='The URI to the script.')
c.argument('script_parameters',
arg_group='Script Action',
help='The parameters for the script.')
# Domain Service
c.argument(
'domain',
arg_group='Domain Service',
validator=validate_domain_service,
completer=get_resource_name_completion_list_under_subscription(
'Microsoft.AAD/domainServices'),
help=
'The name or resource ID of the user\'s Azure Active Directory Domain Service. '
'Required only when create cluster with Enterprise Security Package.'
)
c.argument(
'cluster_users_group_dns',
arg_group='Domain Service',
nargs='+',
completer=cluster_user_group_completion_list,
help=
'A space-delimited list of Distinguished Names for cluster user groups. '
'Required only when create cluster with Enterprise Security Package. '
)
c.argument(
'cluster_admin_password',
arg_group='Domain Service',
help='The domain admin password. '
'Required only when create cluster with Enterprise Security Package.'
)
c.argument(
'cluster_admin_account',
arg_group='Domain Service',
completer=cluster_admin_account_completion_list,
help=
'The domain user account that will have admin privileges on the cluster. '
'Required only when create cluster with Enterprise Security Package.'
)
c.argument(
'ldaps_urls',
arg_group='Domain Service',
nargs='+',
help=
'A space-delimited list of LDAPS protocol URLs to communicate with the Active Directory. '
'Required only when create cluster with Enterprise Security Package.'
)
# Customer Managed Key
c.argument(
'encryption_vault_uri',
arg_group='Customer Managed Key',
help=
'Base key vault URI where the customers key is located eg. https://myvault.vault.azure.net'
)
c.argument('encryption_key_name',
arg_group='Customer Managed Key',
help='Key name that is used for enabling disk encryption.')
c.argument(
'encryption_key_version',
arg_group='Customer Managed Key',
help='Key version that is used for enabling disk encryption.')
c.argument('encryption_algorithm',
arg_type=get_enum_type(JsonWebKeyEncryptionAlgorithm),
arg_group='Customer Managed Key',
help='Algorithm identifier for encryption.')
# Kafka Rest Proxy
c.argument(
'kafka_client_group_id',
arg_group='Kafka Rest Proxy',
help='The client AAD security group id for Kafka Rest Proxy')
c.argument(
'kafka_client_group_name',
arg_group='Kafka Rest Proxy',
help='The client AAD security group name for Kafka Rest Proxy')
# Managed Service Identity
c.argument(
'assign_identity',
arg_group='Managed Service Identity',
validator=validate_msi,
completer=get_resource_name_completion_list_under_subscription(
'Microsoft.ManagedIdentity/userAssignedIdentities'),
help="The name or ID of user assigned identity.")
# resize
with self.argument_context('hdinsight resize') as c:
c.argument(
'target_instance_count',
options_list=['--workernode-count', '-c'],
help='The target worker node instance count for the operation.',
required=True)
# application
with self.argument_context('hdinsight application') as c:
c.argument('cluster_name',
options_list=['--cluster-name'],
completer=get_resource_name_completion_list(
'Microsoft.HDInsight/clusters'),
help='The name of the cluster.')
c.argument('application_name',
arg_type=name_type,
help='The constant value for the application name.')
c.argument('application_type',
options_list=['--type', '-t'],
arg_type=get_enum_type(['CustomApplication', 'RServer']),
help='The application type.')
c.argument('marketplace_identifier',
options_list=['--marketplace-id'],
help='The marketplace identifier.')
c.argument('https_endpoint_access_mode',
arg_group='HTTPS Endpoint',
options_list=['--access-mode'],
help='The access mode for the application.')
c.argument('https_endpoint_destination_port',
arg_group='HTTPS Endpoint',
options_list=['--destination-port'],
help='The destination port to connect to.')
c.argument('sub_domain_suffix',
arg_group='HTTPS Endpoint',
help='The subdomain suffix of the application.')
c.argument(
'disable_gateway_auth',
arg_group='HTTPS Endpoint',
arg_type=get_three_state_flag(),
help='Indicates whether to disable gateway authentication. '
'Default is to enable gateway authentication. Default: false. ')
c.argument('tags', tags_type)
c.argument('ssh_password',
options_list=['--ssh-password', '-P'],
arg_group='SSH',
help='SSH password for the cluster nodes.')
# script action
with self.argument_context('hdinsight script-action') as c:
c.argument('cluster_name',
options_list=['--cluster-name'],
completer=get_resource_name_completion_list(
'Microsoft.HDInsight/clusters'),
help='The name of the cluster.')
c.argument(
'roles',
nargs='+',
completer=get_generic_completion_list(known_role_types),
help=
'A space-delimited list of roles (nodes) where the script will be executed. '
'Valid roles are {}.'.format(', '.join(known_role_types)))
c.argument('persist_on_success',
help='If the scripts needs to be persisted.')
c.argument('script_action_name',
arg_type=name_type,
arg_group=None,
help='The name of the script action.')
c.argument('script_uri', arg_group=None, help='The URI to the script.')
c.argument('script_parameters',
arg_group=None,
help='The parameters for the script.')
c.argument('script_execution_id',
options_list=['--execution-id'],
arg_group=None,
help='The script execution id')
with self.argument_context('hdinsight script-action delete') as c:
c.argument('script_name',
arg_type=name_type,
arg_group=None,
help='The name of the script.')
# Monitoring
with self.argument_context('hdinsight monitor') as c:
c.argument(
'workspace',
validator=validate_workspace,
completer=get_resource_name_completion_list_under_subscription(
'Microsoft.OperationalInsights/workspaces'),
help=
'The name, resource ID or workspace ID of Log Analytics workspace.'
)
c.argument('primary_key',
help='The certificate for the Log Analytics workspace. '
'Required when workspace ID is provided.')
c.ignore('workspace_type')
# VPN connection shared key
register_cli_argument('network vpn-connection shared-key', 'connection_shared_key_name', options_list=('--name', '-n'), id_part='name')
register_cli_argument('network vpn-connection shared-key', 'virtual_network_gateway_connection_name', options_list=('--connection-name',), metavar='NAME', id_part='name')
# Traffic manager profiles
register_cli_argument('network traffic-manager profile', 'traffic_manager_profile_name', name_arg_type, id_part='name', completer=get_resource_name_completion_list('Microsoft.Network/trafficManagerProfiles'))
register_cli_argument('network traffic-manager profile', 'profile_name', name_arg_type, id_part='name', completer=get_resource_name_completion_list('Microsoft.Network/trafficManagerProfiles'))
register_cli_argument('network traffic-manager profile create', 'routing_method', **enum_choice_list(routingMethod))
register_cli_argument('network traffic-manager profile check-dns', 'name', name_arg_type, help='DNS prefix to verify availability for.', required=True)
register_cli_argument('network traffic-manager profile check-dns', 'type', help=argparse.SUPPRESS, default='Microsoft.Network/trafficManagerProfiles')
# Traffic manager endpoints
endpoint_types = ['azureEndpoints', 'externalEndpoints', 'nestedEndpoints']
register_cli_argument('network traffic-manager endpoint', 'endpoint_name', name_arg_type, id_part='child_name', help='Endpoint name.', completer=get_tm_endpoint_completion_list())
register_cli_argument('network traffic-manager endpoint', 'endpoint_type', options_list=('--type',), help='Endpoint type. Values include: {}.'.format(', '.join(endpoint_types)), completer=get_generic_completion_list(endpoint_types))
register_cli_argument('network traffic-manager endpoint', 'profile_name', help='Name of parent profile.', completer=get_resource_name_completion_list('Microsoft.Network/trafficManagerProfiles'), id_part='name')
# DNS
register_cli_argument('network dns zone', 'zone_name', name_arg_type)
register_cli_argument('network dns', 'record_set_name', name_arg_type, help='The name of the RecordSet, relative to the name of the zone.')
register_cli_argument('network dns', 'relative_record_set_name', name_arg_type, help='The name of the RecordSet, relative to the name of the zone.')
register_cli_argument('network dns', 'zone_name', help='The name of the zone without a terminating dot.')
register_cli_argument('network dns record-set create', 'record_set_type', options_list=('--type',), **enum_choice_list(RecordType))
register_cli_argument('network dns record-set create', 'ttl', default=3600)
register_cli_argument('network dns', 'record_type', options_list=('--type',), **enum_choice_list(RecordType))
register_cli_argument('network dns record', 'record_set_name', options_list=('--record-set-name',))
register_cli_argument('network dns record txt add', 'value', nargs='+')
register_cli_argument('network dns record txt remove', 'value', nargs='+')
register_cli_argument('network dns zone import', 'file_name', help='Path to the DNS zone file to import')
register_cli_argument('network dns zone export', 'file_name', help='Path to the DNS zone file to save')
def load_arguments(self, _):
from ._completers import storage_account_completion_list, storage_account_key_completion_list
from knack.arguments import CLIArgumentType
node_size_type = CLIArgumentType(arg_group='Node',
help='The size of the node. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-hadoop-provision-linux-clusters#configure-cluster-size')
with self.argument_context('hdinsight') as c:
c.argument('cluster_name', arg_type=name_type,
completer=get_resource_name_completion_list('Microsoft.HDInsight/clusters'),
help='The name of the cluster.')
c.argument('tags', tags_type)
c.argument('cluster_version', options_list=['--version', '-v'], arg_group='Cluster',
help='The HDInsight cluster version. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-component-versioning#supported-hdinsight-versions')
c.argument('cluster_type', options_list=['--type', '-t'], arg_group='Cluster',
completer=get_generic_completion_list(known_cluster_types),
help='Type of HDInsight cluster, like: {}. '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight-'
'hadoop-provision-linux-clusters#cluster-types'.format(', '.join(known_cluster_types)))
c.argument('component_version', arg_group='Cluster', nargs='*', validator=validate_component_version,
help='The versions of various Hadoop components, in space-'
'separated versions in \'component=version\' format. Example: '
'Spark=2.0 Hadoop=2.7.3 '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight'
'-component-versioning#hadoop-components-available-with-different-'
'hdinsight-versions')
c.argument('cluster_configurations', arg_group='Cluster',
help='Extra configurations of various components, in JSON.')
c.argument('cluster_tier', arg_type=get_enum_type(['standard', 'premium']), arg_group='Cluster',
help='The tier of the cluster: standard or premium.')
c.argument('http_username', options_list=['--http-user', '-u'], arg_group='HTTP',
help='HTTP username for the cluster. Default: admin.')
c.argument('http_password', options_list=['--http-password', '-p'], arg_group='HTTP',
help='HTTP password for the cluster.')
c.argument('ssh_username', options_list=['--ssh-user', '-U'], arg_group='SSH',
help='SSH username for the cluster nodes.')
c.argument('ssh_password', options_list=['--ssh-password', '-P'], arg_group='SSH',
help='SSH password for the cluster nodes. If none specified, uses the HTTP password.')
c.argument('ssh_public_key', options_list=['--ssh-public-key', '-K'], arg_group='SSH',
help='SSH public key for the cluster nodes.')
c.argument('headnode_size', arg_type=node_size_type)
c.argument('workernode_size', arg_type=node_size_type)
c.argument('workernode_data_disks_per_node', arg_group='Node',
help='The number of data disks to use per worker node.')
c.argument('workernode_data_disk_storage_account_type', arg_group='Node',
arg_type=get_enum_type(['standard_lrs', 'premium_lrs']),
help='The type of storage account that will be used for the data disks: standard_lrs or premium_lrs')
c.argument('workernode_data_disk_size', arg_group='Node',
help='The size of the data disk in GB, e.g. 1023.')
c.argument('zookeepernode_size', arg_type=node_size_type)
c.argument('edgenode_size', arg_type=node_size_type)
c.argument('workernode_count', options_list=['--size', '-s'], arg_group='Cluster',
help='The number of worker nodes in the cluster.')
c.argument('storage_account', arg_group='Storage', completer=storage_account_completion_list,
help='The storage account, e.g. "<name>.blob.core.windows.net".')
c.argument('storage_account_key', arg_group='Storage', completer=storage_account_key_completion_list,
help='The storage account key. A key can be retrieved automatically '
'if the user has access to the storage account.')
c.argument('storage_default_container', arg_group='Storage',
help='The storage container the cluster will use. '
'Uses the cluster name if none was specified. (WASB only)')
c.argument('storage_default_filesystem', arg_group='Storage',
help='The storage filesystem the cluster will use. (DFS only)')
c.argument('virtual_network', arg_group='Network',
help='The virtual network resource ID of an existing virtual network.')
c.argument('subnet_name', arg_group='Network',
help='The name of the subnet in the specified virtual network.')
def load_arguments(self, _):
from ._completers import storage_account_completion_list, storage_account_key_completion_list
from knack.arguments import CLIArgumentType
node_size_type = CLIArgumentType(arg_group='Node',
help='The size of the node. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-hadoop-provision-linux-clusters#configure-cluster-size')
# cluster
with self.argument_context('hdinsight') as c:
c.argument('cluster_name', arg_type=name_type,
completer=get_resource_name_completion_list('Microsoft.HDInsight/clusters'),
help='The name of the cluster.')
c.argument('tags', tags_type)
c.argument('cluster_version', options_list=['--version', '-v'], arg_group='Cluster',
help='The HDInsight cluster version. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-component-versioning#supported-hdinsight-versions')
c.argument('cluster_type', options_list=['--type', '-t'], arg_group='Cluster',
completer=get_generic_completion_list(known_cluster_types),
help='Type of HDInsight cluster, like: {}. '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight-'
'hadoop-provision-linux-clusters#cluster-types'.format(', '.join(known_cluster_types)))
c.argument('component_version', arg_group='Cluster', nargs='*', validator=validate_component_version,
help='The versions of various Hadoop components, in space-'
'separated versions in \'component=version\' format. Example: '
'Spark=2.0 Hadoop=2.7.3 '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight'
'-component-versioning#hadoop-components-available-with-different-'
'hdinsight-versions')
c.argument('cluster_configurations', arg_group='Cluster',
help='Extra configurations of various components, in JSON.')
c.argument('cluster_tier', arg_type=get_enum_type(['standard', 'premium']), arg_group='Cluster',
help='The tier of the cluster: standard or premium.')
c.argument('http_username', options_list=['--http-user', '-u'], arg_group='HTTP',
help='HTTP username for the cluster. Default: admin.')
c.argument('http_password', options_list=['--http-password', '-p'], arg_group='HTTP',
help='HTTP password for the cluster.')
c.argument('ssh_username', options_list=['--ssh-user', '-U'], arg_group='SSH',
help='SSH username for the cluster nodes.')
c.argument('ssh_password', options_list=['--ssh-password', '-P'], arg_group='SSH',
help='SSH password for the cluster nodes. If none specified, uses the HTTP password.')
c.argument('ssh_public_key', options_list=['--ssh-public-key', '-K'], arg_group='SSH',
help='SSH public key for the cluster nodes.')
c.argument('headnode_size', arg_type=node_size_type)
c.argument('workernode_size', arg_type=node_size_type)
c.argument('workernode_data_disks_per_node', arg_group='Node',
help='The number of data disks to use per worker node.')
c.argument('workernode_data_disk_storage_account_type', arg_group='Node',
arg_type=get_enum_type(['standard_lrs', 'premium_lrs']),
help='The type of storage account that will be used for the data disks: standard_lrs or premium_lrs')
c.argument('workernode_data_disk_size', arg_group='Node',
help='The size of the data disk in GB, e.g. 1023.')
c.argument('zookeepernode_size', arg_type=node_size_type)
c.argument('edgenode_size', arg_type=node_size_type)
c.argument('workernode_count', options_list=['--size', '-s'], arg_group='Cluster',
help='The number of worker nodes in the cluster.')
c.argument('storage_account', arg_group='Storage', completer=storage_account_completion_list,
help='The storage account, e.g. "<name>.blob.core.windows.net".')
c.argument('storage_account_key', arg_group='Storage', completer=storage_account_key_completion_list,
help='The storage account key. A key can be retrieved automatically '
'if the user has access to the storage account.')
c.argument('storage_default_container', arg_group='Storage',
help='The storage container the cluster will use. '
'Uses the cluster name if none was specified. (WASB only)')
c.argument('storage_default_filesystem', arg_group='Storage',
help='The storage filesystem the cluster will use. (DFS only)')
c.argument('virtual_network', arg_group='Network',
help='The virtual network resource ID of an existing virtual network.')
c.argument('subnet_name', arg_group='Network',
help='The name of the subnet in the specified virtual network.')
c.argument('script_action_name', arg_group='Script Action', help='The name of the script action.')
c.argument('script_uri', arg_group='Script Action', help='The URI to the script.')
c.argument('script_parameters', arg_group='Script Action', help='The parameters for the script.')
# application
with self.argument_context('hdinsight application') as c:
c.argument('application_name', arg_group='Application', help='The constant value for the application name.')
c.argument('application_type', arg_group='Application', help='The application type.')
c.argument('marketplace_identifier', arg_group='Application', help='The marketplace identifier.')
c.argument('https_endpoint_access_mode', arg_group='HTTPS Endpoint',
help='The access mode for the application.')
c.argument('https_endpoint_destination_port', arg_group='HTTPS Endpoint',
help='The destination port to connect to.')
c.argument('https_endpoint_location', arg_group='HTTPS Endpoint', help='The location of the endpoint.')
c.argument('https_endpoint_public_port', arg_group='HTTPS Endpoint', help='The public port to connect to.')
c.argument('ssh_endpoint_destination_port', arg_group='SSH Endpoint',
help='The destination port to connect to.')
c.argument('ssh_endpoint_location', arg_group='SSH Endpoint', help='The location of the endpoint.')
c.argument('ssh_endpoint_public_port', arg_group='SSH Endpoint', help='The public port to connect to.')
c.argument('tags', tags_type)
c.argument('ssh_password', options_list=['--ssh-password', '-P'], arg_group='SSH',
help='SSH password for the cluster nodes.')
# script action
with self.argument_context('hdinsight script-action') as c:
c.argument('roles', arg_group='Script Action',
completer=get_generic_completion_list(known_role_types),
help='A comma-delimited list of roles (nodes) where the script will be executed. '
'Valid roles are {}.'.format(', '.join(known_role_types)))
c.argument('persist_on_success', arg_group='Script Action', help='If the scripts needs to be persisted.')
c.argument('persisted', arg_group='Script Action', help='If only list persisted script actions.')
c.argument('script_name', options_list='--script-action-name', arg_group='Script Action',
help='The name of the script action.')
# OMS
with self.argument_context('hdinsight oms') as c:
c.argument('workspace_id', arg_group='OMS', help='The Operations Management Suite (OMS) workspace ID.')
c.argument('primary_key', arg_group='OMS', help='The Operations Management Suite (OMS) workspace key.')
# VPN connection shared key
register_cli_argument('network vpn-connection shared-key', 'connection_shared_key_name', options_list=('--name', '-n'), id_part='name')
register_cli_argument('network vpn-connection shared-key', 'virtual_network_gateway_connection_name', options_list=('--connection-name',), metavar='NAME', id_part='name')
# Traffic manager profiles
register_cli_argument('network traffic-manager profile', 'traffic_manager_profile_name', name_arg_type, id_part='name', completer=get_resource_name_completion_list('Microsoft.Network/trafficManagerProfiles'))
register_cli_argument('network traffic-manager profile', 'profile_name', name_arg_type, id_part='name', completer=get_resource_name_completion_list('Microsoft.Network/trafficManagerProfiles'))
register_cli_argument('network traffic-manager profile create', 'routing_method', **enum_choice_list(routingMethod))
register_cli_argument('network traffic-manager profile check-dns', 'name', name_arg_type, help='DNS prefix to verify availability for.', required=True)
register_cli_argument('network traffic-manager profile check-dns', 'type', help=argparse.SUPPRESS, default='Microsoft.Network/trafficManagerProfiles')
# Traffic manager endpoints
endpoint_types = ['azureEndpoints', 'externalEndpoints', 'nestedEndpoints']
register_cli_argument('network traffic-manager endpoint', 'endpoint_name', name_arg_type, id_part='child_name', help='Endpoint name.', completer=get_tm_endpoint_completion_list())
register_cli_argument('network traffic-manager endpoint', 'endpoint_type', options_list=('--type',), help='Endpoint type. Values include: {}.'.format(', '.join(endpoint_types)), completer=get_generic_completion_list(endpoint_types))
register_cli_argument('network traffic-manager endpoint', 'profile_name', help='Name of parent profile.', completer=get_resource_name_completion_list('Microsoft.Network/trafficManagerProfiles'), id_part='name')
# DNS
register_cli_argument('network dns zone', 'zone_name', name_arg_type)
register_cli_argument('network dns', 'record_set_name', name_arg_type, help='The name of the RecordSet, relative to the name of the zone.')
register_cli_argument('network dns', 'relative_record_set_name', name_arg_type, help='The name of the RecordSet, relative to the name of the zone.')
register_cli_argument('network dns', 'zone_name', help='The name of the zone without a terminating dot.')
register_cli_argument('network dns record-set create', 'record_set_type', options_list=('--type',), **enum_choice_list(RecordType))
register_cli_argument('network dns record-set create', 'ttl', default=3600)
register_cli_argument('network dns', 'record_type', options_list=('--type',), **enum_choice_list(RecordType))
register_cli_argument('network dns record', 'record_set_name', options_list=('--record-set-name',))
register_cli_argument('network dns record txt add', 'value', nargs='+')
register_cli_argument('network dns record txt remove', 'value', nargs='+')
register_cli_argument('network dns zone import', 'file_name', help='Path to the DNS zone file to import')
register_cli_argument('network dns zone export', 'file_name', help='Path to the DNS zone file to save')
def load_arguments(self, _):
from ._completers import subnet_completion_list
from knack.arguments import CLIArgumentType
from azure.mgmt.hdinsight.models import Tier, JsonWebKeyEncryptionAlgorithm
node_size_type = CLIArgumentType(arg_group='Node',
help='The size of the node. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-hadoop-provision-linux-clusters#configure-cluster-size')
# cluster
with self.argument_context('hdinsight') as c:
# Cluster
c.argument('cluster_name', arg_type=name_type,
completer=get_resource_name_completion_list('Microsoft.HDInsight/clusters'),
help='The name of the cluster.')
c.argument('tags', tags_type)
c.argument('cluster_version', options_list=['--version', '-v'], arg_group='Cluster',
help='The HDInsight cluster version. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-component-versioning#supported-hdinsight-versions')
c.argument('cluster_type', options_list=['--type', '-t'], arg_group='Cluster',
completer=get_generic_completion_list(known_cluster_types),
help='Type of HDInsight cluster, like: {}. '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight-'
'hadoop-provision-linux-clusters#cluster-types'.format(', '.join(known_cluster_types)))
c.argument('component_version', arg_group='Cluster', nargs='*', validator=validate_component_version,
help='The versions of various Hadoop components, in space-'
'separated versions in \'component=version\' format. Example: '
'Spark=2.0 Hadoop=2.7.3 '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight'
'-component-versioning#hadoop-components-available-with-different-'
'hdinsight-versions')
c.argument('cluster_configurations', arg_group='Cluster',
help='Extra configurations of various components, in JSON.')
c.argument('cluster_tier', arg_type=get_enum_type(Tier), arg_group='Cluster',
help='The tier of the cluster')
c.argument('esp', arg_group='Cluster', arg_type=get_three_state_flag(),
help='Specify to create cluster with Enterprise Security Package')
# HTTP
c.argument('http_username', options_list=['--http-user', '-u'], arg_group='HTTP',
help='HTTP username for the cluster. Default: admin.')
c.argument('http_password', options_list=['--http-password', '-p'], arg_group='HTTP',
help='HTTP password for the cluster.')
# SSH
c.argument('ssh_username', options_list=['--ssh-user', '-U'], arg_group='SSH',
help='SSH username for the cluster nodes.')
c.argument('ssh_password', options_list=['--ssh-password', '-P'], arg_group='SSH',
help='SSH password for the cluster nodes. If none specified, uses the HTTP password.')
c.argument('ssh_public_key', options_list=['--ssh-public-key', '-K'], arg_group='SSH',
help='SSH public key for the cluster nodes.')
# Node
c.argument('headnode_size', arg_type=node_size_type)
c.argument('workernode_size', arg_type=node_size_type)
c.argument('workernode_data_disks_per_node', arg_group='Node',
help='The number of data disks to use per worker node.')
c.argument('workernode_data_disk_storage_account_type', arg_group='Node',
arg_type=get_enum_type(['standard_lrs', 'premium_lrs']),
help='The type of storage account that will be used for the data disks: standard_lrs or premium_lrs')
c.argument('workernode_data_disk_size', arg_group='Node',
help='The size of the data disk in GB, e.g. 1023.')
c.argument('zookeepernode_size', arg_type=node_size_type)
c.argument('edgenode_size', arg_type=node_size_type)
c.argument('workernode_count', options_list=['--size', '-s'], arg_group='Node',
help='The number of worker nodes in the cluster.')
# Storage
c.argument('storage_account', arg_group='Storage', validator=validate_storage_account,
completer=get_resource_name_completion_list('Microsoft.Storage/storageAccounts'),
help='The name or ID of the storage account.')
c.argument('storage_account_key', arg_group='Storage',
help='The storage account key. A key can be retrieved automatically '
'if the user has access to the storage account.')
c.argument('storage_default_container', arg_group='Storage',
help='The storage container the cluster will use. '
'Uses the cluster name if none was specified. (WASB only)')
c.argument('storage_default_filesystem', arg_group='Storage',
help='The storage filesystem the cluster will use. (DFS only)')
c.argument('storage_account_managed_identity', arg_group='Storage', validator=validate_storage_msi,
completer=get_resource_name_completion_list('Microsoft.ManagedIdentity/userAssignedIdentities'),
help='User-assigned managed identity with access to the storage account filesystem. '
'Only required when storage account type is Azure Data Lake Storage Gen2.')
# Network
c.argument('vnet_name', arg_group='Network', validator=validate_subnet,
completer=get_resource_name_completion_list('Microsoft.Network/virtualNetworks'),
help='The name of a virtual network.')
c.argument('subnet', arg_group='Network',
completer=subnet_completion_list,
help='The name or ID of subnet. If name is supplied, `--vnet-name` must be supplied.')
# Script Action
c.argument('script_action_name', arg_group='Script Action', help='The name of the script action.')
c.argument('script_uri', arg_group='Script Action', help='The URI to the script.')
c.argument('script_parameters', arg_group='Script Action', help='The parameters for the script.')
# Domain Service
c.argument('domain', arg_group='Domain Service', validator=validate_domain_service,
help='The name or resource ID of the user\'s Azure Active Directory Domain Service. '
'Required only when create cluster with Enterprise Security Package.')
c.argument('cluster_users_group_dns', arg_group='Domain Service', nargs='+',
help='A space-delimited list of Distinguished Names for cluster user groups. '
'Required only when create cluster with Enterprise Security Package. ')
c.argument('cluster_admin_password', arg_group='Domain Service',
help='The domain admin password. '
'Required only when create cluster with Enterprise Security Package.')
c.argument('cluster_admin_account', arg_group='Domain Service',
help='The domain user account that will have admin privileges on the cluster. '
'Required only when create cluster with Enterprise Security Package.')
c.argument('ldaps_urls', arg_group='Domain Service', nargs='+',
help='A space-delimited list of LDAPS protocol URLs to communicate with the Active Directory. '
'Required only when create cluster with Enterprise Security Package.')
# Customer Managed Key
c.argument('encryption_vault_uri', arg_group='Customer Managed Key',
help='Base key vault URI where the customers key is located eg. https://myvault.vault.azure.net')
c.argument('encryption_key_name', arg_group='Customer Managed Key',
help='Key name that is used for enabling disk encryption.')
c.argument('encryption_key_version', arg_group='Customer Managed Key',
help='Key version that is used for enabling disk encryption.')
c.argument('encryption_algorithm', arg_type=get_enum_type(JsonWebKeyEncryptionAlgorithm),
arg_group='Customer Managed Key', help='Algorithm identifier for encryption.')
# Managed Service Identity
c.argument('assign_identity', arg_group='Managed Service Identity', validator=validate_msi,
help="The name or ID of user assigned identity.")
# application
with self.argument_context('hdinsight application') as c:
c.argument('application_name', arg_group='Application', help='The constant value for the application name.')
c.argument('application_type', arg_group='Application', help='The application type.')
c.argument('marketplace_identifier', arg_group='Application', help='The marketplace identifier.')
c.argument('https_endpoint_access_mode', arg_group='HTTPS Endpoint',
help='The access mode for the application.')
c.argument('https_endpoint_destination_port', arg_group='HTTPS Endpoint',
help='The destination port to connect to.')
c.argument('https_endpoint_location', arg_group='HTTPS Endpoint', help='The location of the endpoint.')
c.argument('https_endpoint_public_port', arg_group='HTTPS Endpoint', help='The public port to connect to.')
c.argument('ssh_endpoint_destination_port', arg_group='SSH Endpoint',
help='The destination port to connect to.')
c.argument('ssh_endpoint_location', arg_group='SSH Endpoint', help='The location of the endpoint.')
c.argument('ssh_endpoint_public_port', arg_group='SSH Endpoint', help='The public port to connect to.')
c.argument('tags', tags_type)
c.argument('ssh_password', options_list=['--ssh-password', '-P'], arg_group='SSH',
help='SSH password for the cluster nodes.')
# script action
with self.argument_context('hdinsight script-action') as c:
c.argument('roles', arg_group='Script Action',
completer=get_generic_completion_list(known_role_types),
help='A comma-delimited list of roles (nodes) where the script will be executed. '
'Valid roles are {}.'.format(', '.join(known_role_types)))
c.argument('persist_on_success', arg_group='Script Action', help='If the scripts needs to be persisted.')
c.argument('persisted', arg_group='Script Action', help='If only list persisted script actions.')
c.argument('script_name', options_list='--script-action-name', arg_group='Script Action',
help='The name of the script action.')
# OMS
with self.argument_context('hdinsight oms') as c:
c.argument('workspace_id', arg_group='OMS', help='The Operations Management Suite (OMS) workspace ID.')
c.argument('primary_key', arg_group='OMS', help='The Operations Management Suite (OMS) workspace key.')
def load_arguments(self, _):
from ._completers import subnet_completion_list
from knack.arguments import CLIArgumentType
from azure.mgmt.hdinsight.models import Tier, JsonWebKeyEncryptionAlgorithm
node_size_type = CLIArgumentType(
arg_group='Node',
help=
'The size of the node. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-hadoop-provision-linux-clusters#configure-cluster-size'
)
# cluster
with self.argument_context('hdinsight') as c:
# Cluster
c.argument('cluster_name',
arg_type=name_type,
completer=get_resource_name_completion_list(
'Microsoft.HDInsight/clusters'),
help='The name of the cluster.')
c.argument('tags', tags_type)
c.argument(
'cluster_version',
options_list=['--version', '-v'],
arg_group='Cluster',
help=
'The HDInsight cluster version. See also: https://docs.microsoft.com/en-us/azure/'
'hdinsight/hdinsight-component-versioning#supported-hdinsight-versions'
)
c.argument(
'cluster_type',
options_list=['--type', '-t'],
arg_group='Cluster',
completer=get_generic_completion_list(known_cluster_types),
help='Type of HDInsight cluster, like: {}. '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight-'
'hadoop-provision-linux-clusters#cluster-types'.format(
', '.join(known_cluster_types)))
c.argument(
'component_version',
arg_group='Cluster',
nargs='*',
validator=validate_component_version,
help='The versions of various Hadoop components, in space-'
'separated versions in \'component=version\' format. Example: '
'Spark=2.0 Hadoop=2.7.3 '
'See also: https://docs.microsoft.com/en-us/azure/hdinsight/hdinsight'
'-component-versioning#hadoop-components-available-with-different-'
'hdinsight-versions')
c.argument('cluster_configurations',
arg_group='Cluster',
help='Extra configurations of various components, in JSON.')
c.argument('cluster_tier',
arg_type=get_enum_type(Tier),
arg_group='Cluster',
help='The tier of the cluster')
c.argument(
'esp',
arg_group='Cluster',
arg_type=get_three_state_flag(),
help='Specify to create cluster with Enterprise Security Package')
# HTTP
c.argument('http_username',
options_list=['--http-user', '-u'],
arg_group='HTTP',
help='HTTP username for the cluster. Default: admin.')
c.argument('http_password',
options_list=['--http-password', '-p'],
arg_group='HTTP',
help='HTTP password for the cluster.')
# SSH
c.argument('ssh_username',
options_list=['--ssh-user', '-U'],
arg_group='SSH',
help='SSH username for the cluster nodes.')
c.argument(
'ssh_password',
options_list=['--ssh-password', '-P'],
arg_group='SSH',
help=
'SSH password for the cluster nodes. If none specified, uses the HTTP password.'
)
c.argument('ssh_public_key',
options_list=['--ssh-public-key', '-K'],
arg_group='SSH',
help='SSH public key for the cluster nodes.')
# Node
c.argument('headnode_size', arg_type=node_size_type)
c.argument('workernode_size', arg_type=node_size_type)
c.argument('workernode_data_disks_per_node',
arg_group='Node',
help='The number of data disks to use per worker node.')
c.argument(
'workernode_data_disk_storage_account_type',
arg_group='Node',
arg_type=get_enum_type(['standard_lrs', 'premium_lrs']),
help=
'The type of storage account that will be used for the data disks: standard_lrs or premium_lrs'
)
c.argument('workernode_data_disk_size',
arg_group='Node',
help='The size of the data disk in GB, e.g. 1023.')
c.argument('zookeepernode_size', arg_type=node_size_type)
c.argument('edgenode_size', arg_type=node_size_type)
c.argument('workernode_count',
options_list=['--size', '-s'],
arg_group='Node',
help='The number of worker nodes in the cluster.')
# Storage
c.argument('storage_account',
arg_group='Storage',
validator=validate_storage_account,
completer=get_resource_name_completion_list(
'Microsoft.Storage/storageAccounts'),
help='The name or ID of the storage account.')
c.argument(
'storage_account_key',
arg_group='Storage',
help='The storage account key. A key can be retrieved automatically '
'if the user has access to the storage account.')
c.argument('storage_default_container',
arg_group='Storage',
help='The storage container the cluster will use. '
'Uses the cluster name if none was specified. (WASB only)')
c.argument(
'storage_default_filesystem',
arg_group='Storage',
help='The storage filesystem the cluster will use. (DFS only)')
# Network
c.argument('vnet_name',
arg_group='Network',
validator=validate_subnet,
completer=get_resource_name_completion_list(
'Microsoft.Network/virtualNetworks'),
help='The name of a virtual network.')
c.argument(
'subnet',
arg_group='Network',
completer=subnet_completion_list,
help=
'The name or ID of subnet. If name is supplied, `--vnet-name` must be supplied.'
)
# Script Action
c.argument('script_action_name',
arg_group='Script Action',
help='The name of the script action.')
c.argument('script_uri',
arg_group='Script Action',
help='The URI to the script.')
c.argument('script_parameters',
arg_group='Script Action',
help='The parameters for the script.')
# Domain Service
c.argument(
'domain',
arg_group='Domain Service',
validator=validate_domain_service,
help=
'The name or resource ID of the user\'s Azure Active Directory Domain Service. '
'Required only when create cluster with Enterprise Security Package.'
)
c.argument(
'cluster_users_group_dns',
arg_group='Domain Service',
nargs='+',
help=
'A space-delimited list of Distinguished Names for cluster user groups. '
'Required only when create cluster with Enterprise Security Package. '
)
c.argument(
'cluster_admin_password',
arg_group='Domain Service',
help='The domain admin password. '
'Required only when create cluster with Enterprise Security Package.'
)
c.argument(
'cluster_admin_account',
arg_group='Domain Service',
help=
'The domain user account that will have admin privileges on the cluster. '
'Required only when create cluster with Enterprise Security Package.'
)
c.argument(
'ldaps_urls',
arg_group='Domain Service',
nargs='+',
help=
'A space-delimited list of LDAPS protocol URLs to communicate with the Active Directory. '
'Required only when create cluster with Enterprise Security Package.'
)
# Customer Managed Key
c.argument(
'encryption_vault_uri',
arg_group='Customer Managed Key',
help=
'Base key vault URI where the customers key is located eg. https://myvault.vault.azure.net'
)
c.argument('encryption_key_name',
arg_group='Customer Managed Key',
help='Key name that is used for enabling disk encryption.')
c.argument(
'encryption_key_version',
arg_group='Customer Managed Key',
help='Key version that is used for enabling disk encryption.')
c.argument('encryption_algorithm',
arg_type=get_enum_type(JsonWebKeyEncryptionAlgorithm),
arg_group='Customer Managed Key',
help='Algorithm identifier for encryption.')
# Managed Service Identity
c.argument('assign_identity',
arg_group='Managed Service Identity',
validator=validate_msi,
help="The name or ID of user assigned identity.")
# application
with self.argument_context('hdinsight application') as c:
c.argument('application_name',
arg_group='Application',
help='The constant value for the application name.')
c.argument('application_type',
arg_group='Application',
help='The application type.')
c.argument('marketplace_identifier',
arg_group='Application',
help='The marketplace identifier.')
c.argument('https_endpoint_access_mode',
arg_group='HTTPS Endpoint',
help='The access mode for the application.')
c.argument('https_endpoint_destination_port',
arg_group='HTTPS Endpoint',
help='The destination port to connect to.')
c.argument('https_endpoint_location',
arg_group='HTTPS Endpoint',
help='The location of the endpoint.')
c.argument('https_endpoint_public_port',
arg_group='HTTPS Endpoint',
help='The public port to connect to.')
c.argument('ssh_endpoint_destination_port',
arg_group='SSH Endpoint',
help='The destination port to connect to.')
c.argument('ssh_endpoint_location',
arg_group='SSH Endpoint',
help='The location of the endpoint.')
c.argument('ssh_endpoint_public_port',
arg_group='SSH Endpoint',
help='The public port to connect to.')
c.argument('tags', tags_type)
c.argument('ssh_password',
options_list=['--ssh-password', '-P'],
arg_group='SSH',
help='SSH password for the cluster nodes.')
# script action
with self.argument_context('hdinsight script-action') as c:
c.argument(
'roles',
arg_group='Script Action',
completer=get_generic_completion_list(known_role_types),
help=
'A comma-delimited list of roles (nodes) where the script will be executed. '
'Valid roles are {}.'.format(', '.join(known_role_types)))
c.argument('persist_on_success',
arg_group='Script Action',
help='If the scripts needs to be persisted.')
c.argument('persisted',
arg_group='Script Action',
help='If only list persisted script actions.')
c.argument('script_name',
options_list='--script-action-name',
arg_group='Script Action',
help='The name of the script action.')
# OMS
with self.argument_context('hdinsight oms') as c:
c.argument('workspace_id',
arg_group='OMS',
help='The Operations Management Suite (OMS) workspace ID.')
c.argument('primary_key',
arg_group='OMS',
help='The Operations Management Suite (OMS) workspace key.')
def load_arguments(self, _):
with self.argument_context('graph query') as c:
c.argument('graph_query', options_list=['--graph-query', '--q', '-q'], required=True, completer=get_generic_completion_list(_QUERY_EXAMPLES), help='Resource Graph query to execute.')
c.argument('first', options_list=['--first'], required=False, type=int, default=100, help='The maximum number of objects to return. Accepted range: 1-5000.')
c.argument('skip', options_list=['--skip'], required=False, type=int, default=0, help='Ignores the first N objects and then gets the remaining objects.')
c.argument('subscriptions', options_list=['--subscriptions', '-s'], nargs='*', required=False, default=None, help='List of subscriptions to run query against. By default all accessible subscriptions are queried.')
