示例#1
0
def _get_client(service, parsed_args):
    account_name = getattr(parsed_args, 'account_name', None) or az_config.get('storage', 'account', None)
    account_key = getattr(parsed_args, 'account_key', None) or az_config.get('storage', 'key', None)
    connection_string = getattr(parsed_args, 'connection_string', None) or az_config.get('storage', 'connection_string', None)
    sas_token = getattr(parsed_args, 'sas_token', None) or az_config.get('storage', 'sas_token', None)
    return get_storage_data_service_client(
        service, account_name, account_key, connection_string, sas_token)
示例#2
0
def add_azure_container_to_cluster_create_parameters(params, container_name, mount_path):
    """Add Azure Storage container to the cluster create parameters.

    :param model.ClusterCreateParameters params: cluster create parameters.
    :param str container_name: container name.
    :param str mount_path: relative mount path for the container.
    """
    if not mount_path:
        raise CLIError('Azure Storage container relative mount path cannot be empty.')
    if params.node_setup is None:
        params.node_setup = models.NodeSetup()
    if params.node_setup.mount_volumes is None:
        params.node_setup.mount_volumes = models.MountVolumes()
    if params.node_setup.mount_volumes.azure_blob_file_systems is None:
        params.node_setup.mount_volumes.azure_blob_file_systems = []
    storage_account_name = az_config.get('batchai', 'storage_account', fallback=None)
    if not storage_account_name:
        raise CLIError(MSG_CONFIGURE_STORAGE_ACCOUNT)
    storage_account_key = az_config.get('batchai', 'storage_key', fallback=None)
    if not storage_account_key:
        raise CLIError(MSG_CONFIGURE_STORAGE_KEY)
    params.node_setup.mount_volumes.azure_blob_file_systems.append(models.AzureBlobFileSystemReference(
        relative_mount_path=mount_path,
        account_name=storage_account_name,
        container_name=container_name,
        credentials=models.AzureStorageCredentialsInfo(account_key=storage_account_key)))
示例#3
0
def add_azure_file_share_to_cluster_create_parameters(params, azure_file_share, mount_path):
    """Add Azure File share to the cluster create parameters.

    :param model.ClusterCreateParameters params: cluster create parameters.
    :param str azure_file_share: name of the azure file share.
    :param str mount_path: relative mount path for Azure File share.
    """
    if not mount_path:
        raise CLIError('Azure File share relative mount path cannot be empty.')
    if params.node_setup is None:
        params.node_setup = models.NodeSetup()
    if params.node_setup.mount_volumes is None:
        params.node_setup.mount_volumes = models.MountVolumes()
    if params.node_setup.mount_volumes.azure_file_shares is None:
        params.node_setup.mount_volumes.azure_file_shares = []
    storage_account_name = az_config.get('batchai', 'storage_account', fallback=None)
    if not storage_account_name:
        raise CLIError(MSG_CONFIGURE_STORAGE_ACCOUNT)
    storage_account_key = az_config.get('batchai', 'storage_key', fallback=None)
    if not storage_account_key:
        raise CLIError(MSG_CONFIGURE_STORAGE_KEY)
    params.node_setup.mount_volumes.azure_file_shares.append(models.AzureFileShareReference(
        relative_mount_path=mount_path,
        account_name=storage_account_name,
        azure_file_url='https://{0}.file.core.windows.net/{1}'.format(storage_account_name, azure_file_share),
        credentials=models.AzureStorageCredentialsInfo(storage_account_key)))
示例#4
0
def validate_client_parameters(namespace):
    """ Retrieves storage connection parameters from environment variables and parses out
    connection string into account name and key """
    n = namespace

    if not n.connection_string:
        n.connection_string = az_config.get('storage', 'connection_string', None)

    # if connection string supplied or in environment variables, extract account key and name
    if n.connection_string:
        conn_dict = validate_key_value_pairs(n.connection_string)
        n.account_name = conn_dict['AccountName']
        n.account_key = conn_dict['AccountKey']

    # otherwise, simply try to retrieve the remaining variables from environment variables
    if not n.account_name:
        n.account_name = az_config.get('storage', 'account', None)
    if not n.account_key:
        n.account_key = az_config.get('storage', 'key', None)
    if not n.sas_token:
        n.sas_token = az_config.get('storage', 'sas_token', None)

    # if account name is specified but no key, attempt to query
    if n.account_name and not n.account_key:
        scf = get_mgmt_service_client(StorageManagementClient)
        acc = next((x for x in scf.storage_accounts.list() if x.name == n.account_name), None)
        if acc:
            from azure.cli.core.commands.arm import parse_resource_id
            rg = parse_resource_id(acc.id)['resource_group']
            n.account_key = \
                scf.storage_accounts.list_keys(rg, n.account_name).keys[0].value  # pylint: disable=no-member
        else:
            raise ValueError("Storage account '{}' not found.".format(n.account_name))
示例#5
0
def validate_client_parameters(namespace):
    """Retrieves Batch connection parameters from environment variables"""
    from azure.mgmt.batch import BatchManagementClient
    from azure.cli.core.commands.client_factory import get_mgmt_service_client
    from azure.cli.core._config import az_config

    # simply try to retrieve the remaining variables from environment variables
    if not namespace.account_name:
        namespace.account_name = az_config.get('batch', 'account', None)
    if not namespace.account_key:
        namespace.account_key = az_config.get('batch', 'access_key', None)
    if not namespace.account_endpoint:
        namespace.account_endpoint = az_config.get('batch', 'endpoint', None)

    # if account name is specified but no key, attempt to query
    if namespace.account_name and namespace.account_endpoint and not namespace.account_key:
        endpoint = urlsplit(namespace.account_endpoint)
        host = endpoint.netloc
        client = get_mgmt_service_client(BatchManagementClient)
        acc = next((x for x in client.batch_account.list()
                    if x.name == namespace.account_name and x.account_endpoint == host), None)
        if acc:
            from azure.cli.core.commands.arm import parse_resource_id
            rg = parse_resource_id(acc.id)['resource_group']
            namespace.account_key = \
                client.batch_account.get_keys(rg, namespace.account_name).primary  # pylint: disable=no-member
        else:
            raise ValueError("Batch account '{}' not found.".format(namespace.account_name))
    else:
        if not namespace.account_name:
            raise ValueError("Need specifiy batch account in command line or enviroment variable.")
        if not namespace.account_endpoint:
            raise ValueError("Need specifiy batch endpoint in command line or enviroment variable.")
示例#6
0
def add_azure_file_share_to_cluster_create_parameters(params, azure_file_share,
                                                      mount_path):
    """Add Azure File share to the cluster create parameters.

    :param model.ClusterCreateParameters params: cluster create parameters.
    :param str azure_file_share: name of the azure file share.
    :param str mount_path: relative mount path for Azure File share.
    """
    if not mount_path:
        raise CLIError('Azure File share relative mount path cannot be empty.')
    if params.node_setup is None:
        params.node_setup = models.NodeSetup()
    if params.node_setup.mount_volumes is None:
        params.node_setup.mount_volumes = models.MountVolumes()
    if params.node_setup.mount_volumes.azure_file_shares is None:
        params.node_setup.mount_volumes.azure_file_shares = []
    storage_account_name = az_config.get('batchai',
                                         'storage_account',
                                         fallback=None)
    if not storage_account_name:
        raise CLIError(MSG_CONFIGURE_STORAGE_ACCOUNT)
    storage_account_key = az_config.get('batchai',
                                        'storage_key',
                                        fallback=None)
    if not storage_account_key:
        raise CLIError(MSG_CONFIGURE_STORAGE_KEY)
    params.node_setup.mount_volumes.azure_file_shares.append(
        models.AzureFileShareReference(
            relative_mount_path=mount_path,
            account_name=storage_account_name,
            azure_file_url='https://{0}.file.core.windows.net/{1}'.format(
                storage_account_name, azure_file_share),
            credentials=models.AzureStorageCredentialsInfo(
                storage_account_key)))
示例#7
0
def _get_client(service, parsed_args):
    account_name = getattr(parsed_args, 'account_name', None) or az_config.get('storage', 'account', None)
    account_key = getattr(parsed_args, 'account_key', None) or az_config.get('storage', 'key', None)
    connection_string = getattr(parsed_args, 'connection_string', None) or az_config.get('storage', 'connection_string', None)
    sas_token = getattr(parsed_args, 'sas_token', None) or az_config.get('storage', 'sas_token', None)
    return get_storage_data_service_client(
        service, account_name, account_key, connection_string, sas_token)
示例#8
0
def _install_or_update(package_list, link, private, pre):
    import pip
    options = ['--isolated', '--disable-pip-version-check', '--upgrade']
    if pre:
        options.append('--pre')
    if _installed_in_user():
        options.append('--user')
    pkg_index_options = ['--find-links', link] if link else []
    if private:
        package_index_url = az_config.get('component',
                                          'package_index_url',
                                          fallback=None)
        package_index_trusted_host = az_config.get(
            'component', 'package_index_trusted_host', fallback=None)
        if package_index_url:
            pkg_index_options += ['--extra-index-url', package_index_url]
        else:
            raise CLIError(
                'AZURE_COMPONENT_PACKAGE_INDEX_URL environment variable not set and not '
                'specified in config. AZURE_COMPONENT_PACKAGE_INDEX_TRUSTED_HOST may '
                'also need to be set.\nIf executing az with sudo, you may want sudo\'s '
                '-E and -H flags.')
        pkg_index_options += ['--trusted-host', package_index_trusted_host
                              ] if package_index_trusted_host else []
    pip_args = ['install'] + options + package_list + pkg_index_options
    _run_pip(pip, pip_args)
    # Fix to make sure that we have empty __init__.py files for the azure site-packages folder.
    nspkg_pip_args = ['install'] + options + ['--force-reinstall', 'azure-nspkg', 'azure-mgmt-nspkg'] + pkg_index_options  # pylint: disable=line-too-long
    _run_pip(pip, nspkg_pip_args)
示例#9
0
def add_azure_container_to_cluster_create_parameters(params, container_name,
                                                     mount_path):
    """Add Azure Storage container to the cluster create parameters.

    :param model.ClusterCreateParameters params: cluster create parameters.
    :param str container_name: container name.
    :param str mount_path: relative mount path for the container.
    """
    if not mount_path:
        raise CLIError(
            'Azure Storage container relative mount path cannot be empty.')
    if params.node_setup is None:
        params.node_setup = models.NodeSetup()
    if params.node_setup.mount_volumes is None:
        params.node_setup.mount_volumes = models.MountVolumes()
    if params.node_setup.mount_volumes.azure_blob_file_systems is None:
        params.node_setup.mount_volumes.azure_blob_file_systems = []
    storage_account_name = az_config.get('batchai',
                                         'storage_account',
                                         fallback=None)
    if not storage_account_name:
        raise CLIError(MSG_CONFIGURE_STORAGE_ACCOUNT)
    storage_account_key = az_config.get('batchai',
                                        'storage_key',
                                        fallback=None)
    if not storage_account_key:
        raise CLIError(MSG_CONFIGURE_STORAGE_KEY)
    params.node_setup.mount_volumes.azure_blob_file_systems.append(
        models.AzureBlobFileSystemReference(
            relative_mount_path=mount_path,
            account_name=storage_account_name,
            container_name=container_name,
            credentials=models.AzureStorageCredentialsInfo(
                account_key=storage_account_key)))
示例#10
0
def validate_client_parameters(namespace):
    """ Retrieves storage connection parameters from environment variables and parses out
    connection string into account name and key """
    n = namespace

    if not n.connection_string:
        n.connection_string = az_config.get('storage', 'connection_string', None)

    # if connection string supplied or in environment variables, extract account key and name
    if n.connection_string:
        conn_dict = validate_key_value_pairs(n.connection_string)
        n.account_name = conn_dict['AccountName']
        n.account_key = conn_dict['AccountKey']

    # otherwise, simply try to retrieve the remaining variables from environment variables
    if not n.account_name:
        n.account_name = az_config.get('storage', 'account', None)
    if not n.account_key:
        n.account_key = az_config.get('storage', 'key', None)
    if not n.sas_token:
        n.sas_token = az_config.get('storage', 'sas_token', None)

    # strip the '?' from sas token. the portal and command line are returns sas token in different
    # forms
    if n.sas_token:
        n.sas_token = n.sas_token.lstrip('?')

    # if account name is specified but no key, attempt to query
    if n.account_name and not n.account_key and not n.sas_token:
        n.account_key = _query_account_key(n.account_name)
示例#11
0
def _install_or_update(package_list, link, private, pre):
    import pip
    options = [
        '--isolated', '--disable-pip-version-check', '--upgrade',
        '--ignore-installed'
    ]
    if pre:
        options.append('--pre')
    if _installed_in_user():
        options.append('--user')
    pkg_index_options = ['--find-links', link] if link else []
    if private:
        package_index_url = az_config.get('component',
                                          'package_index_url',
                                          fallback=None)
        package_index_trusted_host = az_config.get('component', 'package_index_trusted_host', fallback=None)  #pylint: disable=line-too-long
        if package_index_url:
            pkg_index_options += ['--extra-index-url', package_index_url]
        else:
            raise CLIError('AZURE_COMPONENT_PACKAGE_INDEX_URL environment variable not set and not specified in config. '  #pylint: disable=line-too-long
                           'AZURE_COMPONENT_PACKAGE_INDEX_TRUSTED_HOST may also need to be set.\n'
                           'If executing az with sudo, you may want sudo\'s -E and -H flags.')  #pylint: disable=line-too-long
        pkg_index_options += ['--trusted-host', package_index_trusted_host] if package_index_trusted_host else []  #pylint: disable=line-too-long
    pip_args = ['install'] + options + package_list + pkg_index_options
    _run_pip(pip, pip_args)
示例#12
0
def validate_client_parameters(namespace):
    """ Retrieves storage connection parameters from environment variables and parses out
    connection string into account name and key """
    n = namespace

    if not n.connection_string:
        n.connection_string = az_config.get('storage', 'connection_string', None)

    # if connection string supplied or in environment variables, extract account key and name
    if n.connection_string:
        conn_dict = validate_key_value_pairs(n.connection_string)
        n.account_name = conn_dict['AccountName']
        n.account_key = conn_dict['AccountKey']

    # otherwise, simply try to retrieve the remaining variables from environment variables
    if not n.account_name:
        n.account_name = az_config.get('storage', 'account', None)
    if not n.account_key:
        n.account_key = az_config.get('storage', 'key', None)
    if not n.sas_token:
        n.sas_token = az_config.get('storage', 'sas_token', None)

    # if account name is specified but no key, attempt to query
    if n.account_name and not n.account_key:
        scf = get_mgmt_service_client(StorageManagementClient)
        acc = next((x for x in scf.storage_accounts.list() if x.name == n.account_name), None)
        if acc:
            from azure.cli.core.commands.arm import parse_resource_id
            rg = parse_resource_id(acc.id)['resource_group']
            n.account_key = \
                scf.storage_accounts.list_keys(rg, n.account_name).keys[0].value  # pylint: disable=no-member
        else:
            raise ValueError("Storage account '{}' not found.".format(n.account_name))
示例#13
0
def _install_or_update(package_list, link, private, pre):
    import pip
    options = ['--isolated', '--disable-pip-version-check', '--upgrade']
    if pre:
        options.append('--pre')
    if _installed_in_user():
        options.append('--user')
    pkg_index_options = ['--find-links', link] if link else []
    if private:
        package_index_url = az_config.get('component', 'package_index_url', fallback=None)
        package_index_trusted_host = az_config.get('component', 'package_index_trusted_host',
                                                   fallback=None)
        if package_index_url:
            pkg_index_options += ['--extra-index-url', package_index_url]
        else:
            raise CLIError('AZURE_COMPONENT_PACKAGE_INDEX_URL environment variable not set and not '
                           'specified in config. AZURE_COMPONENT_PACKAGE_INDEX_TRUSTED_HOST may '
                           'also need to be set.\nIf executing az with sudo, you may want sudo\'s '
                           '-E and -H flags.')
        pkg_index_options += ['--trusted-host',
                              package_index_trusted_host] if package_index_trusted_host else []
    pip_args = ['install'] + options + package_list + pkg_index_options
    _run_pip(pip, pip_args)
    # Fix to make sure that we have empty __init__.py files for the azure site-packages folder.
    nspkg_pip_args = ['install'] + options + ['--force-reinstall', 'azure-nspkg', 'azure-mgmt-nspkg'] + pkg_index_options  # pylint: disable=line-too-long
    _run_pip(pip, nspkg_pip_args)
示例#14
0
def validate_client_parameters(namespace):
    """ Retrieves storage connection parameters from environment variables and parses out
    connection string into account name and key """
    n = namespace

    if not n.connection_string:
        n.connection_string = az_config.get('storage', 'connection_string',
                                            None)

    # if connection string supplied or in environment variables, extract account key and name
    if n.connection_string:
        conn_dict = validate_key_value_pairs(n.connection_string)
        n.account_name = conn_dict['AccountName']
        n.account_key = conn_dict['AccountKey']

    # otherwise, simply try to retrieve the remaining variables from environment variables
    if not n.account_name:
        n.account_name = az_config.get('storage', 'account', None)
    if not n.account_key:
        n.account_key = az_config.get('storage', 'key', None)
    if not n.sas_token:
        n.sas_token = az_config.get('storage', 'sas_token', None)

    # strip the '?' from sas token. the portal and command line are returns sas token in different
    # forms
    if n.sas_token:
        n.sas_token = n.sas_token.lstrip('?')

    # if account name is specified but no key, attempt to query
    if n.account_name and not n.account_key and not n.sas_token:
        n.account_key = _query_account_key(n.account_name)
示例#15
0
def update_cluster_create_parameters_with_env_variables(params):
    """Replaces placeholders with information from the environment variables.

    Currently we support replacing of storage account name and key in mount volumes.

    :param models.ClusterCreateParameters params: cluster creation parameters to patch.
    """
    storage_account_name = az_config.get(
        'batchai',
        'storage_account',
        fallback=AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER)
    storage_account_key = az_config.get(
        'batchai',
        'storage_key',
        fallback=AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER)
    # Patch parameters of azure file share.
    if params.node_setup and \
            params.node_setup.mount_volumes and \
            params.node_setup.mount_volumes.azure_file_shares:
        for ref in params.node_setup.mount_volumes.azure_file_shares:
            if ref.account_name == AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER:
                ref.account_name = storage_account_name
            if ref.azure_file_url and AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER in ref.azure_file_url:
                ref.azure_file_url = ref.azure_file_url.replace(
                    AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER,
                    storage_account_name)
            if ref.credentials_info and ref.credentials_info.account_key == AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER:
                ref.credentials_info.account_key = storage_account_key
            if not ref.credentials_info:
                ref.credentials_info = models.AzureStorageCredentialsInfo(
                    account_key=storage_account_key)

            # Verify that all placeholders are replaced with real values.
            if (ref.account_name == AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER
                    or AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER
                    in ref.azure_file_url):
                raise CLIError(MSG_CONFIGURE_STORAGE_ACCOUNT)
            if ref.credentials_info.account_key == AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER:
                raise CLIError(MSG_CONFIGURE_STORAGE_KEY)

    # Patch parameters of blob file system.
    if params.node_setup and \
            params.node_setup.mount_volumes and \
            params.node_setup.mount_volumes.azure_blob_file_systems:
        for ref in params.node_setup.mount_volumes.azure_blob_file_systems:
            if ref.account_name == AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER:
                ref.account_name = storage_account_name
            if ref.credentials_info and ref.credentials_info.account_key == \
                    AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER:
                ref.credentials_info.account_key = storage_account_key
            if not ref.credentials_info:
                ref.credentials_info = models.AzureStorageCredentialsInfo(
                    account_key=storage_account_key)
            # Verify that all placeholders are replaced with real values.
            if ref.account_name == AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER:
                raise CLIError(MSG_CONFIGURE_STORAGE_ACCOUNT)
            if ref.credentials_info.account_key == AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER:
                raise CLIError(MSG_CONFIGURE_STORAGE_KEY)
示例#16
0
def get_effective_storage_account_name_and_key(account_name, account_key):
    """Returns storage account name and key to be used.

    :param str or None account_name: storage account name provided as command line argument.
    :param str or None account_key: storage account key provided as command line argument.
    """
    if account_name:
        return account_name, get_storage_account_key(account_name, account_key) or ''
    return az_config.get('batchai', 'storage_account', ''), az_config.get('batchai', 'storage_key', '')
示例#17
0
 def __init__(self, client, account_name, resource_group_name, account_endpoint):
     self.resource_file_cache = {}
     self.resolved_storage_client = None
     self.batch_mgmt_client = client
     self.batch_account_name = account_name
     if not self.batch_account_name:
         self.batch_account_name = az_config.get('batch', 'account', None)
     self.batch_account_endpoint = account_endpoint
     if not self.batch_account_endpoint:
         self.batch_account_endpoint = az_config.get('batch', 'endpoint', None)
     self.batch_resource_group = resource_group_name
示例#18
0
def _get_client(service, parsed_args):
    account_name = parsed_args.account_name or az_config.get(
        'storage', 'account', None)
    account_key = parsed_args.account_key or az_config.get(
        'storage', 'key', None)
    connection_string = parsed_args.connection_string or az_config.get(
        'storage', 'connection_string', None)
    sas_token = parsed_args.sas_token or az_config.get('storage', 'sas_token',
                                                       None)
    return get_data_service_client(service, account_name, account_key,
                                   connection_string, sas_token)
示例#19
0
def update_cluster_create_parameters_with_env_variables(params):
    """Replaces placeholders with information from the environment variables.

    Currently we support replacing of storage account name and key in mount volumes.

    :param models.ClusterCreateParameters params: cluster creation parameters to patch.
    """
    storage_account_name = az_config.get('batchai', 'storage_account',
                                         fallback=AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER)
    storage_account_key = az_config.get('batchai', 'storage_key', fallback=AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER)
    # Patch parameters of azure file share.
    if params.node_setup and \
            params.node_setup.mount_volumes and \
            params.node_setup.mount_volumes.azure_file_shares:
        for ref in params.node_setup.mount_volumes.azure_file_shares:
            if ref.account_name == AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER:
                ref.account_name = storage_account_name
            if ref.azure_file_url and AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER in ref.azure_file_url:
                ref.azure_file_url = ref.azure_file_url.replace(
                    AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER, storage_account_name)
            if ref.credentials and ref.credentials.account_key == AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER:
                ref.credentials.account_key = storage_account_key
            if not ref.credentials:
                ref.credentials = models.AzureStorageCredentialsInfo(account_key=storage_account_key)

            # Verify that all placeholders are replaced with real values.
            if (ref.account_name == AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER or
                    AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER in ref.azure_file_url):
                raise CLIError(MSG_CONFIGURE_STORAGE_ACCOUNT)
            if ref.credentials.account_key == AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER:
                raise CLIError(MSG_CONFIGURE_STORAGE_KEY)

    # Patch parameters of blob file system.
    if params.node_setup and \
            params.node_setup.mount_volumes and \
            params.node_setup.mount_volumes.azure_blob_file_systems:
        for ref in params.node_setup.mount_volumes.azure_blob_file_systems:
            if ref.account_name == AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER:
                ref.account_name = storage_account_name
            if ref.credentials and ref.credentials.account_key == AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER:
                ref.credentials.account_key = storage_account_key
            if not ref.credentials:
                ref.credentials = models.AzureStorageCredentialsInfo(account_key=storage_account_key)
            # Verify that all placeholders are replaced with real values.
            if ref.account_name == AZURE_BATCHAI_STORAGE_ACCOUNT_PLACEHOLDER:
                raise CLIError(MSG_CONFIGURE_STORAGE_ACCOUNT)
            if ref.credentials.account_key == AZURE_BATCHAI_STORAGE_KEY_PLACEHOLDER:
                raise CLIError(MSG_CONFIGURE_STORAGE_KEY)
示例#20
0
def get_acr_api_version():
    '''Returns the api version for container registry
    '''
    customized_api_version = az_config.get('acr', 'apiversion', None)
    if customized_api_version:
        logger.warning('Customized api-version is used: %s', customized_api_version)
    return customized_api_version
示例#21
0
def log_telemetry(name, log_type='event', **kwargs):
    """
    IMPORTANT: do not log events with quotes in the name, properties or measurements;
    those events may fail to upload.  Also, telemetry events must be verified in
    the backend because successful upload does not guarentee success.
    """
    if not user_agrees_to_telemetry():
        return

    try:
        name = _remove_cmd_chars(name)
        _sanitize_inputs(kwargs)

        source = 'az'
        if _in_ci():
            source = 'CI'
        elif ARGCOMPLETE_ENV_NAME in os.environ:
            source = 'completer'

        types = ['event', 'pageview', 'trace']
        if log_type not in types:
            raise ValueError('Type {} is not supported.  Available types: {}'.format(log_type,
                                                                                     types))

        props = {
            'telemetry-version': TELEMETRY_VERSION
            }
        _safe_exec(props, 'time', lambda: str(datetime.datetime.now()))
        _safe_exec(props, 'x-ms-client-request-id',
                   lambda: APPLICATION.session['headers']['x-ms-client-request-id'])
        _safe_exec(props, 'command', lambda: APPLICATION.session.get('command', None))
        _safe_exec(props, 'version', lambda: core_version)
        _safe_exec(props, 'source', lambda: source)
        _safe_exec(props, 'installation-id', _get_installation_id)
        _safe_exec(props, 'python-version', lambda: _remove_symbols(str(platform.python_version())))
        _safe_exec(props, 'shell-type', _get_shell_type)
        _safe_exec(props, 'locale', lambda: '{},{}'.format(locale.getdefaultlocale()[0],
                                                           locale.getdefaultlocale()[1]))
        _safe_exec(props, 'user-machine-id', _get_user_machine_id)
        _safe_exec(props, 'user-azure-id', _get_user_azure_id)
        _safe_exec(props, 'azure-subscription-id', _get_azure_subscription_id)
        _safe_exec(props, 'default-output-type', lambda: az_config.get('core', 'output',
                                                                       fallback='unknown'))
        _safe_exec(props, 'environment', _get_env_string)
        if log_type == 'trace':
            _safe_exec(props, 'trace', _get_stack_trace)
            _safe_exec(props, 'error-hash', _get_error_hash)

        if kwargs:
            props.update(**kwargs)

        telemetry_records.append({
            'name': name,
            'type': log_type,
            'properties': props
            })
    except Exception as ex: #pylint: disable=broad-except
        # Never fail the command because of telemetry, unless debugging
        if _debugging():
            raise ex
示例#22
0
class AzRotatingFileHandler(logging.handlers.RotatingFileHandler):
    from azure.cli.core._environment import get_config_dir
    from azure.cli.core._config import az_config

    ENABLED = az_config.getboolean('logging',
                                   'enable_log_file',
                                   fallback=False)
    LOGFILE_DIR = os.path.expanduser(
        az_config.get('logging',
                      'log_dir',
                      fallback=os.path.join(get_config_dir(), 'logs')))

    def __init__(self):
        logging_file_path = self.get_log_file_path()
        super(AzRotatingFileHandler, self).__init__(logging_file_path,
                                                    maxBytes=10 * 1024 * 1024,
                                                    backupCount=5)
        self.setFormatter(
            logging.Formatter(
                '%(process)d : %(asctime)s : %(levelname)s : %(name)s : %(message)s'
            ))
        self.setLevel(logging.DEBUG)

    def get_log_file_path(self):
        if not os.path.isdir(self.LOGFILE_DIR):
            os.makedirs(self.LOGFILE_DIR)
        return os.path.join(self.LOGFILE_DIR, 'az.log')
示例#23
0
def get_acr_api_version():
    """Returns the api version for container registry """
    customized_api_version = az_config.get('acr', 'apiversion', None)
    if customized_api_version:
        logger.warning('Customized api-version is used: %s',
                       customized_api_version)
    return customized_api_version
示例#24
0
 def _update_default_info(self):
     try:
         options = az_config.config_parser.options(DEFAULTS_SECTION)
         self.config_default = ""
         for opt in options:
             self.config_default += opt + ": " + az_config.get(DEFAULTS_SECTION, opt) + "  "
     except configparser.NoSectionError:
         self.config_default = ""
示例#25
0
 def _update_default_info(self):
     try:
         options = az_config.config_parser.options(DEFAULTS_SECTION)
         self.config_default = ""
         for opt in options:
             self.config_default += opt + ": " + az_config.get(DEFAULTS_SECTION, opt) + "  "
     except configparser.NoSectionError:
         self.config_default = ""
示例#26
0
def _find_configured_default(argument):
    if not (hasattr(argument.type, 'default_name_tooling')
            and argument.type.default_name_tooling):
        return None
    try:
        return az_config.get(DEFAULTS_SECTION,
                             argument.type.default_name_tooling, None)
    except configparser.NoSectionError:
        return None
示例#27
0
def _install_or_update(package_list, link, private, pre, show_logs=False):
    options = ['--isolated', '--disable-pip-version-check', '--upgrade']
    if pre:
        options.append('--pre')
    if not show_logs:
        options.append('--quiet')
    pkg_index_options = ['--find-links', link] if link else []
    if private:
        package_index_url = az_config.get('component', 'package_index_url', fallback=None)
        package_index_trusted_host = az_config.get('component', 'package_index_trusted_host', fallback=None) #pylint: disable=line-too-long
        if package_index_url:
            pkg_index_options += ['--extra-index-url', package_index_url]
        else:
            raise CLIError('AZURE_COMPONENT_PACKAGE_INDEX_URL environment variable not set and not specified in config. ' #pylint: disable=line-too-long
                           'AZURE_COMPONENT_PACKAGE_INDEX_TRUSTED_HOST may also need to be set.') #pylint: disable=line-too-long
        pkg_index_options += ['--trusted-host', package_index_trusted_host] if package_index_trusted_host else [] #pylint: disable=line-too-long
    pip.main(['install'] + options + package_list
             + pkg_index_options)
示例#28
0
文件: custom.py 项目: Azure/azure-cli
def _install_or_update(package_list, link, private, pre):
    import pip
    options = ['--isolated', '--disable-pip-version-check', '--upgrade']
    if pre:
        options.append('--pre')
    pkg_index_options = ['--find-links', link] if link else []
    if private:
        package_index_url = az_config.get('component', 'package_index_url', fallback=None)
        package_index_trusted_host = az_config.get('component', 'package_index_trusted_host', fallback=None) #pylint: disable=line-too-long
        if package_index_url:
            pkg_index_options += ['--extra-index-url', package_index_url]
        else:
            raise CLIError('AZURE_COMPONENT_PACKAGE_INDEX_URL environment variable not set and not specified in config. ' #pylint: disable=line-too-long
                           'AZURE_COMPONENT_PACKAGE_INDEX_TRUSTED_HOST may also need to be set.\n'
                           'If executing az with sudo, you may want sudo\'s -E and -H flags.') #pylint: disable=line-too-long
        pkg_index_options += ['--trusted-host', package_index_trusted_host] if package_index_trusted_host else [] #pylint: disable=line-too-long
    pip_args = ['install'] + options + package_list + pkg_index_options
    _run_pip(pip, pip_args)
示例#29
0
def _install_or_update(package_list, link, private, pre, show_logs=False):
    options = ['--isolated', '--disable-pip-version-check', '--upgrade']
    if pre:
        options.append('--pre')
    if not show_logs:
        options.append('--quiet')
    pkg_index_options = ['--find-links', link] if link else []
    if private:
        package_index_url = az_config.get('component',
                                          'package_index_url',
                                          fallback=None)
        package_index_trusted_host = az_config.get('component', 'package_index_trusted_host', fallback=None)  #pylint: disable=line-too-long
        if package_index_url:
            pkg_index_options += ['--extra-index-url', package_index_url]
        else:
            raise CLIError('AZURE_COMPONENT_PACKAGE_INDEX_URL environment variable not set and not specified in config. '  #pylint: disable=line-too-long
                           'AZURE_COMPONENT_PACKAGE_INDEX_TRUSTED_HOST may also need to be set.')  #pylint: disable=line-too-long
        pkg_index_options += ['--trusted-host', package_index_trusted_host] if package_index_trusted_host else []  #pylint: disable=line-too-long
    pip.main(['install'] + options + package_list + pkg_index_options)
示例#30
0
def get_configured_defaults():
    _reload_config()
    try:
        options = az_config.config_parser.options(DEFAULTS_SECTION)
        defaults = {}
        for opt in options:
            value = az_config.get(DEFAULTS_SECTION, opt)
            if value:
                defaults[opt] = value
        return defaults
    except configparser.NoSectionError:
        return {}
示例#31
0
 def _resolve_default_value_from_cfg_file(self, arg, overrides):
     if 'configured_default' in overrides.settings:
         def_config = overrides.settings.pop('configured_default', None)
         # same blunt mechanism like we handled id-parts, for create command, no name default
         if (self.name.split()[-1] == 'create' and
                 overrides.settings.get('metavar', None) == 'NAME'):
             return
         setattr(arg.type, 'configured_default_applied', True)
         config_value = az_config.get(DEFAULTS_SECTION, def_config, None)
         if config_value:
             overrides.settings['default'] = config_value
             overrides.settings['required'] = False
示例#32
0
 def _register_builtin_arguments(**kwargs):
     global_group = kwargs['global_group']
     global_group.add_argument('--output', '-o', dest='_output_format',
                               choices=['json', 'tsv', 'table', 'jsonc'],
                               default=az_config.get('core', 'output', fallback='json'),
                               help='Output format',
                               type=str.lower)
     # The arguments for verbosity don't get parsed by argparse but we add it here for help.
     global_group.add_argument('--verbose', dest='_log_verbosity_verbose', action='store_true',
                               help='Increase logging verbosity. Use --debug for full debug logs.')
     global_group.add_argument('--debug', dest='_log_verbosity_debug', action='store_true',
                               help='Increase logging verbosity to show all debug logs.')
 def _resolve_default_value_from_cfg_file(self, arg, overrides):
     if 'configured_default' in overrides.settings:
         def_config = overrides.settings.pop('configured_default', None)
         # same blunt mechanism like we handled id-parts, for create command, no name default
         if (self.name.split()[-1] == 'create' and
                 overrides.settings.get('metavar', None) == 'NAME'):
             return
         setattr(arg.type, 'configured_default_applied', True)
         config_value = az_config.get(DEFAULTS_SECTION, def_config, None)
         if config_value:
             overrides.settings['default'] = config_value
             overrides.settings['required'] = False
示例#34
0
 def _register_builtin_arguments(**kwargs):
     global_group = kwargs['global_group']
     global_group.add_argument('--output', '-o', dest='_output_format',
                               choices=['json', 'tsv', 'table', 'jsonc','pandas'],
                               default=az_config.get('core', 'output', fallback='json'),
                               help='Output format',
                               type=str.lower)
     # The arguments for verbosity don't get parsed by argparse but we add it here for help.
     global_group.add_argument('--verbose', dest='_log_verbosity_verbose', action='store_true',
                               help='Increase logging verbosity. Use --debug for full debug logs.')  # pylint: disable=line-too-long
     global_group.add_argument('--debug', dest='_log_verbosity_debug', action='store_true',
                               help='Increase logging verbosity to show all debug logs.')
def load_node_agent_skus(prefix, **kwargs):  # pylint: disable=unused-argument
    from msrest.exceptions import ClientRequestError
    from azure.batch.models import BatchErrorException
    from azure.cli.command_modules.batch._client_factory import account_client_factory
    from azure.cli.core._config import az_config
    all_images = []
    client_creds = {}
    client_creds['account_name'] = az_config.get('batch', 'account', None)
    client_creds['account_key'] = az_config.get('batch', 'access_key', None)
    client_creds['account_endpoint'] = az_config.get('batch', 'endpoint', None)
    try:
        client = account_client_factory(client_creds)
        skus = client.list_node_agent_skus()
        for sku in skus:
            for image in sku['verifiedImageReferences']:
                all_images.append("{}:{}:{}:{}".format(image['publisher'],
                                                       image['offer'],
                                                       image['sku'],
                                                       image['version']))
        return all_images
    except (ClientRequestError, BatchErrorException):
        return []
示例#36
0
def validate_client_parameters(namespace):
    """Retrieves Batch connection parameters from environment variables"""
    from azure.mgmt.batch import BatchManagementClient
    from azure.cli.core.commands.client_factory import get_mgmt_service_client
    from azure.cli.core._config import az_config

    # simply try to retrieve the remaining variables from environment variables
    if not namespace.account_name:
        namespace.account_name = az_config.get('batch', 'account', None)
    if not namespace.account_key:
        namespace.account_key = az_config.get('batch', 'access_key', None)
    if not namespace.account_endpoint:
        namespace.account_endpoint = az_config.get('batch', 'endpoint', None)

    # if account name is specified but no key, attempt to query if we use shared key auth
    if namespace.account_name and namespace.account_endpoint and not namespace.account_key:
        if az_config.get('batch', 'auth_mode', 'shared_key') == 'shared_key':
            endpoint = urlsplit(namespace.account_endpoint)
            host = endpoint.netloc
            client = get_mgmt_service_client(BatchManagementClient)
            acc = next((x for x in client.batch_account.list()
                        if x.name == namespace.account_name and x.account_endpoint == host), None)
            if acc:
                from azure.cli.core.commands.arm import parse_resource_id
                rg = parse_resource_id(acc.id)['resource_group']
                namespace.account_key = \
                    client.batch_account.get_keys(rg,  # pylint: disable=no-member
                                                  namespace.account_name).primary
            else:
                raise ValueError("Batch account '{}' not found.".format(namespace.account_name))
    else:
        if not namespace.account_name:
            raise ValueError("Specify batch account in command line or environment variable.")
        if not namespace.account_endpoint:
            raise ValueError("Specify batch endpoint in command line or environment variable.")

    if az_config.get('batch', 'auth_mode', 'shared_key') == 'aad':
        namespace.account_key = None
示例#37
0
def load_node_agent_skus(prefix, **kwargs):  # pylint: disable=unused-argument
    from msrest.exceptions import ClientRequestError
    from azure.batch.models import BatchErrorException
    from azure.cli.command_modules.batch._client_factory import account_client_factory
    from azure.cli.core._config import az_config
    all_images = []
    client_creds = {}
    client_creds['account_name'] = az_config.get('batch', 'account', None)
    client_creds['account_key'] = az_config.get('batch', 'access_key', None)
    client_creds['account_endpoint'] = az_config.get('batch', 'endpoint', None)
    try:
        client = account_client_factory(client_creds)
        skus = client.list_node_agent_skus()
        for sku in skus:
            for image in sku['verifiedImageReferences']:
                all_images.append("{}:{}:{}:{}".format(
                    image['publisher'],
                    image['offer'],
                    image['sku'],
                    image['version']))
        return all_images
    except (ClientRequestError, BatchErrorException):
        return []
示例#38
0
def validate_source_uri(namespace):
    usage_string = 'invalid usage: supply only one of the following argument sets:' + \
                   '\n\t   --source-uri' + \
                   '\n\tOR --source-container --source-blob [--source-snapshot] [--source-sas]' + \
                   '\n\tOR --source-share --source-path [--source-sas]'
    ns = vars(namespace)
    validate_client_parameters(namespace)  # must run first to resolve storage account
    storage_acc = ns.get('account_name', None) or az_config.get('storage', 'account', None)
    uri = ns.get('copy_source', None)
    container = ns.pop('source_container', None)
    blob = ns.pop('source_blob', None)
    sas = ns.pop('source_sas', None)
    snapshot = ns.pop('source_snapshot', None)
    share = ns.pop('source_share', None)
    path = ns.pop('source_path', None)
    if uri:
        if any([container, blob, sas, snapshot, share, path]):
            raise ValueError(usage_string)
        else:
            # simplest scenario--no further processing necessary
            return

    valid_blob_source = container and blob and not share and not path
    valid_file_source = share and path and not container and not blob and not snapshot

    if (not valid_blob_source and not valid_file_source) or (valid_blob_source and valid_file_source):  # pylint: disable=line-too-long
        raise ValueError(usage_string)

    query_params = []
    if sas:
        query_params.append(sas)
    if snapshot:
        query_params.append(snapshot)

    uri = 'https://{0}.{1}.{6}/{2}/{3}{4}{5}'.format(
        storage_acc,
        'blob' if valid_blob_source else 'file',
        container if valid_blob_source else share,
        blob if valid_blob_source else path,
        '?' if query_params else '',
        '&'.join(query_params),
        CLOUD.suffixes.storage_endpoint)

    namespace.copy_source = uri
示例#39
0
def validate_source_uri(namespace):
    usage_string = 'invalid usage: supply only one of the following argument sets:' + \
                   '\n\t   --source-uri' + \
                   '\n\tOR --source-container --source-blob [--source-snapshot] [--source-sas]' + \
                   '\n\tOR --source-share --source-path [--source-sas]'
    ns = vars(namespace)
    validate_client_parameters(namespace)  # must run first to resolve storage account
    storage_acc = ns.get('account_name', None) or az_config.get('storage', 'account', None)
    uri = ns.get('copy_source', None)
    container = ns.pop('source_container', None)
    blob = ns.pop('source_blob', None)
    sas = ns.pop('source_sas', None)
    snapshot = ns.pop('source_snapshot', None)
    share = ns.pop('source_share', None)
    path = ns.pop('source_path', None)
    if uri:
        if any([container, blob, sas, snapshot, share, path]):
            raise ValueError(usage_string)
        else:
            # simplest scenario--no further processing necessary
            return

    valid_blob_source = container and blob and not share and not path
    valid_file_source = share and path and not container and not blob and not snapshot

    if (not valid_blob_source and not valid_file_source) or (valid_blob_source and valid_file_source):  # pylint: disable=line-too-long
        raise ValueError(usage_string)

    query_params = []
    if sas:
        query_params.append(sas)
    if snapshot:
        query_params.append(snapshot)

    uri = 'https://{0}.{1}.{6}/{2}/{3}{4}{5}'.format(
        storage_acc,
        'blob' if valid_blob_source else 'file',
        container if valid_blob_source else share,
        blob if valid_blob_source else path,
        '?' if query_params else '',
        '&'.join(query_params),
        CLOUD.suffixes.storage_endpoint)

    namespace.copy_source = uri
示例#40
0
 def _register_builtin_arguments(**kwargs):
     global_group = kwargs["global_group"]
     global_group.add_argument(
         "--output",
         "-o",
         dest="_output_format",
         choices=["json", "tsv", "list", "table", "jsonc"],
         default=az_config.get("core", "output", fallback="json"),
         help="Output format",
         type=str.lower,
     )
     # The arguments for verbosity don't get parsed by argparse but we add it here for help.
     global_group.add_argument(
         "--verbose",
         dest="_log_verbosity_verbose",
         action="store_true",
         help="Increase logging verbosity. Use --debug for full debug logs.",
     )  # pylint: disable=line-too-long
     global_group.add_argument(
         "--debug",
         dest="_log_verbosity_debug",
         action="store_true",
         help="Increase logging verbosity to show all debug logs.",
     )
示例#41
0
import time
import uuid
from subprocess import CalledProcessError, check_output

import requests
import yaml
import azure.cli.core.azlogging as azlogging
from azure.cli.core._config import az_config
from azure.cli.core._profile import _SERVICE_PRINCIPAL, CredsCache, Profile
# pylint: disable=too-few-public-methods,too-many-arguments,no-self-use,too-many-locals,line-too-long
from azure.cli.core.util import CLIError

logger = azlogging.get_az_logger(__name__)

BASE_URL = az_config.get('container',
                         'service_url',
                         fallback='https://api.mindaro.microsoft.io')
SUBSCRIPTION_URL = "/subscriptions/{subscription_id}"
RESOURCE_BASE_URL = SUBSCRIPTION_URL + "/resourceGroups/{resource_group_name}"
CONTAINER_SERVICE_PROVIDER = "/providers/Microsoft.ContainerService"
CONTAINER_SERVICE_RESOURCE_URL = RESOURCE_BASE_URL + \
                                 CONTAINER_SERVICE_PROVIDER + \
                                 "/containerServices/{container_service_name}"

SERVICE_URL = BASE_URL + SUBSCRIPTION_URL
API_VERSION = "2016-11-01-preview"
SERVICE_RESOURCE_ID = "https://mindaro.microsoft.io/"

DOCKERFILE_FILE = 'Dockerfile'
DOCKER_COMPOSE_FILE = 'docker-compose.yml'
DOCKER_COMPOSE_EXPECTED_VERSION = '2'
示例#42
0
import logging
from logging.handlers import RotatingFileHandler

import colorama

from azure.cli.core._environment import get_config_dir
from azure.cli.core._config import az_config

AZ_LOGFILE_NAME = 'az.log'
DEFAULT_LOG_DIR = os.path.join(get_config_dir(), 'logs')

ENABLE_LOG_FILE = az_config.getboolean('logging',
                                       'enable_log_file',
                                       fallback=False)
LOG_DIR = os.path.expanduser(
    az_config.get('logging', 'log_dir', fallback=DEFAULT_LOG_DIR))

CONSOLE_LOG_CONFIGS = [
    # (default)
    {
        'az': logging.WARNING,
        'root': logging.CRITICAL,
    },
    # --verbose
    {
        'az': logging.INFO,
        'root': logging.CRITICAL,
    },
    # --debug
    {
        'az': logging.DEBUG,
示例#43
0
def _get_client(service, parsed_args):
    account_name = parsed_args.account_name or az_config.get('storage', 'account', None)
    account_key = parsed_args.account_key or az_config.get('storage', 'key', None)
    connection_string = parsed_args.connection_string or az_config.get('storage', 'connection_string', None)
    sas_token = parsed_args.sas_token or az_config.get('storage', 'sas_token', None)
    return get_data_service_client(service, account_name, account_key, connection_string, sas_token)
import os
import platform
import logging
from logging.handlers import RotatingFileHandler

import colorama

from azure.cli.core._environment import get_config_dir
from azure.cli.core._config import az_config

AZ_LOGFILE_NAME = 'az.log'
DEFAULT_LOG_DIR = os.path.join(get_config_dir(), 'logs')

ENABLE_LOG_FILE = az_config.getboolean('logging', 'enable_log_file', fallback=False)
LOG_DIR = os.path.expanduser(az_config.get('logging', 'log_dir', fallback=DEFAULT_LOG_DIR))

CONSOLE_LOG_CONFIGS = [
    # (default)
    {
        'az': logging.WARNING,
        'root': logging.CRITICAL,
    },
    # --verbose
    {
        'az': logging.INFO,
        'root': logging.CRITICAL,
    },
    # --debug
    {
        'az': logging.DEBUG,
示例#45
0
    def on_input_timeout(self, cli):
        """
        When there is a pause in typing
        Brings up the metadata for the command if
        there is a valid command already typed
        """
        rows, cols = get_window_dim()
        rows = int(rows)
        cols = int(cols)
        document = cli.current_buffer.document
        text = document.text
        command = ""
        all_params = ""
        example = ""
        empty_space = ""
        for i in range(cols):
            empty_space += " "
        any_documentation = False
        is_command = True
        for word in text.split():
            if word.startswith("-"):
                is_command = False
            if is_command:
                if not word == 'az':
                    command += str(word) + " "

            if self.completer.is_completable(command.rstrip()):
                cmdstp = command.rstrip()
                any_documentation = True

                if word in self.completer.command_parameters[cmdstp] and \
                self.completer.has_description(cmdstp + " " + word):
                    all_params = word + ":\n" + \
                    self.completer.get_param_description(cmdstp+ \
                    " " + word)

                self.description_docs = u"%s" % \
                self.completer.get_description(cmdstp)

                if cmdstp in self.completer.command_examples:
                    string_example = ""
                    for example in self.completer.command_examples[cmdstp]:
                        for part in example:
                            string_example += part
                    example = self.space_examples(
                        self.completer.command_examples[cmdstp], rows)

        if not any_documentation:
            self.description_docs = u''

        self.param_docs = u"%s" % all_params
        self.example_docs = u'%s' % example

        options = az_config.config_parser.options(DEFAULTS_SECTION)
        self.config_default = ""
        for opt in options:
            self.config_default += opt + ": " + az_config.get(DEFAULTS_SECTION, opt) + "  "
        

        settings, empty_space = self._toolbar_info(cols, empty_space)

        cli.buffers['description'].reset(
            initial_document=Document(self.description_docs, cursor_position=0)
        )
        cli.buffers['parameter'].reset(
            initial_document=Document(self.param_docs)
        )
        cli.buffers['examples'].reset(
            initial_document=Document(self.example_docs)
        )
        cli.buffers['bottom_toolbar'].reset(
            initial_document=Document(u'%s%s%s' % \
            (NOTIFICATIONS, settings, empty_space))
        )
        cli.buffers['default_values'].reset(
            initial_document=Document(u'%s' %self.config_default)
        )
        cli.request_redraw()
示例#46
0
import re
import time
import uuid
from subprocess import CalledProcessError, check_output

import requests
import yaml
import azure.cli.core.azlogging as azlogging
from azure.cli.core._config import az_config
from azure.cli.core._profile import _SERVICE_PRINCIPAL, CredsCache, Profile
# pylint: disable=too-few-public-methods,too-many-arguments,no-self-use,too-many-locals,line-too-long
from azure.cli.core.util import CLIError

logger = azlogging.get_az_logger(__name__)

BASE_URL = az_config.get('container', 'service_url', fallback='https://api.mindaro.microsoft.io')
SUBSCRIPTION_URL = "/subscriptions/{subscription_id}"
RESOURCE_BASE_URL = SUBSCRIPTION_URL + "/resourceGroups/{resource_group_name}"
CONTAINER_SERVICE_PROVIDER = "/providers/Microsoft.ContainerService"
CONTAINER_SERVICE_RESOURCE_URL = (RESOURCE_BASE_URL + CONTAINER_SERVICE_PROVIDER + "/containerServices/{container_service_name}")

SERVICE_URL = BASE_URL + SUBSCRIPTION_URL
API_VERSION = "2016-11-01-preview"
SERVICE_RESOURCE_ID = "https://mindaro.microsoft.io/"

DOCKERFILE_FILE = 'Dockerfile'
DOCKER_COMPOSE_FILE = 'docker-compose.yml'
DOCKER_COMPOSE_EXPECTED_VERSION = '2'

def add_release(
        target_name,
示例#47
0
def log_telemetry(name, log_type='event', **kwargs):
    """
    IMPORTANT: do not log events with quotes in the name, properties or measurements;
    those events may fail to upload.  Also, telemetry events must be verified in
    the backend because successful upload does not guarentee success.
    """
    if not user_agrees_to_telemetry():
        return

    # Now we now we want to log telemetry, get the profile
    try:
        from azure.cli.core._profile import Profile
        profile = Profile()
    except: #pylint: disable=bare-except
        profile = None

    try:
        name = _remove_cmd_chars(name)
        _sanitize_inputs(kwargs)

        source = 'az'
        if _in_ci():
            source = 'CI'
        elif ARGCOMPLETE_ENV_NAME in os.environ:
            source = 'completer'

        types = ['event', 'pageview', 'trace']
        if log_type not in types:
            raise ValueError('Type {} is not supported.  Available types: {}'.format(log_type,
                                                                                     types))

        props = {
            'telemetry-version': TELEMETRY_VERSION
            }
        _safe_exec(props, 'time', lambda: str(datetime.datetime.now()))
        _safe_exec(props, 'x-ms-client-request-id',
                   lambda: APPLICATION.session['headers']['x-ms-client-request-id'])
        _safe_exec(props, 'command', lambda: APPLICATION.session.get('command', None))
        _safe_exec(props, 'version', lambda: core_version)
        _safe_exec(props, 'source', lambda: source)
        _safe_exec(props, 'installation-id', lambda: _get_installation_id(profile))
        _safe_exec(props, 'python-version', lambda: _remove_symbols(str(platform.python_version())))
        _safe_exec(props, 'shell-type', _get_shell_type)
        _safe_exec(props, 'locale', lambda: '{},{}'.format(locale.getdefaultlocale()[0],
                                                           locale.getdefaultlocale()[1]))
        _safe_exec(props, 'user-machine-id', _get_user_machine_id)
        _safe_exec(props, 'user-azure-id', lambda: _get_user_azure_id(profile))
        _safe_exec(props, 'azure-subscription-id', lambda: _get_azure_subscription_id(profile))
        _safe_exec(props, 'default-output-type', lambda: az_config.get('core', 'output',
                                                                       fallback='unknown'))
        _safe_exec(props, 'environment', _get_env_string)
        if log_type == 'trace':
            _safe_exec(props, 'trace', _get_stack_trace)
            _safe_exec(props, 'error-hash', _get_error_hash)

        if kwargs:
            props.update(**kwargs)

        telemetry_records.append({
            'name': name,
            'type': log_type,
            'properties': props
            })
    except Exception as ex: #pylint: disable=broad-except
        # Never fail the command because of telemetry, unless debugging
        if _debugging():
            raise ex