3394 строки
109 KiB
Python
3394 строки
109 KiB
Python
# Copyright (c) Microsoft Corporation
|
|
#
|
|
# All rights reserved.
|
|
#
|
|
# MIT License
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a
|
|
# copy of this software and associated documentation files (the "Software"),
|
|
# to deal in the Software without restriction, including without limitation
|
|
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
# and/or sell copies of the Software, and to permit persons to whom the
|
|
# Software is furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in
|
|
# all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
# DEALINGS IN THE SOFTWARE.
|
|
|
|
# compat imports
|
|
from __future__ import (
|
|
absolute_import, division, print_function, unicode_literals
|
|
)
|
|
from builtins import ( # noqa
|
|
bytes, dict, int, list, object, range, str, ascii, chr, hex, input,
|
|
next, oct, open, pow, round, super, filter, map, zip)
|
|
# stdlib imports
|
|
import collections
|
|
import datetime
|
|
try:
|
|
import pathlib2 as pathlib
|
|
except ImportError:
|
|
import pathlib
|
|
# non-stdlib imports
|
|
import dateutil.parser
|
|
# local imports
|
|
from . import task_factory
|
|
from . import util
|
|
|
|
# global defines
|
|
_METADATA_VERSION_NAME = 'batch_shipyard_version'
|
|
_GLUSTER_DEFAULT_VOLNAME = 'gv0'
|
|
_GLUSTER_ON_COMPUTE_VOLUME = '.gluster/{}'.format(_GLUSTER_DEFAULT_VOLNAME)
|
|
_TENSORBOARD_DOCKER_IMAGE = (
|
|
'gcr.io/tensorflow/tensorflow',
|
|
'/usr/local/lib/python2.7/dist-packages/tensorflow'
|
|
'/tensorboard/tensorboard.py',
|
|
6006
|
|
)
|
|
_GPU_COMPUTE_INSTANCES = frozenset((
|
|
'standard_nc6', 'standard_nc12', 'standard_nc24', 'standard_nc24r',
|
|
))
|
|
_GPU_VISUALIZATION_INSTANCES = frozenset((
|
|
'standard_nv6', 'standard_nv12', 'standard_nv24',
|
|
))
|
|
_GPU_INSTANCES = _GPU_COMPUTE_INSTANCES.union(_GPU_VISUALIZATION_INSTANCES)
|
|
_RDMA_INSTANCES = frozenset((
|
|
'standard_a8', 'standard_a9', 'standard_h16r', 'standard_h16mr',
|
|
'standard_nc24r',
|
|
))
|
|
_PREMIUM_STORAGE_INSTANCE_PREFIXES = frozenset((
|
|
'standard_ds', 'standard_gs',
|
|
))
|
|
_PREMIUM_STORAGE_INSTANCE_SUFFIXES = frozenset((
|
|
's', 's_v3',
|
|
))
|
|
_VM_TCP_NO_TUNE = (
|
|
'basic_a0', 'basic_a1', 'basic_a2', 'basic_a3', 'basic_a4', 'standard_a0',
|
|
'standard_a1', 'standard_a2', 'standard_a3', 'standard_a5', 'standard_a6',
|
|
'standard_a1_v2', 'standard_a2_v2', 'standard_a3_v2', 'standard_a4_v2',
|
|
'standard_a2m_v2', 'standard_a4m_v2', 'standard_d1', 'standard_d2',
|
|
'standard_d1_v2', 'standard_f1'
|
|
)
|
|
# named tuples
|
|
PoolVmCountSettings = collections.namedtuple(
|
|
'PoolVmCountSettings', [
|
|
'dedicated',
|
|
'low_priority',
|
|
]
|
|
)
|
|
PoolVmPlatformImageSettings = collections.namedtuple(
|
|
'PoolVmPlatformImageSettings', [
|
|
'publisher',
|
|
'offer',
|
|
'sku',
|
|
]
|
|
)
|
|
PoolVmCustomImageSettings = collections.namedtuple(
|
|
'PoolVmCustomImageSettings', [
|
|
'image_uris',
|
|
'node_agent',
|
|
]
|
|
)
|
|
PoolAutoscaleScenarioSettings = collections.namedtuple(
|
|
'PoolAutoscaleScenarioSettings', [
|
|
'name',
|
|
'maximum_vm_count',
|
|
'node_deallocation_option',
|
|
'sample_lookback_interval',
|
|
'required_sample_percentage',
|
|
'rebalance_preemption_percentage',
|
|
'bias_last_sample',
|
|
'bias_node_type',
|
|
]
|
|
)
|
|
PoolAutoscaleSettings = collections.namedtuple(
|
|
'PoolAutoscaleSettings', [
|
|
'evaluation_interval',
|
|
'formula',
|
|
'scenario',
|
|
]
|
|
)
|
|
PoolAutopoolSettings = collections.namedtuple(
|
|
'PoolAutopoolSettings', [
|
|
'pool_lifetime',
|
|
'keep_alive',
|
|
]
|
|
)
|
|
PoolSettings = collections.namedtuple(
|
|
'PoolSettings', [
|
|
'id', 'vm_size', 'vm_count', 'resize_timeout', 'max_tasks_per_node',
|
|
'inter_node_communication_enabled', 'vm_configuration',
|
|
'reboot_on_start_task_failed',
|
|
'block_until_all_global_resources_loaded',
|
|
'transfer_files_on_pool_creation', 'input_data', 'resource_files',
|
|
'gpu_driver', 'ssh', 'additional_node_prep_commands',
|
|
'virtual_network', 'autoscale', 'node_fill_type',
|
|
]
|
|
)
|
|
SSHSettings = collections.namedtuple(
|
|
'SSHSettings', [
|
|
'username', 'expiry_days', 'ssh_public_key', 'ssh_public_key_data',
|
|
'ssh_private_key', 'generate_docker_tunnel_script',
|
|
'generated_file_export_path', 'hpn_server_swap',
|
|
]
|
|
)
|
|
AADSettings = collections.namedtuple(
|
|
'AADSettings', [
|
|
'directory_id', 'application_id', 'auth_key', 'rsa_private_key_pem',
|
|
'x509_cert_sha1_thumbprint', 'user', 'password', 'endpoint',
|
|
'token_cache_file',
|
|
]
|
|
)
|
|
KeyVaultCredentialsSettings = collections.namedtuple(
|
|
'KeyVaultCredentialsSettings', [
|
|
'aad', 'keyvault_uri', 'keyvault_credentials_secret_id',
|
|
]
|
|
)
|
|
ManagementCredentialsSettings = collections.namedtuple(
|
|
'ManagementCredentialsSettings', [
|
|
'aad', 'subscription_id',
|
|
]
|
|
)
|
|
BatchCredentialsSettings = collections.namedtuple(
|
|
'BatchCredentialsSettings', [
|
|
'aad', 'account', 'account_key', 'account_service_url',
|
|
'resource_group', 'subscription_id', 'location',
|
|
]
|
|
)
|
|
StorageCredentialsSettings = collections.namedtuple(
|
|
'StorageCredentialsSettings', [
|
|
'account', 'account_key', 'endpoint',
|
|
]
|
|
)
|
|
BatchShipyardSettings = collections.namedtuple(
|
|
'BatchShipyardSettings', [
|
|
'storage_account_settings', 'storage_entity_prefix',
|
|
'generated_sas_expiry_days', 'use_shipyard_docker_image',
|
|
'store_timing_metrics',
|
|
]
|
|
)
|
|
DockerRegistrySettings = collections.namedtuple(
|
|
'DockerRegistrySettings', [
|
|
'allow_public_docker_hub_pull_on_missing',
|
|
'storage_account', 'container', 'server', 'port',
|
|
'user', 'password',
|
|
]
|
|
)
|
|
DataReplicationSettings = collections.namedtuple(
|
|
'DataReplicationSettings', [
|
|
'peer_to_peer', 'concurrent_source_downloads',
|
|
]
|
|
)
|
|
PeerToPeerSettings = collections.namedtuple(
|
|
'PeerToPeerSettings', [
|
|
'enabled', 'compression', 'direct_download_seed_bias',
|
|
]
|
|
)
|
|
SourceSettings = collections.namedtuple(
|
|
'SourceSettings', [
|
|
'path', 'include', 'exclude'
|
|
]
|
|
)
|
|
DestinationSettings = collections.namedtuple(
|
|
'DestinationSettings', [
|
|
'storage_account_settings', 'shared_data_volume',
|
|
'relative_destination_path', 'data_transfer'
|
|
]
|
|
)
|
|
DataTransferSettings = collections.namedtuple(
|
|
'DataTransferSettings', [
|
|
'method', 'ssh_private_key', 'scp_ssh_extra_options',
|
|
'rsync_extra_options', 'split_files_megabytes',
|
|
'max_parallel_transfers_per_node',
|
|
'container', 'file_share', 'blobxfer_extra_options',
|
|
]
|
|
)
|
|
JobScheduleSettings = collections.namedtuple(
|
|
'JobScheduleSettings', [
|
|
'do_not_run_until', 'do_not_run_after', 'start_window',
|
|
'recurrence_interval',
|
|
]
|
|
)
|
|
JobManagerSettings = collections.namedtuple(
|
|
'JobManagerSettings', [
|
|
'allow_low_priority_node', 'run_exclusive', 'monitor_task_completion',
|
|
]
|
|
)
|
|
JobRecurrenceSettings = collections.namedtuple(
|
|
'JobRecurrenceSettings', [
|
|
'schedule', 'job_manager',
|
|
]
|
|
)
|
|
UserIdentitySettings = collections.namedtuple(
|
|
'UserIdentitySettings', [
|
|
'default_pool_admin', 'specific_user_uid', 'specific_user_gid',
|
|
]
|
|
)
|
|
TaskFactoryStorageSettings = collections.namedtuple(
|
|
'TaskFactoryStorageSettings', [
|
|
'storage_settings', 'storage_link_name', 'container', 'file_share',
|
|
'include', 'exclude',
|
|
]
|
|
)
|
|
TaskSettings = collections.namedtuple(
|
|
'TaskSettings', [
|
|
'id', 'image', 'name', 'docker_run_options', 'environment_variables',
|
|
'environment_variables_keyvault_secret_id', 'envfile',
|
|
'resource_files', 'command', 'infiniband', 'gpu', 'depends_on',
|
|
'depends_on_range', 'max_task_retries', 'max_wall_time',
|
|
'retention_time', 'docker_run_cmd', 'docker_exec_cmd',
|
|
'multi_instance',
|
|
]
|
|
)
|
|
MultiInstanceSettings = collections.namedtuple(
|
|
'MultiInstanceSettings', [
|
|
'num_instances', 'coordination_command', 'resource_files',
|
|
]
|
|
)
|
|
ResourceFileSettings = collections.namedtuple(
|
|
'ResourceFileSettings', [
|
|
'file_path', 'blob_source', 'file_mode',
|
|
]
|
|
)
|
|
ManagedDisksSettings = collections.namedtuple(
|
|
'ManagedDisksSettings', [
|
|
'resource_group', 'premium', 'disk_size_gb', 'disk_names',
|
|
]
|
|
)
|
|
VirtualNetworkSettings = collections.namedtuple(
|
|
'VirtualNetworkSettings', [
|
|
'name', 'resource_group', 'address_space', 'subnet_name',
|
|
'subnet_address_prefix', 'existing_ok', 'create_nonexistant',
|
|
]
|
|
)
|
|
SambaAccountSettings = collections.namedtuple(
|
|
'SambaAccountSettings', [
|
|
'username', 'password', 'uid', 'gid',
|
|
]
|
|
)
|
|
SambaSettings = collections.namedtuple(
|
|
'SambaSettings', [
|
|
'share_name', 'account', 'read_only', 'create_mask',
|
|
'directory_mask',
|
|
]
|
|
)
|
|
FileServerSettings = collections.namedtuple(
|
|
'FileServerSettings', [
|
|
'type', 'mountpoint', 'mount_options', 'server_options', 'samba',
|
|
]
|
|
)
|
|
InboundNetworkSecurityRule = collections.namedtuple(
|
|
'InboundNetworkSecurityRule', [
|
|
'destination_port_range', 'source_address_prefix', 'protocol',
|
|
]
|
|
)
|
|
NetworkSecuritySettings = collections.namedtuple(
|
|
'NetworkSecuritySettings', [
|
|
'inbound',
|
|
]
|
|
)
|
|
MappedVmDiskSettings = collections.namedtuple(
|
|
'MappedVmDiskSettings', [
|
|
'disk_array', 'filesystem', 'raid_level',
|
|
]
|
|
)
|
|
PublicIpSettings = collections.namedtuple(
|
|
'PublicIpSettings', [
|
|
'enabled', 'static',
|
|
]
|
|
)
|
|
StorageClusterSettings = collections.namedtuple(
|
|
'StorageClusterSettings', [
|
|
'id', 'resource_group', 'virtual_network', 'network_security',
|
|
'file_server', 'vm_count', 'vm_size', 'fault_domains', 'public_ip',
|
|
'hostname_prefix', 'ssh', 'vm_disk_map',
|
|
]
|
|
)
|
|
RemoteFsSettings = collections.namedtuple(
|
|
'RemoteFsSettings', [
|
|
'location', 'managed_disks', 'storage_cluster',
|
|
]
|
|
)
|
|
|
|
|
|
def _kv_read_checked(conf, key, default=None):
|
|
# type: (dict, str, obj) -> obj
|
|
"""Read a key as some value with a check against None and length
|
|
:param dict conf: configuration dict
|
|
:param str key: conf key
|
|
:param obj default: default to assign
|
|
:rtype: obj or None
|
|
:return: value of key
|
|
"""
|
|
try:
|
|
ret = conf[key]
|
|
if util.is_none_or_empty(ret):
|
|
raise KeyError()
|
|
except KeyError:
|
|
ret = default
|
|
return ret
|
|
|
|
|
|
def _kv_read(conf, key, default=None):
|
|
# type: (dict, str, obj) -> obj
|
|
"""Read a key as some value
|
|
:param dict conf: configuration dict
|
|
:param str key: conf key
|
|
:param obj default: default to assign
|
|
:rtype: obj or None
|
|
:return: value of key
|
|
"""
|
|
try:
|
|
ret = conf[key]
|
|
except KeyError:
|
|
ret = default
|
|
return ret
|
|
|
|
|
|
def get_metadata_version_name():
|
|
# type: (None) -> str
|
|
"""Get metadata version name
|
|
:rtype: str
|
|
:return: metadata version name
|
|
"""
|
|
return _METADATA_VERSION_NAME
|
|
|
|
|
|
def get_tensorboard_docker_image():
|
|
# type: (None) -> Tuple[str, str]
|
|
"""Get tensorboard docker image
|
|
:rtype: tuple
|
|
:return: (tensorboard docker image,
|
|
absolute path to tensorboard.py, container port)
|
|
"""
|
|
return _TENSORBOARD_DOCKER_IMAGE
|
|
|
|
|
|
def get_gluster_default_volume_name():
|
|
# type: (None) -> str
|
|
"""Get gluster default volume name
|
|
:rtype: str
|
|
:return: gluster default volume name
|
|
"""
|
|
return _GLUSTER_DEFAULT_VOLNAME
|
|
|
|
|
|
def get_gluster_on_compute_volume():
|
|
# type: (None) -> str
|
|
"""Get gluster on compute volume mount suffix
|
|
:rtype: str
|
|
:return: gluster on compute volume mount
|
|
"""
|
|
return _GLUSTER_ON_COMPUTE_VOLUME
|
|
|
|
|
|
def can_tune_tcp(vm_size):
|
|
# type: (str) -> bool
|
|
"""Check if TCP tuning on compute node should be performed
|
|
:param str vm_size: vm size
|
|
:rtype: bool
|
|
:return: True if VM should be tuned
|
|
"""
|
|
if vm_size.lower() in _VM_TCP_NO_TUNE:
|
|
return False
|
|
return True
|
|
|
|
|
|
def is_gpu_pool(vm_size):
|
|
# type: (str) -> bool
|
|
"""Check if pool is GPU capable
|
|
:param str vm_size: vm size
|
|
:rtype: bool
|
|
:return: if gpus are present
|
|
"""
|
|
if vm_size.lower() in _GPU_INSTANCES:
|
|
return True
|
|
return False
|
|
|
|
|
|
def is_gpu_compute_pool(vm_size):
|
|
# type: (str) -> bool
|
|
"""Check if pool is for GPU compute
|
|
:param str vm_size: vm size
|
|
:rtype: bool
|
|
:return: if compute gpus are present
|
|
"""
|
|
if vm_size.lower() in _GPU_COMPUTE_INSTANCES:
|
|
return True
|
|
return False
|
|
|
|
|
|
def is_gpu_visualization_pool(vm_size):
|
|
# type: (str) -> bool
|
|
"""Check if pool is for GPU visualization
|
|
:param str vm_size: vm size
|
|
:rtype: bool
|
|
:return: if visualization gpus are present
|
|
"""
|
|
if vm_size.lower() in _GPU_VISUALIZATION_INSTANCES:
|
|
return True
|
|
return False
|
|
|
|
|
|
def get_gpu_type_from_vm_size(vm_size):
|
|
# type: (str) -> str
|
|
"""Get GPU type as string
|
|
:param str vm_size: vm size
|
|
:rtype: str
|
|
:return: compute for gpgpu and visualization for viz
|
|
"""
|
|
if is_gpu_compute_pool(vm_size):
|
|
return 'compute'
|
|
elif is_gpu_visualization_pool(vm_size):
|
|
return 'visualization'
|
|
else:
|
|
return None
|
|
|
|
|
|
def gpu_configuration_check(config, vm_size=None):
|
|
# type: (dict, str) -> bool
|
|
"""Check if OS is allowed with a GPU VM
|
|
:param dict config: configuration dict
|
|
:param str vm_size: vm size
|
|
:rtype: bool
|
|
:return: if configuration is allowed
|
|
"""
|
|
# if this is not a gpu sku, always allow
|
|
if util.is_none_or_empty(vm_size):
|
|
vm_size = pool_settings(config).vm_size
|
|
if not is_gpu_pool(vm_size):
|
|
return True
|
|
# always allow gpu with custom images
|
|
node_agent = pool_custom_image_node_agent(config)
|
|
if util.is_not_empty(node_agent):
|
|
return True
|
|
# check for platform image support
|
|
publisher = pool_publisher(config, lower=True)
|
|
offer = pool_offer(config, lower=True)
|
|
sku = pool_sku(config, lower=True)
|
|
if (publisher == 'canonical' and offer == 'ubuntuserver' and
|
|
sku > '16.04'):
|
|
return True
|
|
elif (publisher == 'openlogic' and
|
|
(offer == 'centos' or offer == 'centos-hpc') and sku == '7.3'):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def is_rdma_pool(vm_size):
|
|
# type: (str) -> bool
|
|
"""Check if pool is IB/RDMA capable
|
|
:param str vm_size: vm size
|
|
:rtype: bool
|
|
:return: if rdma is present
|
|
"""
|
|
if vm_size.lower() in _RDMA_INSTANCES:
|
|
return True
|
|
return False
|
|
|
|
|
|
def is_premium_storage_vm_size(vm_size):
|
|
# type: (str) -> bool
|
|
"""Check if vm size is premium storage compatible
|
|
:pararm str vm_size: vm size
|
|
:rtype: bool
|
|
:return: if vm size is premium storage compatible
|
|
"""
|
|
if any([vm_size.lower().endswith(x)
|
|
for x in _PREMIUM_STORAGE_INSTANCE_SUFFIXES]):
|
|
return True
|
|
elif any([vm_size.lower().startswith(x)
|
|
for x in _PREMIUM_STORAGE_INSTANCE_PREFIXES]):
|
|
return True
|
|
return False
|
|
|
|
|
|
def temp_disk_mountpoint(config, offer=None):
|
|
# type: (dict) -> str
|
|
"""Get temporary disk mountpoint
|
|
:param dict config: configuration object
|
|
:param str offer: offer override
|
|
:rtype: str
|
|
:return: temporary disk mount point
|
|
"""
|
|
if offer is None:
|
|
vmconfig = _populate_pool_vm_configuration(config)
|
|
if isinstance(vmconfig, PoolVmPlatformImageSettings):
|
|
offer = pool_offer(config, lower=True)
|
|
else:
|
|
if vmconfig.node_agent.lower().startswith('batch.node.ubuntu'):
|
|
offer = 'ubuntuserver'
|
|
else:
|
|
offer = None
|
|
else:
|
|
offer = offer.lower()
|
|
if offer == 'ubuntuserver':
|
|
return '/mnt'
|
|
else:
|
|
return '/mnt/resource'
|
|
|
|
|
|
def verbose(config):
|
|
# type: (dict) -> bool
|
|
"""Get verbose setting
|
|
:param dict config: configuration object
|
|
:rtype: bool
|
|
:return: verbose setting
|
|
"""
|
|
return config['_verbose']
|
|
|
|
|
|
def set_auto_confirm(config, flag):
|
|
# type: (dict, bool) -> None
|
|
"""Set autoconfirm setting
|
|
:param dict config: configuration object
|
|
:param bool flag: flag to set
|
|
"""
|
|
config['_auto_confirm'] = flag
|
|
|
|
|
|
# POOL CONFIG
|
|
def pool_specification(config):
|
|
# type: (dict) -> dict
|
|
"""Get Pool specification config block
|
|
:param dict config: configuration object
|
|
:rtype: dict
|
|
:return: pool specification
|
|
"""
|
|
return config['pool_specification']
|
|
|
|
|
|
def _pool_vm_count(config, conf=None):
|
|
# type: (dict, dict) -> PoolVmCountSettings
|
|
"""Get Pool vm count settings
|
|
:param dict config: configuration object
|
|
:param dict conf: vm_count object
|
|
:rtype: PoolVmCountSettings
|
|
:return: pool vm count settings
|
|
"""
|
|
if conf is None:
|
|
conf = pool_specification(config)['vm_count']
|
|
if isinstance(conf, int):
|
|
conf = {'dedicated': conf}
|
|
return PoolVmCountSettings(
|
|
dedicated=_kv_read(conf, 'dedicated', 0),
|
|
low_priority=_kv_read(conf, 'low_priority', 0),
|
|
)
|
|
|
|
|
|
def pool_vm_configuration(config, key):
|
|
# type: (dict, str) -> dict
|
|
"""Get Pool VM configuration
|
|
:param dict config: configuration object
|
|
:param str key: vm config key
|
|
:rtype: str
|
|
:return: pool vm config
|
|
"""
|
|
try:
|
|
conf = _kv_read_checked(
|
|
config['pool_specification']['vm_configuration'], key)
|
|
except KeyError:
|
|
conf = None
|
|
if conf is None:
|
|
return config['pool_specification']
|
|
else:
|
|
return conf
|
|
|
|
|
|
def _populate_pool_vm_configuration(config):
|
|
# type: (dict) -> dict
|
|
"""Populate Pool VM configuration
|
|
:param dict config: configuration object
|
|
:rtype: PoolVmPlatformImageSettings or PoolVmCustomImageSettings
|
|
:return: pool vm config
|
|
"""
|
|
conf = pool_vm_configuration(config, 'platform_image')
|
|
if 'publisher' in conf:
|
|
return PoolVmPlatformImageSettings(
|
|
publisher=conf['publisher'],
|
|
offer=conf['offer'],
|
|
sku=conf['sku'],
|
|
)
|
|
else:
|
|
conf = pool_vm_configuration(config, 'custom_image')
|
|
return PoolVmCustomImageSettings(
|
|
image_uris=conf['image_uris'],
|
|
node_agent=conf['node_agent'],
|
|
)
|
|
|
|
|
|
def pool_autoscale_settings(config):
|
|
# type: (dict) -> PoolAutoscaleSettings
|
|
"""Get Pool autoscale settings
|
|
:param dict config: configuration object
|
|
:rtype: PoolAutoscaleSettings
|
|
:return: pool autoscale settings from specification
|
|
"""
|
|
conf = pool_specification(config)
|
|
conf = _kv_read_checked(conf, 'autoscale', {})
|
|
ei = _kv_read_checked(conf, 'evaluation_interval')
|
|
if util.is_not_empty(ei):
|
|
ei = util.convert_string_to_timedelta(ei)
|
|
else:
|
|
ei = datetime.timedelta(minutes=15)
|
|
scenconf = _kv_read_checked(conf, 'scenario')
|
|
if scenconf is not None:
|
|
mvc = _kv_read_checked(scenconf, 'maximum_vm_count')
|
|
if mvc is None:
|
|
raise ValueError('maximum_vm_count must be specified')
|
|
ndo = _kv_read_checked(
|
|
scenconf, 'node_deallocation_option', 'taskcompletion')
|
|
if (ndo is not None and
|
|
ndo not in (
|
|
'requeue', 'terminate', 'taskcompletion', 'retaineddata')):
|
|
raise ValueError(
|
|
'invalid node_deallocation_option: {}'.format(ndo))
|
|
sli = _kv_read_checked(scenconf, 'sample_lookback_interval')
|
|
if util.is_not_empty(sli):
|
|
sli = util.convert_string_to_timedelta(sli)
|
|
else:
|
|
sli = datetime.timedelta(minutes=10)
|
|
scenario = PoolAutoscaleScenarioSettings(
|
|
name=_kv_read_checked(scenconf, 'name').lower(),
|
|
maximum_vm_count=_pool_vm_count(config, conf=mvc),
|
|
node_deallocation_option=ndo,
|
|
sample_lookback_interval=sli,
|
|
required_sample_percentage=_kv_read(
|
|
scenconf, 'required_sample_percentage', 70),
|
|
rebalance_preemption_percentage=_kv_read(
|
|
scenconf, 'rebalance_preemption_percentage', None),
|
|
bias_last_sample=_kv_read(
|
|
scenconf, 'bias_last_sample', True),
|
|
bias_node_type=_kv_read_checked(
|
|
scenconf, 'bias_node_type', 'auto').lower(),
|
|
)
|
|
else:
|
|
scenario = None
|
|
return PoolAutoscaleSettings(
|
|
evaluation_interval=ei,
|
|
formula=_kv_read_checked(conf, 'formula'),
|
|
scenario=scenario,
|
|
)
|
|
|
|
|
|
def is_pool_autoscale_enabled(config, pas=None):
|
|
# type: (dict, PoolAutoscaleSettings) -> bool
|
|
"""Check if pool autoscale is enabled
|
|
:param dict config: configuration object
|
|
:param PoolAutoscaleSettings pas: pool autoscale settings
|
|
:rtype: bool
|
|
:return: if pool autoscale is enabled
|
|
"""
|
|
if pas is None:
|
|
pas = pool_autoscale_settings(config)
|
|
return util.is_not_empty(pas.formula) or pas.scenario is not None
|
|
|
|
|
|
def pool_settings(config):
|
|
# type: (dict) -> PoolSettings
|
|
"""Get Pool settings
|
|
:param dict config: configuration object
|
|
:rtype: PoolSettings
|
|
:return: pool settings from specification
|
|
"""
|
|
conf = pool_specification(config)
|
|
try:
|
|
max_tasks_per_node = conf['max_tasks_per_node']
|
|
except KeyError:
|
|
max_tasks_per_node = 1
|
|
resize_timeout = _kv_read_checked(conf, 'resize_timeout')
|
|
if util.is_not_empty(resize_timeout):
|
|
resize_timeout = util.convert_string_to_timedelta(resize_timeout)
|
|
else:
|
|
resize_timeout = None
|
|
try:
|
|
inter_node_communication_enabled = conf[
|
|
'inter_node_communication_enabled']
|
|
except KeyError:
|
|
inter_node_communication_enabled = False
|
|
try:
|
|
reboot_on_start_task_failed = conf['reboot_on_start_task_failed']
|
|
except KeyError:
|
|
reboot_on_start_task_failed = False
|
|
try:
|
|
block_until_all_gr = conf['block_until_all_global_resources_loaded']
|
|
except KeyError:
|
|
block_until_all_gr = True
|
|
try:
|
|
transfer_files_on_pool_creation = conf[
|
|
'transfer_files_on_pool_creation']
|
|
except KeyError:
|
|
transfer_files_on_pool_creation = False
|
|
try:
|
|
input_data = conf['input_data']
|
|
if util.is_none_or_empty(input_data):
|
|
raise KeyError()
|
|
except KeyError:
|
|
input_data = None
|
|
# get additional resource files
|
|
try:
|
|
rfs = conf['resource_files']
|
|
if util.is_none_or_empty(rfs):
|
|
raise KeyError()
|
|
resource_files = []
|
|
for rf in rfs:
|
|
try:
|
|
fm = rf['file_mode']
|
|
if util.is_none_or_empty(fm):
|
|
raise KeyError()
|
|
except KeyError:
|
|
fm = None
|
|
resource_files.append(
|
|
ResourceFileSettings(
|
|
file_path=rf['file_path'],
|
|
blob_source=rf['blob_source'],
|
|
file_mode=fm,
|
|
)
|
|
)
|
|
except KeyError:
|
|
resource_files = None
|
|
# ssh settings
|
|
try:
|
|
sshconf = conf['ssh']
|
|
ssh_username = _kv_read_checked(sshconf, 'username')
|
|
if util.is_none_or_empty(ssh_username):
|
|
raise KeyError()
|
|
except KeyError:
|
|
ssh_username = None
|
|
ssh_expiry_days = None
|
|
ssh_public_key = None
|
|
ssh_public_key_data = None
|
|
ssh_private_key = None
|
|
ssh_gen_docker_tunnel = None
|
|
ssh_gen_file_path = None
|
|
ssh_hpn = None
|
|
else:
|
|
ssh_expiry_days = _kv_read(sshconf, 'expiry_days', 30)
|
|
if ssh_expiry_days <= 0:
|
|
ssh_expiry_days = 30
|
|
ssh_public_key = _kv_read_checked(sshconf, 'ssh_public_key')
|
|
if util.is_not_empty(ssh_public_key):
|
|
ssh_public_key = pathlib.Path(ssh_public_key)
|
|
ssh_public_key_data = _kv_read_checked(sshconf, 'ssh_public_key_data')
|
|
ssh_private_key = _kv_read_checked(sshconf, 'ssh_private_key')
|
|
if util.is_not_empty(ssh_private_key):
|
|
ssh_private_key = pathlib.Path(ssh_private_key)
|
|
if (ssh_public_key is not None and
|
|
util.is_not_empty(ssh_public_key_data)):
|
|
raise ValueError(
|
|
'cannot specify both an SSH public key file and data')
|
|
if (ssh_public_key is None and
|
|
util.is_none_or_empty(ssh_public_key_data) and
|
|
ssh_private_key is not None):
|
|
raise ValueError(
|
|
'cannot specify an SSH private key with no public '
|
|
'key specified')
|
|
ssh_gen_docker_tunnel = _kv_read(
|
|
sshconf, 'generate_docker_tunnel_script', False)
|
|
ssh_gen_file_path = _kv_read_checked(
|
|
sshconf, 'generated_file_export_path', '.')
|
|
ssh_hpn = _kv_read(sshconf, 'hpn_server_swap', False)
|
|
try:
|
|
gpu_driver = conf['gpu']['nvidia_driver']['source']
|
|
if util.is_none_or_empty(gpu_driver):
|
|
raise KeyError()
|
|
except KeyError:
|
|
gpu_driver = None
|
|
try:
|
|
additional_node_prep_commands = conf['additional_node_prep_commands']
|
|
if util.is_none_or_empty(additional_node_prep_commands):
|
|
raise KeyError()
|
|
except KeyError:
|
|
additional_node_prep_commands = []
|
|
return PoolSettings(
|
|
id=conf['id'],
|
|
vm_size=conf['vm_size'].lower(), # normalize
|
|
vm_count=_pool_vm_count(config),
|
|
resize_timeout=resize_timeout,
|
|
max_tasks_per_node=max_tasks_per_node,
|
|
inter_node_communication_enabled=inter_node_communication_enabled,
|
|
vm_configuration=_populate_pool_vm_configuration(config),
|
|
reboot_on_start_task_failed=reboot_on_start_task_failed,
|
|
block_until_all_global_resources_loaded=block_until_all_gr,
|
|
transfer_files_on_pool_creation=transfer_files_on_pool_creation,
|
|
input_data=input_data,
|
|
resource_files=resource_files,
|
|
ssh=SSHSettings(
|
|
username=ssh_username,
|
|
expiry_days=ssh_expiry_days,
|
|
ssh_public_key=ssh_public_key,
|
|
ssh_public_key_data=ssh_public_key_data,
|
|
ssh_private_key=ssh_private_key,
|
|
generate_docker_tunnel_script=ssh_gen_docker_tunnel,
|
|
generated_file_export_path=ssh_gen_file_path,
|
|
hpn_server_swap=ssh_hpn,
|
|
),
|
|
gpu_driver=gpu_driver,
|
|
additional_node_prep_commands=additional_node_prep_commands,
|
|
virtual_network=virtual_network_settings(
|
|
conf,
|
|
default_existing_ok=True,
|
|
default_create_nonexistant=False,
|
|
),
|
|
autoscale=pool_autoscale_settings(config),
|
|
node_fill_type=_kv_read_checked(conf, 'node_fill_type'),
|
|
)
|
|
|
|
|
|
def set_block_until_all_global_resources_loaded(config, flag):
|
|
# type: (dict, bool) -> None
|
|
"""Set block until all global resources setting
|
|
:param dict config: configuration object
|
|
:param bool flag: flag to set
|
|
"""
|
|
config['pool_specification'][
|
|
'block_until_all_global_resources_loaded'] = flag
|
|
|
|
|
|
def set_inter_node_communication_enabled(config, flag):
|
|
# type: (dict, bool) -> None
|
|
"""Set inter node comm setting
|
|
:param dict config: configuration object
|
|
:param bool flag: flag to set
|
|
"""
|
|
config['pool_specification']['inter_node_communication_enabled'] = flag
|
|
|
|
|
|
def set_ssh_public_key(config, pubkey):
|
|
# type: (dict, str) -> None
|
|
"""Set SSH public key setting
|
|
:param dict config: configuration object
|
|
:param str pubkey: public key to set
|
|
"""
|
|
if 'ssh' not in config['pool_specification']:
|
|
config['pool_specification']['ssh'] = {}
|
|
config['pool_specification']['ssh']['ssh_public_key'] = pubkey
|
|
|
|
|
|
def set_hpn_server_swap(config, flag):
|
|
# type: (dict, bool) -> None
|
|
"""Set SSH HPN server swap setting
|
|
:param dict config: configuration object
|
|
:param bool flag: flag to set
|
|
"""
|
|
if 'ssh' not in config['pool_specification']:
|
|
config['pool_specification']['ssh'] = {}
|
|
config['pool_specification']['ssh']['hpn_server_swap'] = flag
|
|
|
|
|
|
def pool_id(config, lower=False):
|
|
# type: (dict, bool) -> str
|
|
"""Get Pool id
|
|
:param dict config: configuration object
|
|
:param bool lower: lowercase return
|
|
:rtype: str
|
|
:return: pool id
|
|
"""
|
|
id = config['pool_specification']['id']
|
|
return id.lower() if lower else id
|
|
|
|
|
|
def pool_publisher(config, lower=False):
|
|
# type: (dict, bool) -> str
|
|
"""Get Pool publisher
|
|
:param dict config: configuration object
|
|
:param bool lower: lowercase return
|
|
:rtype: str
|
|
:return: pool publisher
|
|
"""
|
|
conf = pool_vm_configuration(config, 'platform_image')
|
|
pub = _kv_read_checked(conf, 'publisher')
|
|
return pub.lower() if lower and util.is_not_empty(pub) else pub
|
|
|
|
|
|
def pool_offer(config, lower=False):
|
|
# type: (dict, bool) -> str
|
|
"""Get Pool offer
|
|
:param dict config: configuration object
|
|
:param bool lower: lowercase return
|
|
:rtype: str
|
|
:return: pool offer
|
|
"""
|
|
conf = pool_vm_configuration(config, 'platform_image')
|
|
offer = _kv_read_checked(conf, 'offer')
|
|
return offer.lower() if lower and util.is_not_empty(offer) else offer
|
|
|
|
|
|
def pool_sku(config, lower=False):
|
|
# type: (dict, bool) -> str
|
|
"""Get Pool sku
|
|
:param dict config: configuration object
|
|
:param bool lower: lowercase return
|
|
:rtype: str
|
|
:return: pool sku
|
|
"""
|
|
conf = pool_vm_configuration(config, 'platform_image')
|
|
sku = _kv_read_checked(conf, 'sku')
|
|
return sku.lower() if lower and util.is_not_empty(sku) else sku
|
|
|
|
|
|
def pool_custom_image_node_agent(config):
|
|
# type: (dict) -> str
|
|
"""Get Pool node agent from custom image
|
|
:param dict config: configuration object
|
|
:rtype: str
|
|
:return: pool node agent
|
|
"""
|
|
conf = pool_vm_configuration(config, 'custom_image')
|
|
return _kv_read_checked(conf, 'node_agent')
|
|
|
|
|
|
# CREDENTIALS SETTINGS
|
|
def raw_credentials(config, omit_keyvault):
|
|
# type: (dict, bool) -> dict
|
|
"""Get raw credentials dictionary
|
|
:param dict config: configuration object
|
|
:param bool omit_keyvault: omit keyvault settings if present
|
|
:rtype: dict
|
|
:return: credentials dict
|
|
"""
|
|
conf = config['credentials']
|
|
if omit_keyvault:
|
|
conf.pop('keyvault', None)
|
|
return conf
|
|
|
|
|
|
def _aad_credentials(
|
|
conf, default_endpoint=None, default_token_cache_file=None):
|
|
# type: (dict, str) -> AADSettings
|
|
"""Retrieve AAD Settings
|
|
:param dict config: configuration object
|
|
:param str default_endpoint: default endpoint
|
|
:param str default_token_cache_file: default token cache file
|
|
:rtype: AADSettings
|
|
:return: AAD settings
|
|
"""
|
|
if 'aad' in conf:
|
|
aad_directory_id = _kv_read_checked(conf['aad'], 'directory_id')
|
|
aad_application_id = _kv_read_checked(conf['aad'], 'application_id')
|
|
aad_auth_key = _kv_read_checked(conf['aad'], 'auth_key')
|
|
aad_user = _kv_read_checked(conf['aad'], 'user')
|
|
aad_password = _kv_read_checked(conf['aad'], 'password')
|
|
aad_cert_private_key = _kv_read_checked(
|
|
conf['aad'], 'rsa_private_key_pem')
|
|
aad_cert_thumbprint = _kv_read_checked(
|
|
conf['aad'], 'x509_cert_sha1_thumbprint')
|
|
aad_endpoint = _kv_read_checked(
|
|
conf['aad'], 'endpoint', default_endpoint)
|
|
if 'token_cache' not in conf['aad']:
|
|
conf['aad']['token_cache'] = {}
|
|
token_cache_enabled = _kv_read(
|
|
conf['aad']['token_cache'], 'enabled', True)
|
|
if token_cache_enabled:
|
|
token_cache_file = _kv_read_checked(
|
|
conf['aad']['token_cache'], 'filename',
|
|
default_token_cache_file)
|
|
else:
|
|
token_cache_file = None
|
|
return AADSettings(
|
|
directory_id=aad_directory_id,
|
|
application_id=aad_application_id,
|
|
auth_key=aad_auth_key,
|
|
user=aad_user,
|
|
password=aad_password,
|
|
rsa_private_key_pem=aad_cert_private_key,
|
|
x509_cert_sha1_thumbprint=aad_cert_thumbprint,
|
|
endpoint=aad_endpoint,
|
|
token_cache_file=token_cache_file,
|
|
)
|
|
else:
|
|
return AADSettings(
|
|
directory_id=None,
|
|
application_id=None,
|
|
auth_key=None,
|
|
user=None,
|
|
password=None,
|
|
rsa_private_key_pem=None,
|
|
x509_cert_sha1_thumbprint=None,
|
|
endpoint=default_endpoint,
|
|
token_cache_file=None,
|
|
)
|
|
|
|
|
|
def credentials_keyvault(config):
|
|
# type: (dict) -> KeyVaultCredentialsSettings
|
|
"""Get KeyVault settings
|
|
:param dict config: configuration object
|
|
:rtype: KeyVaultCredentialsSettings
|
|
:return: Key Vault settings
|
|
"""
|
|
try:
|
|
conf = config['credentials']['keyvault']
|
|
except (KeyError, TypeError):
|
|
conf = {}
|
|
keyvault_uri = _kv_read_checked(conf, 'uri')
|
|
keyvault_credentials_secret_id = _kv_read_checked(
|
|
conf, 'credentials_secret_id')
|
|
return KeyVaultCredentialsSettings(
|
|
aad=_aad_credentials(
|
|
conf,
|
|
default_endpoint='https://vault.azure.net',
|
|
default_token_cache_file=(
|
|
'.batch_shipyard_aad_keyvault_token.json'
|
|
),
|
|
),
|
|
keyvault_uri=keyvault_uri,
|
|
keyvault_credentials_secret_id=keyvault_credentials_secret_id,
|
|
)
|
|
|
|
|
|
def credentials_management(config):
|
|
# type: (dict) -> ManagementCredentialsSettings
|
|
"""Get Management settings
|
|
:param dict config: configuration object
|
|
:rtype: ManagementCredentialsSettings
|
|
:return: Management settings
|
|
"""
|
|
try:
|
|
conf = config['credentials']['management']
|
|
except (KeyError, TypeError):
|
|
conf = {}
|
|
subscription_id = _kv_read_checked(conf, 'subscription_id')
|
|
return ManagementCredentialsSettings(
|
|
aad=_aad_credentials(
|
|
conf,
|
|
default_endpoint='https://management.core.windows.net/',
|
|
default_token_cache_file=(
|
|
'.batch_shipyard_aad_management_token.json'
|
|
),
|
|
),
|
|
subscription_id=subscription_id,
|
|
)
|
|
|
|
|
|
def credentials_batch(config):
|
|
# type: (dict) -> BatchCredentialsSettings
|
|
"""Get Batch credentials
|
|
:param dict config: configuration object
|
|
:rtype: BatchCredentialsSettings
|
|
:return: batch creds
|
|
"""
|
|
conf = config['credentials']['batch']
|
|
account = _kv_read_checked(conf, 'account')
|
|
account_key = _kv_read_checked(conf, 'account_key')
|
|
account_service_url = conf['account_service_url']
|
|
resource_group = _kv_read_checked(conf, 'resource_group')
|
|
# get subscription id from management section
|
|
try:
|
|
subscription_id = _kv_read_checked(
|
|
config['credentials']['management'], 'subscription_id')
|
|
except (KeyError, TypeError):
|
|
subscription_id = None
|
|
# parse location from url
|
|
tmp = account_service_url.split('.')
|
|
location = tmp[1]
|
|
acct_from_url = tmp[0].split('/')[-1]
|
|
if util.is_none_or_empty(account):
|
|
account = acct_from_url
|
|
else:
|
|
# ensure url account matches account name
|
|
if account != acct_from_url:
|
|
raise ValueError(
|
|
('Specified account {} is a mismatch with service '
|
|
'url {}').format(account, account_service_url))
|
|
return BatchCredentialsSettings(
|
|
aad=_aad_credentials(
|
|
conf,
|
|
default_endpoint='https://batch.core.windows.net/',
|
|
default_token_cache_file=(
|
|
'.batch_shipyard_aad_batch_token.json'
|
|
),
|
|
),
|
|
account=account,
|
|
account_key=account_key,
|
|
account_service_url=conf['account_service_url'],
|
|
resource_group=resource_group,
|
|
location=location,
|
|
subscription_id=subscription_id,
|
|
)
|
|
|
|
|
|
def credentials_batch_account_key_secret_id(config):
|
|
# type: (dict) -> str
|
|
"""Get Batch account key KeyVault Secret Id
|
|
:param dict config: configuration object
|
|
:rtype: str
|
|
:return: keyvault secret id
|
|
"""
|
|
try:
|
|
secid = config[
|
|
'credentials']['batch']['account_key_keyvault_secret_id']
|
|
if util.is_none_or_empty(secid):
|
|
raise KeyError()
|
|
except KeyError:
|
|
return None
|
|
return secid
|
|
|
|
|
|
def set_credentials_batch_account_key(config, bakey):
|
|
# type: (dict, str) -> None
|
|
"""Set Batch account key
|
|
:param dict config: configuration object
|
|
:param str bakey: batch account key
|
|
"""
|
|
config['credentials']['batch']['account_key'] = bakey
|
|
|
|
|
|
def credentials_storage(config, ssel):
|
|
# type: (dict, str) -> StorageCredentialsSettings
|
|
"""Get specific storage credentials
|
|
:param dict config: configuration object
|
|
:param str ssel: storage selector link
|
|
:rtype: StorageCredentialsSettings
|
|
:return: storage creds
|
|
"""
|
|
conf = config['credentials']['storage'][ssel]
|
|
try:
|
|
ep = conf['endpoint']
|
|
if util.is_none_or_empty(ep):
|
|
raise KeyError()
|
|
except KeyError:
|
|
ep = 'core.windows.net'
|
|
return StorageCredentialsSettings(
|
|
account=conf['account'],
|
|
account_key=conf['account_key'],
|
|
endpoint=ep,
|
|
)
|
|
|
|
|
|
def iterate_storage_credentials(config):
|
|
# type: (dict) -> str
|
|
"""Iterate storage credential storage select links
|
|
:param dict config: configuration object
|
|
:rtype: str
|
|
:return: storage selector link
|
|
"""
|
|
for conf in config['credentials']['storage']:
|
|
yield conf
|
|
|
|
|
|
def credentials_storage_account_key_secret_id(config, ssel):
|
|
# type: (dict, str) -> str
|
|
"""Get Storage account key KeyVault Secret Id
|
|
:param dict config: configuration object
|
|
:param str ssel: storage selector link
|
|
:rtype: str
|
|
:return: keyvault secret id
|
|
"""
|
|
try:
|
|
secid = config[
|
|
'credentials']['storage'][ssel]['account_key_keyvault_secret_id']
|
|
if util.is_none_or_empty(secid):
|
|
raise KeyError()
|
|
except KeyError:
|
|
return None
|
|
return secid
|
|
|
|
|
|
def set_credentials_storage_account_key(config, ssel, sakey):
|
|
# type: (dict, str, str) -> None
|
|
"""Set Storage account key
|
|
:param dict config: configuration object
|
|
:param str ssel: storage selector link
|
|
:param str sakey: storage account key
|
|
"""
|
|
config['credentials']['storage'][ssel]['account_key'] = sakey
|
|
|
|
|
|
def docker_registry_login(config, server):
|
|
# type: (dict, str) -> tuple
|
|
"""Get docker registry login settings
|
|
:param dict config: configuration object
|
|
:param str server: credentials for login server to retrieve
|
|
:rtype: tuple
|
|
:return: (user, pw)
|
|
"""
|
|
try:
|
|
user = config['credentials']['docker_registry'][server]['username']
|
|
pw = config['credentials']['docker_registry'][server]['password']
|
|
if util.is_none_or_empty(user) or util.is_none_or_empty(pw):
|
|
raise KeyError()
|
|
except KeyError:
|
|
user = None
|
|
pw = None
|
|
return user, pw
|
|
|
|
|
|
def iterate_docker_registry_servers(config):
|
|
# type: (dict) -> str
|
|
"""Iterate docker registry servers
|
|
:param dict config: configuration object
|
|
:rtype: str
|
|
:return: docker registry name
|
|
"""
|
|
try:
|
|
for conf in config['credentials']['docker_registry']:
|
|
yield conf
|
|
except KeyError:
|
|
pass
|
|
|
|
|
|
def credentials_docker_registry_password_secret_id(config, dr):
|
|
# type: (dict, str) -> str
|
|
"""Get Docker registry password KeyVault Secret Id
|
|
:param dict config: configuration object
|
|
:param str dr: docker registry link
|
|
:rtype: str
|
|
:return: keyvault secret id
|
|
"""
|
|
try:
|
|
secid = config['credentials'][
|
|
'docker_registry'][dr]['password_keyvault_secret_id']
|
|
if util.is_none_or_empty(secid):
|
|
raise KeyError()
|
|
except KeyError:
|
|
return None
|
|
return secid
|
|
|
|
|
|
def set_credentials_docker_registry_password(config, dr, password):
|
|
# type: (dict, str, str) -> None
|
|
"""Set Docker registry password
|
|
:param dict config: configuration object
|
|
:param str dr: docker registry link
|
|
:param str password: password
|
|
"""
|
|
config['credentials']['docker_registry'][dr]['password'] = password
|
|
|
|
|
|
# GLOBAL SETTINGS
|
|
def batch_shipyard_settings(config):
|
|
# type: (dict) -> BatchShipyardSettings
|
|
"""Get batch shipyard settings
|
|
:param dict config: configuration object
|
|
:rtype: BatchShipyardSettings
|
|
:return: batch shipyard settings
|
|
"""
|
|
conf = config['batch_shipyard']
|
|
stlink = conf['storage_account_settings']
|
|
if util.is_none_or_empty(stlink):
|
|
raise ValueError('batch_shipyard:storage_account_settings is invalid')
|
|
try:
|
|
sep = conf['storage_entity_prefix']
|
|
if sep is None:
|
|
raise KeyError()
|
|
except KeyError:
|
|
sep = 'shipyard'
|
|
try:
|
|
sasexpiry = conf['generated_sas_expiry_days']
|
|
except KeyError:
|
|
sasexpiry = None
|
|
try:
|
|
use_shipyard_image = conf['use_shipyard_docker_image']
|
|
except KeyError:
|
|
use_shipyard_image = True
|
|
try:
|
|
store_timing = conf['store_timing_metrics']
|
|
except KeyError:
|
|
store_timing = False
|
|
return BatchShipyardSettings(
|
|
storage_account_settings=stlink,
|
|
storage_entity_prefix=sep,
|
|
generated_sas_expiry_days=sasexpiry,
|
|
use_shipyard_docker_image=use_shipyard_image,
|
|
store_timing_metrics=store_timing,
|
|
)
|
|
|
|
|
|
def set_use_shipyard_docker_image(config, flag):
|
|
# type: (dict, bool) -> None
|
|
"""Set shipyard docker image use
|
|
:param dict config: configuration object
|
|
:param bool flag: flag to set
|
|
"""
|
|
config['batch_shipyard']['use_shipyard_docker_image'] = flag
|
|
|
|
|
|
def batch_shipyard_encryption_enabled(config):
|
|
# type: (dict) -> bool
|
|
"""Get credential encryption enabled setting
|
|
:param dict config: configuration object
|
|
:rtype: bool
|
|
:return: if credential encryption is enabled
|
|
"""
|
|
try:
|
|
encrypt = config['batch_shipyard']['encryption']['enabled']
|
|
except KeyError:
|
|
encrypt = False
|
|
return encrypt
|
|
|
|
|
|
def set_batch_shipyard_encryption_enabled(config, flag):
|
|
# type: (dict, bool) -> None
|
|
"""Set credential encryption enabled setting
|
|
:param dict config: configuration object
|
|
:param bool flag: flag to set
|
|
"""
|
|
if 'encryption' not in config['batch_shipyard']:
|
|
config['batch_shipyard']['encryption'] = {}
|
|
config['batch_shipyard']['encryption']['enabled'] = flag
|
|
|
|
|
|
def batch_shipyard_encryption_pfx_filename(config):
|
|
# type: (dict) -> str
|
|
"""Get filename of pfx cert
|
|
:param dict config: configuration object
|
|
:rtype: str
|
|
:return: pfx filename
|
|
"""
|
|
try:
|
|
pfxfile = config['batch_shipyard']['encryption']['pfx']['filename']
|
|
except KeyError:
|
|
pfxfile = None
|
|
return pfxfile
|
|
|
|
|
|
def batch_shipyard_encryption_pfx_passphrase(config):
|
|
# type: (dict) -> str
|
|
"""Get passphrase of pfx cert
|
|
:param dict config: configuration object
|
|
:rtype: str
|
|
:return: pfx passphrase
|
|
"""
|
|
try:
|
|
passphrase = config['batch_shipyard']['encryption'][
|
|
'pfx']['passphrase']
|
|
except KeyError:
|
|
passphrase = None
|
|
return passphrase
|
|
|
|
|
|
def batch_shipyard_encryption_pfx_sha1_thumbprint(config):
|
|
# type: (dict) -> str
|
|
"""Get sha1 tp of pfx cert
|
|
:param dict config: configuration object
|
|
:rtype: str
|
|
:return: pfx sha1 thumbprint
|
|
"""
|
|
try:
|
|
tp = config['batch_shipyard']['encryption']['pfx']['sha1_thumbprint']
|
|
except KeyError:
|
|
tp = None
|
|
return tp
|
|
|
|
|
|
def set_batch_shipyard_encryption_pfx_sha1_thumbprint(config, tp):
|
|
# type: (dict, str) -> None
|
|
"""Set sha1 tp of pfx cert
|
|
:param dict config: configuration object
|
|
"""
|
|
config['batch_shipyard']['encryption']['pfx']['sha1_thumbprint'] = tp
|
|
|
|
|
|
def batch_shipyard_encryption_public_key_pem(config):
|
|
# type: (dict) -> str
|
|
"""Get filename of pem public key
|
|
:param dict config: configuration object
|
|
:rtype: str
|
|
:return: pem filename
|
|
"""
|
|
try:
|
|
pem = config['batch_shipyard']['encryption']['public_key_pem']
|
|
except KeyError:
|
|
pem = None
|
|
return pem
|
|
|
|
|
|
def docker_registry_private_settings(config):
|
|
# type: (dict) -> DockerRegistrySettings
|
|
"""Get docker private registry backed to azure storage settings
|
|
:param dict config: configuration object
|
|
:rtype: DockerRegistrySettings
|
|
:return: docker registry settings
|
|
"""
|
|
try:
|
|
pregpubpull = config['docker_registry']['private'][
|
|
'allow_public_docker_hub_pull_on_missing']
|
|
except KeyError:
|
|
pregpubpull = False
|
|
try:
|
|
server = config['docker_registry']['private']['server']
|
|
if util.is_none_or_empty(server):
|
|
raise KeyError()
|
|
server = server.split(':')
|
|
if len(server) == 1:
|
|
port = 80
|
|
elif len(server) == 2:
|
|
port = int(server[1])
|
|
else:
|
|
raise ValueError('invalid docker registry server specification')
|
|
server = server[0]
|
|
# get login
|
|
user, pw = docker_registry_login(config, server)
|
|
if util.is_none_or_empty(user) or util.is_none_or_empty(pw):
|
|
raise ValueError(
|
|
'Docker registry login settings not specified for: {}'.format(
|
|
server))
|
|
except KeyError:
|
|
server = None
|
|
port = None
|
|
user = None
|
|
pw = None
|
|
try:
|
|
sa = config['docker_registry']['private']['azure_storage'][
|
|
'storage_account_settings']
|
|
if util.is_none_or_empty(sa):
|
|
raise KeyError()
|
|
cont = config['docker_registry']['private']['azure_storage'][
|
|
'container']
|
|
if util.is_none_or_empty(cont):
|
|
raise KeyError()
|
|
except KeyError:
|
|
sa = None
|
|
cont = None
|
|
if server is not None and sa is not None:
|
|
raise ValueError(
|
|
'cannot specify both a private registry server host and a '
|
|
'private registry backed by Azure Storage')
|
|
return DockerRegistrySettings(
|
|
allow_public_docker_hub_pull_on_missing=pregpubpull,
|
|
storage_account=sa,
|
|
container=cont,
|
|
server=server,
|
|
port=port,
|
|
user=user,
|
|
password=pw,
|
|
)
|
|
|
|
|
|
def data_replication_settings(config):
|
|
# type: (dict) -> DataReplicationSettings
|
|
"""Get data replication settings
|
|
:param dict config: configuration object
|
|
:rtype: DataReplicationSettings
|
|
:return: data replication settings
|
|
"""
|
|
try:
|
|
conf = config['data_replication']
|
|
except KeyError:
|
|
conf = {}
|
|
try:
|
|
concurrent_source_downloads = conf['concurrent_source_downloads']
|
|
except KeyError:
|
|
concurrent_source_downloads = 10
|
|
try:
|
|
conf = config['data_replication']['peer_to_peer']
|
|
except KeyError:
|
|
conf = {}
|
|
try:
|
|
p2p_enabled = conf['enabled']
|
|
except KeyError:
|
|
p2p_enabled = False
|
|
try:
|
|
p2p_compression = conf['compression']
|
|
except KeyError:
|
|
p2p_compression = True
|
|
pool_vm_count = _pool_vm_count(config)
|
|
total_vm_count = pool_vm_count.dedicated + pool_vm_count.low_priority
|
|
try:
|
|
p2p_direct_download_seed_bias = conf['direct_download_seed_bias']
|
|
if (p2p_direct_download_seed_bias is None or
|
|
p2p_direct_download_seed_bias < 1):
|
|
raise KeyError()
|
|
except KeyError:
|
|
p2p_direct_download_seed_bias = total_vm_count // 10
|
|
if p2p_direct_download_seed_bias < 1:
|
|
p2p_direct_download_seed_bias = 1
|
|
return DataReplicationSettings(
|
|
peer_to_peer=PeerToPeerSettings(
|
|
enabled=p2p_enabled,
|
|
compression=p2p_compression,
|
|
direct_download_seed_bias=p2p_direct_download_seed_bias
|
|
),
|
|
concurrent_source_downloads=concurrent_source_downloads,
|
|
)
|
|
|
|
|
|
def set_peer_to_peer_enabled(config, flag):
|
|
# type: (dict, bool) -> None
|
|
"""Set peer to peer enabled setting
|
|
:param dict config: configuration object
|
|
:param bool flag: flag to set
|
|
"""
|
|
if 'data_replication' not in config:
|
|
config['data_replication'] = {}
|
|
if 'peer_to_peer' not in config['data_replication']:
|
|
config['data_replication']['peer_to_peer'] = {}
|
|
config['data_replication']['peer_to_peer']['enabled'] = flag
|
|
|
|
|
|
def global_resources_docker_images(config):
|
|
# type: (dict) -> list
|
|
"""Get list of docker images
|
|
:param dict config: configuration object
|
|
:rtype: list
|
|
:return: docker images
|
|
"""
|
|
try:
|
|
images = config['global_resources']['docker_images']
|
|
if util.is_none_or_empty(images):
|
|
raise KeyError()
|
|
except KeyError:
|
|
images = []
|
|
return images
|
|
|
|
|
|
def global_resources_files(config):
|
|
# type: (dict) -> list
|
|
"""Get list of global files ingress
|
|
:param dict config: configuration object
|
|
:rtype: list
|
|
:return: global files ingress list
|
|
"""
|
|
try:
|
|
files = config['global_resources']['files']
|
|
if util.is_none_or_empty(files):
|
|
raise KeyError()
|
|
except KeyError:
|
|
files = []
|
|
return files
|
|
|
|
|
|
def is_direct_transfer(filespair):
|
|
# type: (dict) -> bool
|
|
"""Determine if src/dst pair for files ingress is a direct compute node
|
|
transfer
|
|
:param dict filespair: src/dst pair
|
|
:rtype: bool
|
|
:return: if ingress is direct
|
|
"""
|
|
return 'storage_account_settings' not in filespair['destination']
|
|
|
|
|
|
def files_source_settings(conf):
|
|
# type: (dict) -> SourceSettings
|
|
"""Get global resources files source
|
|
:param dict conf: configuration block
|
|
:rtype: SourceSettings
|
|
:return: source settings
|
|
"""
|
|
path = conf['source']['path']
|
|
if util.is_none_or_empty(path):
|
|
raise ValueError('global resource files path is invalid')
|
|
try:
|
|
include = conf['source']['include']
|
|
if util.is_none_or_empty(include):
|
|
raise KeyError()
|
|
except KeyError:
|
|
include = None
|
|
try:
|
|
exclude = conf['source']['exclude']
|
|
if util.is_none_or_empty(exclude):
|
|
raise KeyError()
|
|
except KeyError:
|
|
exclude = None
|
|
return SourceSettings(path=path, include=include, exclude=exclude)
|
|
|
|
|
|
def files_destination_settings(fdict):
|
|
# type: (dict) -> DestinationSettings
|
|
"""Get global resources files destination
|
|
:param dict fdict: configuration block
|
|
:rtype: DestinationSettings
|
|
:return: destination settings
|
|
"""
|
|
conf = fdict['destination']
|
|
try:
|
|
shared = conf['shared_data_volume']
|
|
except KeyError:
|
|
shared = None
|
|
try:
|
|
storage = conf['storage_account_settings']
|
|
except KeyError:
|
|
storage = None
|
|
try:
|
|
rdp = conf['relative_destination_path']
|
|
if rdp is not None:
|
|
rdp = rdp.lstrip('/').rstrip('/')
|
|
if len(rdp) == 0:
|
|
rdp = None
|
|
except KeyError:
|
|
rdp = None
|
|
try:
|
|
method = conf['data_transfer']['method'].lower()
|
|
except KeyError:
|
|
if storage is None:
|
|
raise RuntimeError(
|
|
'no transfer method specified for data transfer of '
|
|
'source: {} to {} rdp={}'.format(
|
|
files_source_settings(fdict).path, shared, rdp))
|
|
else:
|
|
method = None
|
|
try:
|
|
ssh_eo = conf['data_transfer']['scp_ssh_extra_options']
|
|
if ssh_eo is None:
|
|
raise KeyError()
|
|
except KeyError:
|
|
ssh_eo = ''
|
|
try:
|
|
rsync_eo = conf['data_transfer']['rsync_extra_options']
|
|
if rsync_eo is None:
|
|
raise KeyError()
|
|
except KeyError:
|
|
rsync_eo = ''
|
|
try:
|
|
mpt = conf['data_transfer']['max_parallel_transfers_per_node']
|
|
if mpt is not None and mpt <= 0:
|
|
raise KeyError()
|
|
except KeyError:
|
|
mpt = None
|
|
# ensure valid mpt number
|
|
if mpt is None:
|
|
mpt = 1
|
|
try:
|
|
split = conf['data_transfer']['split_files_megabytes']
|
|
if split is not None and split <= 0:
|
|
raise KeyError()
|
|
# convert to bytes
|
|
if split is not None:
|
|
split <<= 20
|
|
except KeyError:
|
|
split = None
|
|
ssh_private_key = _kv_read_checked(
|
|
conf['data_transfer'], 'ssh_private_key')
|
|
if util.is_not_empty(ssh_private_key):
|
|
ssh_private_key = pathlib.Path(ssh_private_key)
|
|
try:
|
|
container = conf['data_transfer']['container']
|
|
if util.is_none_or_empty(container):
|
|
raise KeyError()
|
|
except KeyError:
|
|
container = None
|
|
try:
|
|
fshare = conf['data_transfer']['file_share']
|
|
if util.is_none_or_empty(fshare):
|
|
raise KeyError()
|
|
except KeyError:
|
|
fshare = None
|
|
try:
|
|
bx_eo = conf['data_transfer']['blobxfer_extra_options']
|
|
if bx_eo is None:
|
|
bx_eo = ''
|
|
except KeyError:
|
|
bx_eo = ''
|
|
return DestinationSettings(
|
|
storage_account_settings=storage,
|
|
shared_data_volume=shared,
|
|
relative_destination_path=rdp,
|
|
data_transfer=DataTransferSettings(
|
|
container=container,
|
|
file_share=fshare,
|
|
blobxfer_extra_options=bx_eo,
|
|
method=method,
|
|
ssh_private_key=ssh_private_key,
|
|
scp_ssh_extra_options=ssh_eo,
|
|
rsync_extra_options=rsync_eo,
|
|
split_files_megabytes=split,
|
|
max_parallel_transfers_per_node=mpt,
|
|
)
|
|
)
|
|
|
|
|
|
def global_resources_data_volumes(config):
|
|
# type: (dict) -> dict
|
|
"""Get data volumes dictionary
|
|
:param dict config: configuration object
|
|
:rtype: dict
|
|
:return: data volumes
|
|
"""
|
|
try:
|
|
dv = config['global_resources']['docker_volumes']['data_volumes']
|
|
if util.is_none_or_empty(dv):
|
|
raise KeyError()
|
|
except KeyError:
|
|
dv = {}
|
|
return dv
|
|
|
|
|
|
def global_resources_shared_data_volumes(config):
|
|
# type: (dict) -> dict
|
|
"""Get shared data volumes dictionary
|
|
:param dict config: configuration object
|
|
:rtype: dict
|
|
:return: shared data volumes
|
|
"""
|
|
try:
|
|
sdv = config['global_resources']['docker_volumes'][
|
|
'shared_data_volumes']
|
|
if util.is_none_or_empty(sdv):
|
|
raise KeyError()
|
|
except KeyError:
|
|
sdv = {}
|
|
return sdv
|
|
|
|
|
|
def shared_data_volume_driver(sdv, sdvkey):
|
|
# type: (dict, str) -> str
|
|
"""Get shared data volume driver
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: str
|
|
:return: volume driver
|
|
"""
|
|
return sdv[sdvkey]['volume_driver']
|
|
|
|
|
|
def shared_data_volume_container_path(sdv, sdvkey):
|
|
# type: (dict, str) -> str
|
|
"""Get shared data volume container path
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: str
|
|
:return: container path
|
|
"""
|
|
return sdv[sdvkey]['container_path']
|
|
|
|
|
|
def shared_data_volume_mount_options(sdv, sdvkey):
|
|
# type: (dict, str) -> str
|
|
"""Get shared data volume mount options
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: str
|
|
:return: shared data volume mount options
|
|
"""
|
|
try:
|
|
mo = sdv[sdvkey]['mount_options']
|
|
except KeyError:
|
|
mo = None
|
|
return mo
|
|
|
|
|
|
def azure_file_storage_account_settings(sdv, sdvkey):
|
|
# type: (dict, str) -> str
|
|
"""Get azure file storage account link
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: str
|
|
:return: storage account link
|
|
"""
|
|
return sdv[sdvkey]['storage_account_settings']
|
|
|
|
|
|
def azure_file_share_name(sdv, sdvkey):
|
|
# type: (dict, str) -> str
|
|
"""Get azure file share name
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: str
|
|
:return: azure file share name
|
|
"""
|
|
return sdv[sdvkey]['azure_file_share_name']
|
|
|
|
|
|
def gluster_volume_type(sdv, sdvkey):
|
|
# type: (dict, str) -> str
|
|
"""Get gluster volume type
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: str
|
|
:return: gluster volume type
|
|
"""
|
|
try:
|
|
vt = sdv[sdvkey]['volume_type']
|
|
if util.is_none_or_empty(vt):
|
|
raise KeyError()
|
|
except KeyError:
|
|
vt = 'replica'
|
|
return vt
|
|
|
|
|
|
def gluster_volume_options(sdv, sdvkey):
|
|
# type: (dict, str) -> str
|
|
"""Get gluster volume options
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: str
|
|
:return: gluster volume options
|
|
"""
|
|
try:
|
|
vo = sdv[sdvkey]['volume_options']
|
|
if util.is_none_or_empty(vo):
|
|
raise KeyError()
|
|
except KeyError:
|
|
vo = None
|
|
return vo
|
|
|
|
|
|
def is_shared_data_volume_azure_file(sdv, sdvkey):
|
|
# type: (dict, str) -> bool
|
|
"""Determine if shared data volume is an azure file share
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: bool
|
|
:return: if shared data volume is azure file
|
|
"""
|
|
return shared_data_volume_driver(sdv, sdvkey).lower() == 'azurefile'
|
|
|
|
|
|
def is_shared_data_volume_gluster_on_compute(sdv, sdvkey):
|
|
# type: (dict, str) -> bool
|
|
"""Determine if shared data volume is a glusterfs share on compute
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: bool
|
|
:return: if shared data volume is glusterfs on compute
|
|
"""
|
|
return shared_data_volume_driver(
|
|
sdv, sdvkey).lower() == 'glusterfs_on_compute'
|
|
|
|
|
|
def is_shared_data_volume_storage_cluster(sdv, sdvkey):
|
|
# type: (dict, str) -> bool
|
|
"""Determine if shared data volume is a storage cluster
|
|
:param dict sdv: shared_data_volume configuration object
|
|
:param str sdvkey: key to sdv
|
|
:rtype: bool
|
|
:return: if shared data volume is storage_cluster
|
|
"""
|
|
return shared_data_volume_driver(sdv, sdvkey).lower() == 'storage_cluster'
|
|
|
|
|
|
# INPUT AND OUTPUT DATA SETTINGS
|
|
def input_data(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve input data config block
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: input data config block
|
|
"""
|
|
try:
|
|
id = conf['input_data']
|
|
if util.is_none_or_empty(id):
|
|
raise KeyError()
|
|
except KeyError:
|
|
id = None
|
|
return id
|
|
|
|
|
|
def output_data(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve output data config block
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: output data config block
|
|
"""
|
|
try:
|
|
od = conf['output_data']
|
|
if util.is_none_or_empty(od):
|
|
raise KeyError()
|
|
except KeyError:
|
|
od = None
|
|
return od
|
|
|
|
|
|
def data_storage_account_settings(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve input data storage account settings link
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: storage account link
|
|
"""
|
|
return conf['storage_account_settings']
|
|
|
|
|
|
def data_container(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve input data blob container name
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: container name
|
|
"""
|
|
try:
|
|
container = conf['container']
|
|
if util.is_none_or_empty(container):
|
|
raise KeyError()
|
|
except KeyError:
|
|
container = None
|
|
return container
|
|
|
|
|
|
def data_file_share(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve input data file share name
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: file share name
|
|
"""
|
|
try:
|
|
fshare = conf['file_share']
|
|
if util.is_none_or_empty(fshare):
|
|
raise KeyError()
|
|
except KeyError:
|
|
fshare = None
|
|
return fshare
|
|
|
|
|
|
def data_blobxfer_extra_options(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve input data blobxfer extra options
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: blobxfer extra options
|
|
"""
|
|
try:
|
|
eo = conf['blobxfer_extra_options']
|
|
if eo is None:
|
|
eo = ''
|
|
except KeyError:
|
|
eo = ''
|
|
return eo
|
|
|
|
|
|
def data_include(conf, one_allowable):
|
|
# type: (dict, bool) -> str
|
|
"""Retrieve input data include fileters
|
|
:param dict conf: configuration object
|
|
:param bool one_allowable: if only one include filter is allowed
|
|
:rtype: str
|
|
:return: include filters
|
|
"""
|
|
if one_allowable:
|
|
try:
|
|
include = conf['include']
|
|
if include is not None:
|
|
if len(include) == 0:
|
|
include = ''
|
|
elif len(include) == 1:
|
|
include = include[0]
|
|
else:
|
|
raise ValueError(
|
|
'include for input_data from {} cannot exceed '
|
|
'1 filter'.format(data_storage_account_settings(conf)))
|
|
else:
|
|
include = ''
|
|
except KeyError:
|
|
include = ''
|
|
else:
|
|
try:
|
|
include = conf['include']
|
|
if include is not None and len(include) == 0:
|
|
include = ''
|
|
else:
|
|
include = ';'.join(include)
|
|
except KeyError:
|
|
include = ''
|
|
if include is None:
|
|
include = ''
|
|
return include
|
|
|
|
|
|
def data_exclude(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve input data exclude filters
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: exclude filters
|
|
"""
|
|
try:
|
|
exclude = conf['exclude']
|
|
if exclude is not None and len(exclude) == 0:
|
|
exclude = ''
|
|
else:
|
|
exclude = ';'.join(exclude)
|
|
except KeyError:
|
|
exclude = ''
|
|
if exclude is None:
|
|
exclude = ''
|
|
return exclude
|
|
|
|
|
|
def input_data_destination(conf, on_task):
|
|
# type: (dict, bool) -> str
|
|
"""Retrieve input data destination
|
|
:param dict conf: configuration object
|
|
:param bool on_task: if input data is on the task spec
|
|
:rtype: str
|
|
:return: destination
|
|
"""
|
|
try:
|
|
dst = conf['destination']
|
|
if util.is_none_or_empty(dst):
|
|
raise KeyError()
|
|
except KeyError:
|
|
if on_task:
|
|
dst = '$AZ_BATCH_TASK_WORKING_DIR'
|
|
else:
|
|
raise
|
|
return dst
|
|
|
|
|
|
def input_data_job_id(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve input data job id
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: job id
|
|
"""
|
|
return conf['job_id']
|
|
|
|
|
|
def input_data_task_id(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve input data task id
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: task id
|
|
"""
|
|
return conf['task_id']
|
|
|
|
|
|
def output_data_source(conf):
|
|
# type: (dict) -> str
|
|
"""Retrieve output data source
|
|
:param dict conf: configuration object
|
|
:rtype: str
|
|
:return: source
|
|
"""
|
|
try:
|
|
src = conf['source']
|
|
if util.is_none_or_empty(src):
|
|
raise KeyError()
|
|
except KeyError:
|
|
src = '$AZ_BATCH_TASK_DIR'
|
|
return src
|
|
|
|
|
|
# JOBS SETTINGS
|
|
def job_specifications(config):
|
|
# type: (dict) -> dict
|
|
"""Get job specifications config block
|
|
:param dict config: configuration object
|
|
:rtype: dict
|
|
:return: job specifications
|
|
"""
|
|
return config['job_specifications']
|
|
|
|
|
|
def autogenerated_task_id_prefix(config):
|
|
# type: (dict) -> str
|
|
"""Get the autogenerated task id prefix to use
|
|
:param dict config: configuration object
|
|
:rtype: str
|
|
:return: auto-gen task id prefix
|
|
"""
|
|
conf = _kv_read_checked(
|
|
config['batch_shipyard'], 'autogenerated_task_id', {}
|
|
)
|
|
# do not use _kv_read_checked for prefix we want to allow empty string
|
|
try:
|
|
prefix = conf['prefix']
|
|
if prefix is None:
|
|
raise KeyError()
|
|
except KeyError:
|
|
prefix = 'task-'
|
|
return prefix
|
|
|
|
|
|
def autogenerated_task_id_zfill(config):
|
|
# type: (dict) -> int
|
|
"""Get the autogenerated task zfill setting to use
|
|
:param dict config: configuration object
|
|
:rtype: int
|
|
:return: auto-gen task number zfill
|
|
"""
|
|
conf = _kv_read_checked(
|
|
config['batch_shipyard'], 'autogenerated_task_id', {}
|
|
)
|
|
return _kv_read(conf, 'zfill_width', 5)
|
|
|
|
|
|
def job_tasks(config, conf):
|
|
# type: (dict, dict) -> list
|
|
"""Get all tasks for job
|
|
:param dict config: configuration object
|
|
:param dict conf: job configuration object
|
|
:rtype: list
|
|
:return: list of tasks
|
|
"""
|
|
for _task in conf['tasks']:
|
|
if 'task_factory' in _task:
|
|
# get storage settings if applicable
|
|
if 'file' in _task['task_factory']:
|
|
az = _task['task_factory']['file']['azure_storage']
|
|
tfstorage = TaskFactoryStorageSettings(
|
|
storage_settings=credentials_storage(
|
|
config, data_storage_account_settings(az)),
|
|
storage_link_name=az['storage_account_settings'],
|
|
container=data_container(az),
|
|
file_share=data_file_share(az),
|
|
include=_kv_read_checked(az, 'include'),
|
|
exclude=_kv_read_checked(az, 'exclude'),
|
|
)
|
|
else:
|
|
tfstorage = None
|
|
for task in task_factory.generate_task(_task, tfstorage):
|
|
yield task
|
|
else:
|
|
yield _task
|
|
|
|
|
|
def job_id(conf):
|
|
# type: (dict) -> str
|
|
"""Get job id of a job specification
|
|
:param dict conf: job configuration object
|
|
:rtype: str
|
|
:return: job id
|
|
"""
|
|
return conf['id']
|
|
|
|
|
|
def job_auto_complete(conf):
|
|
# type: (dict) -> bool
|
|
"""Get job (and multi-instance) autocomplete setting
|
|
:param dict conf: job configuration object
|
|
:rtype: bool
|
|
:return: job autocomplete
|
|
"""
|
|
try:
|
|
ac = conf['auto_complete']
|
|
except KeyError:
|
|
ac = False
|
|
return ac
|
|
|
|
|
|
def job_auto_pool(conf):
|
|
# type: (dict) -> PoolAutopoolSettings
|
|
"""Get job autopool setting
|
|
:param dict conf: job configuration object
|
|
:rtype: PoolAutopoolSettings
|
|
:return: job autopool settings
|
|
"""
|
|
ap = _kv_read_checked(conf, 'auto_pool')
|
|
if ap is not None:
|
|
return PoolAutopoolSettings(
|
|
pool_lifetime=_kv_read_checked(
|
|
ap, 'pool_lifetime', 'job').lower(),
|
|
keep_alive=_kv_read(ap, 'keep_alive', False),
|
|
)
|
|
else:
|
|
return None
|
|
|
|
|
|
def job_recurrence(conf):
|
|
# type: (dict) -> JobRecurrenceSettings
|
|
"""Get job recurrence setting
|
|
:param dict conf: job configuration object
|
|
:rtype: JobRecurrenceSettings
|
|
:return: job recurrence settings
|
|
"""
|
|
rec = _kv_read_checked(conf, 'recurrence')
|
|
if rec is not None:
|
|
do_not_run_until = _kv_read_checked(
|
|
rec['schedule'], 'do_not_run_until')
|
|
if do_not_run_until is not None:
|
|
do_not_run_until = dateutil.parser.parse(do_not_run_until)
|
|
do_not_run_after = _kv_read_checked(
|
|
rec['schedule'], 'do_not_run_after')
|
|
if do_not_run_after is not None:
|
|
do_not_run_after = dateutil.parser.parse(do_not_run_after)
|
|
start_window = _kv_read_checked(rec['schedule'], 'start_window')
|
|
if start_window is not None:
|
|
start_window = util.convert_string_to_timedelta(start_window)
|
|
recurrence_interval = util.convert_string_to_timedelta(
|
|
_kv_read_checked(rec['schedule'], 'recurrence_interval')
|
|
)
|
|
jm = _kv_read_checked(rec, 'job_manager', {})
|
|
return JobRecurrenceSettings(
|
|
schedule=JobScheduleSettings(
|
|
do_not_run_until=do_not_run_until,
|
|
do_not_run_after=do_not_run_after,
|
|
start_window=start_window,
|
|
recurrence_interval=recurrence_interval,
|
|
),
|
|
job_manager=JobManagerSettings(
|
|
allow_low_priority_node=_kv_read(
|
|
jm, 'allow_low_priority_node', True),
|
|
run_exclusive=_kv_read(jm, 'run_exclusive', False),
|
|
monitor_task_completion=_kv_read(
|
|
jm, 'monitor_task_completion', False),
|
|
)
|
|
)
|
|
else:
|
|
return None
|
|
|
|
|
|
def job_priority(conf):
|
|
# type: (dict) -> int
|
|
"""Get job priority setting
|
|
:param dict conf: job configuration object
|
|
:rtype: bool
|
|
:return: job autocomplete
|
|
"""
|
|
pri = _kv_read(conf, 'priority', 0)
|
|
if pri < -1000 or pri > 1000:
|
|
raise ValueError('job priority is invalid: {}'.format(pri))
|
|
return pri
|
|
|
|
|
|
def job_environment_variables(conf):
|
|
# type: (dict) -> str
|
|
"""Get env vars of a job specification
|
|
:param dict conf: job configuration object
|
|
:rtype: list
|
|
:return: job env vars
|
|
"""
|
|
try:
|
|
env_vars = conf['environment_variables']
|
|
if util.is_none_or_empty(env_vars):
|
|
raise KeyError()
|
|
except KeyError:
|
|
env_vars = {}
|
|
return env_vars
|
|
|
|
|
|
def job_environment_variables_keyvault_secret_id(conf):
|
|
# type: (dict) -> str
|
|
"""Get keyvault env vars of a job specification
|
|
:param dict conf: job configuration object
|
|
:rtype: list
|
|
:return: job env vars
|
|
"""
|
|
try:
|
|
secid = conf['environment_variables_keyvault_secret_id']
|
|
if util.is_none_or_empty(secid):
|
|
raise KeyError()
|
|
except KeyError:
|
|
secid = None
|
|
return secid
|
|
|
|
|
|
def job_max_task_retries(conf):
|
|
# type: (dict) -> int
|
|
"""Get number of times a task should be retried in a particular job
|
|
:param dict conf: job configuration object
|
|
:rtype: int
|
|
:return: max task retry count
|
|
"""
|
|
try:
|
|
max_task_retries = conf['max_task_retries']
|
|
if max_task_retries is None:
|
|
raise KeyError()
|
|
except KeyError:
|
|
max_task_retries = None
|
|
return max_task_retries
|
|
|
|
|
|
def job_max_wall_time(conf):
|
|
# type: (dict) -> int
|
|
"""Get maximum wall time for any task of a job
|
|
:param dict conf: job configuration object
|
|
:rtype: datetime.timedelta
|
|
:return: max wall time
|
|
"""
|
|
max_wall_time = _kv_read_checked(conf, 'max_wall_time')
|
|
if util.is_not_empty(max_wall_time):
|
|
max_wall_time = util.convert_string_to_timedelta(max_wall_time)
|
|
return max_wall_time
|
|
|
|
|
|
def job_allow_run_on_missing(conf):
|
|
# type: (dict) -> int
|
|
"""Get allow task run on missing image
|
|
:param dict conf: job configuration object
|
|
:rtype: bool
|
|
:return: allow run on missing image
|
|
"""
|
|
try:
|
|
allow = conf['allow_run_on_missing_image']
|
|
if allow is None:
|
|
raise KeyError()
|
|
except KeyError:
|
|
allow = False
|
|
return allow
|
|
|
|
|
|
def has_depends_on_task(conf):
|
|
# type: (dict) -> bool
|
|
"""Determines if task has task dependencies
|
|
:param dict conf: job configuration object
|
|
:rtype: bool
|
|
:return: task has task dependencies
|
|
"""
|
|
if ('depends_on' in conf and util.is_not_empty(conf['depends_on']) or
|
|
'depends_on_range' in conf and
|
|
util.is_not_empty(conf['depends_on_range'])):
|
|
if 'id' not in conf or util.is_none_or_empty(conf['id']):
|
|
raise ValueError(
|
|
'task id is not specified, but depends_on or '
|
|
'depends_on_range is set')
|
|
return True
|
|
return False
|
|
|
|
|
|
def is_multi_instance_task(conf):
|
|
# type: (dict) -> bool
|
|
"""Determines if task is multi-isntance
|
|
:param dict conf: task configuration object
|
|
:rtype: bool
|
|
:return: task is multi-instance
|
|
"""
|
|
return 'multi_instance' in conf
|
|
|
|
|
|
def task_name(conf):
|
|
# type: (dict) -> str
|
|
"""Get task name
|
|
:param dict conf: task configuration object
|
|
:rtype: str
|
|
:return: task name
|
|
"""
|
|
try:
|
|
name = conf['name']
|
|
if util.is_none_or_empty(name):
|
|
raise KeyError()
|
|
except KeyError:
|
|
name = None
|
|
return name
|
|
|
|
|
|
def task_docker_image(conf):
|
|
# type: (dict) -> str
|
|
"""Get docker image used by task
|
|
:param dict conf: task configuration object
|
|
:rtype: str
|
|
:return: docker image used by task
|
|
"""
|
|
try:
|
|
di = conf['image']
|
|
if util.is_none_or_empty(di):
|
|
raise KeyError()
|
|
except KeyError:
|
|
di = None
|
|
return di
|
|
|
|
|
|
def set_task_name(conf, name):
|
|
# type: (dict, str) -> None
|
|
"""Set task name
|
|
:param dict conf: task configuration object
|
|
:param str name: task name to set
|
|
"""
|
|
conf['name'] = name
|
|
|
|
|
|
def task_id(conf):
|
|
# type: (dict) -> str
|
|
"""Get task id
|
|
:param dict conf: task configuration object
|
|
:rtype: str
|
|
:return: task id
|
|
"""
|
|
try:
|
|
id = conf['id']
|
|
if util.is_none_or_empty(id):
|
|
raise KeyError()
|
|
except KeyError:
|
|
id = None
|
|
return id
|
|
|
|
|
|
def set_task_id(conf, id):
|
|
# type: (dict, str) -> None
|
|
"""Set task id
|
|
:param dict conf: task configuration object
|
|
:param str id: task id to set
|
|
"""
|
|
conf['id'] = id
|
|
|
|
|
|
def task_settings(cloud_pool, config, poolconf, jobspec, conf, missing_images):
|
|
# type: (azure.batch.models.CloudPool, dict, PoolSettings, dict,
|
|
# dict, list) -> TaskSettings
|
|
"""Get task settings
|
|
:param azure.batch.models.CloudPool cloud_pool: cloud pool object
|
|
:param dict config: configuration dict
|
|
:param PoolSettings poolconf: pool settings
|
|
:param dict jobspec: job specification
|
|
:param dict conf: task configuration object
|
|
:param list missing_images: list of missing docker images on pool
|
|
:rtype: TaskSettings
|
|
:return: task settings
|
|
"""
|
|
# id must be populated by the time this function is invoked
|
|
task_id = conf['id']
|
|
if util.is_none_or_empty(task_id):
|
|
raise ValueError('task id is invalid')
|
|
# check task id length
|
|
if len(task_id) > 64:
|
|
raise ValueError('task id exceeds 64 characters')
|
|
image = conf['image']
|
|
if util.is_none_or_empty(image):
|
|
raise ValueError('image is invalid')
|
|
# check if image is in missing image list
|
|
if image in missing_images:
|
|
# get private registry settings
|
|
preg = docker_registry_private_settings(config)
|
|
if util.is_not_empty(preg.storage_account):
|
|
registry = 'localhost:5000/'
|
|
elif util.is_not_empty(preg.server):
|
|
registry = '{}/'.format(preg.server)
|
|
else:
|
|
registry = ''
|
|
del preg
|
|
image = '{}{}'.format(registry, image)
|
|
# get some pool props
|
|
if cloud_pool is None:
|
|
pool_id = poolconf.id
|
|
vm_size = poolconf.vm_size
|
|
inter_node_comm = poolconf.inter_node_communication_enabled
|
|
is_custom_image = isinstance(
|
|
poolconf.vm_configuration, PoolVmCustomImageSettings)
|
|
if is_custom_image:
|
|
publisher = None
|
|
offer = None
|
|
sku = None
|
|
node_agent = poolconf.vm_configuration.node_agent
|
|
else:
|
|
publisher = poolconf.vm_configuration.publisher.lower()
|
|
offer = poolconf.vm_configuration.offer.lower()
|
|
sku = poolconf.vm_configuration.sku.lower()
|
|
else:
|
|
pool_id = cloud_pool.id
|
|
vm_size = cloud_pool.vm_size.lower()
|
|
inter_node_comm = cloud_pool.enable_inter_node_communication
|
|
is_custom_image = (
|
|
cloud_pool.virtual_machine_configuration.os_disk is not None
|
|
)
|
|
if is_custom_image:
|
|
publisher = None
|
|
offer = None
|
|
sku = None
|
|
node_agent = cloud_pool.virtual_machine_configuration.\
|
|
node_agent_sku_id.lower()
|
|
else:
|
|
publisher = cloud_pool.virtual_machine_configuration.\
|
|
image_reference.publisher.lower()
|
|
offer = cloud_pool.virtual_machine_configuration.\
|
|
image_reference.offer.lower()
|
|
sku = cloud_pool.virtual_machine_configuration.\
|
|
image_reference.sku.lower()
|
|
# get user identity settings
|
|
ui = _kv_read_checked(jobspec, 'user_identity', {})
|
|
ui_default_pool_admin = _kv_read(ui, 'default_pool_admin', False)
|
|
ui_specific = _kv_read(ui, 'specific_user', {})
|
|
ui_specific_uid = _kv_read(ui_specific, 'uid')
|
|
ui_specific_gid = _kv_read(ui_specific, 'gid')
|
|
del ui
|
|
del ui_specific
|
|
if ui_default_pool_admin and ui_specific_uid is not None:
|
|
raise ValueError(
|
|
'cannot specify both default_pool_admin and '
|
|
'specific_user:uid/gid at the same time')
|
|
ui = UserIdentitySettings(
|
|
default_pool_admin=ui_default_pool_admin,
|
|
specific_user_uid=ui_specific_uid,
|
|
specific_user_gid=ui_specific_gid,
|
|
)
|
|
# get depends on
|
|
try:
|
|
depends_on = conf['depends_on']
|
|
if util.is_none_or_empty(depends_on):
|
|
raise KeyError()
|
|
except KeyError:
|
|
depends_on = None
|
|
try:
|
|
depends_on_range = conf['depends_on_range']
|
|
if util.is_none_or_empty(depends_on_range):
|
|
raise KeyError()
|
|
if len(depends_on_range) != 2:
|
|
raise ValueError('depends_on_range requires 2 elements exactly')
|
|
if not (isinstance(depends_on_range[0], int) and
|
|
isinstance(depends_on_range[1], int)):
|
|
raise ValueError('depends_on_range requires integral members only')
|
|
except KeyError:
|
|
depends_on_range = None
|
|
# get additional resource files
|
|
try:
|
|
rfs = conf['resource_files']
|
|
if util.is_none_or_empty(rfs):
|
|
raise KeyError()
|
|
resource_files = []
|
|
for rf in rfs:
|
|
try:
|
|
fm = rf['file_mode']
|
|
if util.is_none_or_empty(fm):
|
|
raise KeyError()
|
|
except KeyError:
|
|
fm = None
|
|
resource_files.append(
|
|
ResourceFileSettings(
|
|
file_path=rf['file_path'],
|
|
blob_source=rf['blob_source'],
|
|
file_mode=fm,
|
|
)
|
|
)
|
|
except KeyError:
|
|
resource_files = None
|
|
# get generic run opts
|
|
try:
|
|
run_opts = conf['additional_docker_run_options']
|
|
except KeyError:
|
|
run_opts = []
|
|
# parse remove container option
|
|
rm_container = False
|
|
try:
|
|
rm_container = conf['remove_container_after_exit']
|
|
except KeyError:
|
|
rm_container = _kv_read(jobspec, 'remove_container_after_exit', True)
|
|
if rm_container and '--rm' not in run_opts:
|
|
run_opts.append('--rm')
|
|
del rm_container
|
|
# parse /dev/shm option
|
|
shm_size = None
|
|
try:
|
|
shm_size = conf['shm_size']
|
|
except KeyError:
|
|
shm_size = _kv_read_checked(jobspec, 'shm_size')
|
|
if (util.is_not_empty(shm_size) and
|
|
not any(x.startswith('--shm-size=') for x in run_opts)):
|
|
run_opts.append('--shm-size={}'.format(shm_size))
|
|
del shm_size
|
|
# parse name option, if not specified use task id
|
|
try:
|
|
name = conf['name']
|
|
if util.is_none_or_empty(name):
|
|
raise KeyError()
|
|
except KeyError:
|
|
name = task_id
|
|
set_task_name(conf, name)
|
|
run_opts.append('--name {}'.format(name))
|
|
# parse labels option
|
|
try:
|
|
labels = conf['labels']
|
|
if util.is_not_empty(labels):
|
|
for label in labels:
|
|
run_opts.append('-l {}'.format(label))
|
|
del labels
|
|
except KeyError:
|
|
pass
|
|
# parse ports option
|
|
try:
|
|
ports = conf['ports']
|
|
if util.is_not_empty(ports):
|
|
for port in ports:
|
|
run_opts.append('-p {}'.format(port))
|
|
del ports
|
|
except KeyError:
|
|
pass
|
|
# parse entrypoint
|
|
try:
|
|
entrypoint = conf['entrypoint']
|
|
if util.is_not_empty(entrypoint):
|
|
run_opts.append('--entrypoint {}'.format(entrypoint))
|
|
del entrypoint
|
|
except KeyError:
|
|
pass
|
|
# get command
|
|
try:
|
|
command = conf['command']
|
|
if util.is_none_or_empty(command):
|
|
raise KeyError()
|
|
except KeyError:
|
|
command = None
|
|
# parse data volumes
|
|
data_volumes = _kv_read_checked(jobspec, 'data_volumes')
|
|
try:
|
|
tdv = conf['data_volumes']
|
|
if util.is_not_empty(tdv):
|
|
if util.is_not_empty(data_volumes):
|
|
# check for intersection
|
|
if len(set(data_volumes).intersection(set(tdv))) > 0:
|
|
raise ValueError('data volumes must be unique')
|
|
data_volumes.extend(tdv)
|
|
else:
|
|
data_volumes = tdv
|
|
del tdv
|
|
except KeyError:
|
|
pass
|
|
if util.is_not_empty(data_volumes):
|
|
dv = global_resources_data_volumes(config)
|
|
for dvkey in data_volumes:
|
|
try:
|
|
hostpath = dv[dvkey]['host_path']
|
|
if util.is_none_or_empty(hostpath):
|
|
raise KeyError()
|
|
except KeyError:
|
|
hostpath = None
|
|
if util.is_not_empty(hostpath):
|
|
run_opts.append('-v {}:{}'.format(
|
|
hostpath, dv[dvkey]['container_path']))
|
|
else:
|
|
run_opts.append('-v {}'.format(
|
|
dv[dvkey]['container_path']))
|
|
del data_volumes
|
|
# parse shared data volumes
|
|
shared_data_volumes = _kv_read_checked(jobspec, 'shared_data_volumes')
|
|
try:
|
|
tsdv = conf['shared_data_volumes']
|
|
if util.is_not_empty(tsdv):
|
|
if util.is_not_empty(shared_data_volumes):
|
|
# check for intersection
|
|
if len(set(shared_data_volumes).intersection(set(tsdv))) > 0:
|
|
raise ValueError('shared data volumes must be unique')
|
|
shared_data_volumes.extend(tsdv)
|
|
else:
|
|
shared_data_volumes = tsdv
|
|
del tsdv
|
|
except KeyError:
|
|
pass
|
|
if util.is_not_empty(shared_data_volumes):
|
|
sdv = global_resources_shared_data_volumes(config)
|
|
for sdvkey in shared_data_volumes:
|
|
if is_shared_data_volume_gluster_on_compute(sdv, sdvkey):
|
|
run_opts.append('-v {}/{}:{}'.format(
|
|
'$AZ_BATCH_NODE_SHARED_DIR',
|
|
get_gluster_on_compute_volume(),
|
|
shared_data_volume_container_path(sdv, sdvkey)))
|
|
elif is_shared_data_volume_storage_cluster(sdv, sdvkey):
|
|
run_opts.append('-v {}/{}:{}'.format(
|
|
'$AZ_BATCH_NODE_SHARED_DIR',
|
|
sdvkey,
|
|
shared_data_volume_container_path(sdv, sdvkey)))
|
|
else:
|
|
run_opts.append('-v {}:{}'.format(
|
|
sdvkey, shared_data_volume_container_path(sdv, sdvkey)))
|
|
del shared_data_volumes
|
|
# append user identity options
|
|
attach_ui = False
|
|
if ui.default_pool_admin:
|
|
# run as the default pool admin user. note that this is *undocumented*
|
|
# behavior and may break at anytime
|
|
run_opts.append('-u `id -u _azbatch`:`id -g _azbatch`')
|
|
attach_ui = True
|
|
elif ui.specific_user_uid is not None:
|
|
if ui.specific_user_gid is None:
|
|
raise ValueError(
|
|
'cannot specify a user identity uid without a gid')
|
|
run_opts.append(
|
|
'-u {}:{}'.format(ui.specific_user_uid, ui.specific_user_gid))
|
|
attach_ui = True
|
|
if attach_ui:
|
|
run_opts.append('-v /etc/passwd:/etc/passwd:ro')
|
|
run_opts.append('-v /etc/group:/etc/group:ro')
|
|
run_opts.append('-v /etc/sudoers:/etc/sudoers:ro')
|
|
del attach_ui
|
|
del ui
|
|
# env vars
|
|
try:
|
|
env_vars = conf['environment_variables']
|
|
if util.is_none_or_empty(env_vars):
|
|
raise KeyError()
|
|
except KeyError:
|
|
env_vars = {}
|
|
try:
|
|
ev_secid = conf['environment_variables_keyvault_secret_id']
|
|
if util.is_none_or_empty(ev_secid):
|
|
raise KeyError()
|
|
except KeyError:
|
|
ev_secid = None
|
|
# max_task_retries
|
|
try:
|
|
max_task_retries = conf['max_task_retries']
|
|
if max_task_retries is None:
|
|
raise KeyError()
|
|
except KeyError:
|
|
max_task_retries = None
|
|
# max wall time
|
|
try:
|
|
max_wall_time = conf['max_wall_time']
|
|
if max_wall_time is None:
|
|
raise KeyError()
|
|
else:
|
|
max_wall_time = util.convert_string_to_timedelta(max_wall_time)
|
|
except KeyError:
|
|
max_wall_time = None
|
|
# retention time
|
|
try:
|
|
retention_time = conf['retention_time']
|
|
except KeyError:
|
|
retention_time = _kv_read_checked(jobspec, 'retention_time')
|
|
if util.is_not_empty(retention_time):
|
|
retention_time = util.convert_string_to_timedelta(retention_time)
|
|
else:
|
|
retention_time = None
|
|
# infiniband
|
|
infiniband = False
|
|
try:
|
|
infiniband = conf['infiniband']
|
|
except KeyError:
|
|
infiniband = _kv_read(jobspec, 'infiniband', False)
|
|
# gpu
|
|
gpu = False
|
|
try:
|
|
gpu = conf['gpu']
|
|
except KeyError:
|
|
gpu = _kv_read(jobspec, 'gpu', False)
|
|
# adjust for gpu settings
|
|
if gpu:
|
|
if not is_gpu_pool(vm_size):
|
|
raise RuntimeError(
|
|
('cannot initialize a gpu task on nodes without '
|
|
'gpus, pool: {} vm_size: {}').format(pool_id, vm_size))
|
|
# TODO other images as they become available with gpu support
|
|
if ((sku is None and node_agent != 'batch.node.ubuntu 16.04') or
|
|
(publisher != 'canonical' and offer != 'ubuntuserver' and
|
|
(sku is not None and sku < '16.04'))):
|
|
raise ValueError(
|
|
('Unsupported gpu VM config, publisher={} offer={} '
|
|
'sku={}').format(publisher, offer, sku))
|
|
# set docker commands with nvidia docker wrapper
|
|
docker_run_cmd = 'nvidia-docker run'
|
|
docker_exec_cmd = 'nvidia-docker exec'
|
|
else:
|
|
# set normal run and exec commands
|
|
docker_run_cmd = 'docker run'
|
|
docker_exec_cmd = 'docker exec'
|
|
# adjust for infiniband
|
|
if infiniband:
|
|
if not inter_node_comm:
|
|
raise RuntimeError(
|
|
('cannot initialize an infiniband task on a '
|
|
'non-internode communication enabled '
|
|
'pool: {}').format(pool_id))
|
|
if not is_rdma_pool(vm_size):
|
|
raise RuntimeError(
|
|
('cannot initialize an infiniband task on nodes '
|
|
'without RDMA, pool: {} vm_size: {}').format(
|
|
pool_id, vm_size))
|
|
# common run opts
|
|
run_opts.append('--net=host')
|
|
run_opts.append('--ulimit memlock=9223372036854775807')
|
|
run_opts.append('--device=/dev/infiniband/rdma_cm')
|
|
run_opts.append('--device=/dev/infiniband/uverbs0')
|
|
run_opts.append('-v /opt/intel:/opt/intel:ro')
|
|
# only centos-hpc and sles-hpc:12-sp1 are supported
|
|
# for infiniband
|
|
if (publisher == 'openlogic' and offer == 'centos-hpc' or
|
|
node_agent.startswith('batch.node.centos')):
|
|
run_opts.append('-v /etc/rdma:/etc/rdma:ro')
|
|
run_opts.append('-v /etc/rdma/dat.conf:/etc/dat.conf:ro')
|
|
elif (publisher == 'suse' and offer == 'sles-hpc' and
|
|
sku == '12-sp1' or node_agent.startswith('batch.node.opensuse')):
|
|
run_opts.append('-v /etc/dat.conf:/etc/dat.conf:ro')
|
|
run_opts.append('-v /etc/dat.conf:/etc/rdma/dat.conf:ro')
|
|
run_opts.append('--device=/dev/hvnd_rdma')
|
|
else:
|
|
raise ValueError(
|
|
('Unsupported infiniband VM config, publisher={} '
|
|
'offer={}').format(publisher, offer))
|
|
# mount batch root dir
|
|
run_opts.append(
|
|
'-v $AZ_BATCH_NODE_ROOT_DIR:$AZ_BATCH_NODE_ROOT_DIR')
|
|
# set working directory if not already set
|
|
if not any((x.startswith('-w ') or x.startswith('--workdir '))
|
|
for x in run_opts):
|
|
run_opts.append('-w $AZ_BATCH_TASK_WORKING_DIR')
|
|
# always add option for envfile
|
|
envfile = '.shipyard.envlist'
|
|
run_opts.append('--env-file {}'.format(envfile))
|
|
# populate mult-instance settings
|
|
if is_multi_instance_task(conf):
|
|
if not inter_node_comm:
|
|
raise RuntimeError(
|
|
('cannot run a multi-instance task on a '
|
|
'non-internode communication enabled '
|
|
'pool: {}').format(pool_id))
|
|
# container must be named
|
|
if util.is_none_or_empty(name):
|
|
raise ValueError(
|
|
'multi-instance task must be invoked with a named '
|
|
'container')
|
|
# docker exec command cannot be empty/None
|
|
if util.is_none_or_empty(command):
|
|
raise ValueError(
|
|
'multi-instance task must have an application command')
|
|
# set docker run as coordination command
|
|
try:
|
|
run_opts.remove('--rm')
|
|
except ValueError:
|
|
pass
|
|
# run in detached mode
|
|
run_opts.append('-d')
|
|
# ensure host networking stack is used
|
|
if '--net=host' not in run_opts:
|
|
run_opts.append('--net=host')
|
|
# get coordination command
|
|
try:
|
|
coordination_command = conf[
|
|
'multi_instance']['coordination_command']
|
|
if util.is_none_or_empty(coordination_command):
|
|
raise KeyError()
|
|
except KeyError:
|
|
coordination_command = None
|
|
cc_args = [
|
|
'env | grep AZ_BATCH_ >> {}'.format(envfile),
|
|
'{} {} {}{}'.format(
|
|
docker_run_cmd,
|
|
' '.join(run_opts),
|
|
image,
|
|
'{}'.format(' ' + coordination_command)
|
|
if coordination_command else '')
|
|
]
|
|
# get num instances
|
|
num_instances = conf['multi_instance']['num_instances']
|
|
if not isinstance(num_instances, int):
|
|
# TODO remove deprecation path
|
|
if (num_instances == 'pool_specification_vm_count_dedicated' or
|
|
num_instances == 'pool_specification_vm_count'):
|
|
pool_vm_count = _pool_vm_count(config)
|
|
num_instances = pool_vm_count.dedicated
|
|
elif num_instances == 'pool_specification_vm_count_low_priority':
|
|
pool_vm_count = _pool_vm_count(config)
|
|
num_instances = pool_vm_count.low_priority
|
|
elif (num_instances == 'pool_current_dedicated' or
|
|
num_instances == 'pool_current_low_priority'):
|
|
if cloud_pool is None:
|
|
raise RuntimeError(
|
|
('Cannot retrieve current dedicated count for '
|
|
'pool: {}. Ensure pool exists.)'.format(pool_id)))
|
|
if num_instances == 'pool_current_dedicated':
|
|
num_instances = cloud_pool.current_dedicated_nodes
|
|
elif num_instances == 'pool_current_low_priority':
|
|
num_instances = cloud_pool.current_low_priority_nodes
|
|
else:
|
|
raise ValueError(
|
|
('multi instance num instances setting '
|
|
'invalid: {}').format(num_instances))
|
|
# get common resource files
|
|
try:
|
|
mi_rfs = conf['multi_instance']['resource_files']
|
|
if util.is_none_or_empty(mi_rfs):
|
|
raise KeyError()
|
|
mi_resource_files = []
|
|
for rf in mi_rfs:
|
|
try:
|
|
fm = rf['file_mode']
|
|
if util.is_none_or_empty(fm):
|
|
raise KeyError()
|
|
except KeyError:
|
|
fm = None
|
|
mi_resource_files.append(
|
|
ResourceFileSettings(
|
|
file_path=rf['file_path'],
|
|
blob_source=rf['blob_source'],
|
|
file_mode=fm,
|
|
)
|
|
)
|
|
except KeyError:
|
|
mi_resource_files = None
|
|
else:
|
|
num_instances = 0
|
|
cc_args = None
|
|
mi_resource_files = None
|
|
return TaskSettings(
|
|
id=task_id,
|
|
image=image,
|
|
name=name,
|
|
docker_run_options=run_opts,
|
|
environment_variables=env_vars,
|
|
environment_variables_keyvault_secret_id=ev_secid,
|
|
envfile=envfile,
|
|
resource_files=resource_files,
|
|
max_task_retries=max_task_retries,
|
|
max_wall_time=max_wall_time,
|
|
retention_time=retention_time,
|
|
command=command,
|
|
infiniband=infiniband,
|
|
gpu=gpu,
|
|
depends_on=depends_on,
|
|
depends_on_range=depends_on_range,
|
|
docker_run_cmd=docker_run_cmd,
|
|
docker_exec_cmd=docker_exec_cmd,
|
|
multi_instance=MultiInstanceSettings(
|
|
num_instances=num_instances,
|
|
coordination_command=cc_args,
|
|
resource_files=mi_resource_files,
|
|
),
|
|
)
|
|
|
|
|
|
# REMOTEFS SETTINGS
|
|
def virtual_network_settings(
|
|
config, default_resource_group=None, default_existing_ok=False,
|
|
default_create_nonexistant=True):
|
|
# type: (dict) -> VirtualNetworkSettings
|
|
"""Get virtual network settings
|
|
:param dict config: configuration dict
|
|
:param str default_resource_group: default resource group
|
|
:param bool default_existing_ok: default existing ok
|
|
:param bool default_create_nonexistant: default create nonexistant
|
|
:rtype: VirtualNetworkSettings
|
|
:return: virtual network settings
|
|
"""
|
|
conf = _kv_read_checked(config, 'virtual_network', {})
|
|
name = _kv_read_checked(conf, 'name')
|
|
resource_group = _kv_read_checked(
|
|
conf, 'resource_group', default_resource_group)
|
|
address_space = _kv_read_checked(conf, 'address_space')
|
|
existing_ok = _kv_read(conf, 'existing_ok', default_existing_ok)
|
|
create_nonexistant = _kv_read(
|
|
conf, 'create_nonexistant', default_create_nonexistant)
|
|
sub_conf = _kv_read_checked(conf, 'subnet', {})
|
|
subnet_name = _kv_read_checked(sub_conf, 'name')
|
|
subnet_address_prefix = _kv_read_checked(sub_conf, 'address_prefix')
|
|
return VirtualNetworkSettings(
|
|
name=name,
|
|
resource_group=resource_group,
|
|
address_space=address_space,
|
|
subnet_name=subnet_name,
|
|
subnet_address_prefix=subnet_address_prefix,
|
|
existing_ok=existing_ok,
|
|
create_nonexistant=create_nonexistant,
|
|
)
|
|
|
|
|
|
def fileserver_settings(config, vm_count):
|
|
conf = _kv_read_checked(config, 'file_server', {})
|
|
sc_fs_type = _kv_read_checked(conf, 'type')
|
|
if util.is_none_or_empty(sc_fs_type):
|
|
raise ValueError('file_server:type must be specified')
|
|
sc_fs_type = sc_fs_type.lower()
|
|
# cross check against number of vms
|
|
if ((sc_fs_type == 'nfs' and vm_count != 1) or
|
|
(sc_fs_type == 'glusterfs' and vm_count <= 1)):
|
|
raise ValueError(
|
|
('invalid combination of file_server:type {} and '
|
|
'vm_count {}').format(sc_fs_type, vm_count))
|
|
sc_fs_mountpoint = _kv_read_checked(conf, 'mountpoint')
|
|
if util.is_none_or_empty(sc_fs_mountpoint):
|
|
raise ValueError('file_server must be specified')
|
|
sc_mo = _kv_read_checked(conf, 'mount_options')
|
|
# get server options
|
|
so_conf = _kv_read_checked(conf, 'server_options', {})
|
|
# get samba options
|
|
sc_samba = _kv_read_checked(conf, 'samba', {})
|
|
smb_share_name = _kv_read_checked(sc_samba, 'share_name')
|
|
sc_samba_account = _kv_read_checked(sc_samba, 'account', {})
|
|
smb_account = SambaAccountSettings(
|
|
username=_kv_read_checked(sc_samba_account, 'username', 'nobody'),
|
|
password=_kv_read_checked(sc_samba_account, 'password'),
|
|
uid=_kv_read(sc_samba_account, 'uid'),
|
|
gid=_kv_read(sc_samba_account, 'gid'),
|
|
)
|
|
if smb_account.username != 'nobody':
|
|
if util.is_none_or_empty(smb_account.password):
|
|
raise ValueError(
|
|
'samba account password is invalid for username {}'.format(
|
|
smb_account.username))
|
|
if '\n' in smb_account.password:
|
|
raise ValueError(
|
|
'samba account password contains invalid characters')
|
|
if smb_account.uid is None or smb_account.gid is None:
|
|
raise ValueError(
|
|
('samba account uid and/or gid is invalid for '
|
|
'username {}').format(smb_account.username))
|
|
smb_ro = _kv_read(sc_samba, 'read_only', False)
|
|
if smb_ro:
|
|
smb_ro = 'yes'
|
|
else:
|
|
smb_ro = 'no'
|
|
smb_cm = _kv_read_checked(sc_samba, 'create_mask', '0700')
|
|
smb_dm = _kv_read_checked(sc_samba, 'directory_mask', '0700')
|
|
return FileServerSettings(
|
|
type=sc_fs_type,
|
|
mountpoint=sc_fs_mountpoint,
|
|
mount_options=sc_mo,
|
|
server_options=so_conf,
|
|
samba=SambaSettings(
|
|
share_name=smb_share_name,
|
|
account=smb_account,
|
|
read_only=smb_ro,
|
|
create_mask=smb_cm,
|
|
directory_mask=smb_dm,
|
|
),
|
|
)
|
|
|
|
|
|
def remotefs_settings(config, sc_id=None):
|
|
# type: (dict, str) -> RemoteFsSettings
|
|
"""Get remote fs settings
|
|
:param dict config: configuration dict
|
|
:param str sc_id: storage cluster id
|
|
:rtype: RemoteFsSettings
|
|
:return: remote fs settings
|
|
"""
|
|
# general settings
|
|
conf = config['remote_fs']
|
|
resource_group = _kv_read_checked(conf, 'resource_group')
|
|
location = conf['location']
|
|
if util.is_none_or_empty(location):
|
|
raise ValueError('invalid location in remote_fs')
|
|
# managed disk settings
|
|
md_conf = conf['managed_disks']
|
|
md_rg = _kv_read_checked(md_conf, 'resource_group', resource_group)
|
|
if util.is_none_or_empty(md_rg):
|
|
raise ValueError('invalid managed_disks:resource_group in remote_fs')
|
|
md_premium = _kv_read(md_conf, 'premium', False)
|
|
md_disk_size_gb = _kv_read(md_conf, 'disk_size_gb')
|
|
md_disk_names = _kv_read_checked(md_conf, 'disk_names')
|
|
md = ManagedDisksSettings(
|
|
resource_group=md_rg,
|
|
premium=md_premium,
|
|
disk_size_gb=md_disk_size_gb,
|
|
disk_names=md_disk_names,
|
|
)
|
|
if util.is_none_or_empty(sc_id):
|
|
return RemoteFsSettings(
|
|
location=location,
|
|
managed_disks=md,
|
|
storage_cluster=None,
|
|
)
|
|
# storage cluster settings
|
|
try:
|
|
sc_conf = conf['storage_clusters'][sc_id]
|
|
except KeyError:
|
|
raise ValueError(
|
|
('Storage cluster {} is not defined in the given fs '
|
|
'configuration file').format(sc_id))
|
|
sc_rg = _kv_read_checked(sc_conf, 'resource_group', resource_group)
|
|
if util.is_none_or_empty(md_rg):
|
|
raise ValueError('invalid resource_group in remote_fs')
|
|
sc_vm_count = _kv_read(sc_conf, 'vm_count', 1)
|
|
sc_vm_size = _kv_read_checked(sc_conf, 'vm_size')
|
|
sc_fault_domains = _kv_read(sc_conf, 'fault_domains', 2)
|
|
if sc_fault_domains < 2 or sc_fault_domains > 3:
|
|
raise ValueError('fault_domains must be in range [2, 3]: {}'.format(
|
|
sc_fault_domains))
|
|
sc_hostname_prefix = _kv_read_checked(sc_conf, 'hostname_prefix')
|
|
# public ip settings
|
|
pip_conf = _kv_read_checked(sc_conf, 'public_ip', {})
|
|
sc_pip_enabled = _kv_read(pip_conf, 'enabled', True)
|
|
sc_pip_static = _kv_read(pip_conf, 'static', False)
|
|
# sc network security settings
|
|
ns_conf = sc_conf['network_security']
|
|
sc_ns_inbound = {
|
|
'ssh': InboundNetworkSecurityRule(
|
|
destination_port_range='22',
|
|
source_address_prefix=_kv_read_checked(ns_conf, 'ssh', ['*']),
|
|
protocol='tcp',
|
|
),
|
|
}
|
|
if not isinstance(sc_ns_inbound['ssh'].source_address_prefix, list):
|
|
raise ValueError('expected list for ssh network security rule')
|
|
if 'nfs' in ns_conf:
|
|
sc_ns_inbound['nfs'] = InboundNetworkSecurityRule(
|
|
destination_port_range='2049',
|
|
source_address_prefix=_kv_read_checked(ns_conf, 'nfs'),
|
|
protocol='tcp',
|
|
)
|
|
if not isinstance(sc_ns_inbound['nfs'].source_address_prefix, list):
|
|
raise ValueError('expected list for nfs network security rule')
|
|
if 'glusterfs' in ns_conf:
|
|
# glusterd and management ports
|
|
sc_ns_inbound['glusterfs-management'] = InboundNetworkSecurityRule(
|
|
destination_port_range='24007-24008',
|
|
source_address_prefix=_kv_read_checked(ns_conf, 'glusterfs'),
|
|
protocol='tcp',
|
|
)
|
|
# gluster brick ports: only 1 port per vm is needed as there will
|
|
# only be 1 brick per vm (brick is spread across RAID)
|
|
sc_ns_inbound['glusterfs-bricks'] = InboundNetworkSecurityRule(
|
|
destination_port_range='49152',
|
|
source_address_prefix=_kv_read_checked(ns_conf, 'glusterfs'),
|
|
protocol='tcp',
|
|
)
|
|
# only need to check one for glusterfs
|
|
if not isinstance(
|
|
sc_ns_inbound['glusterfs-management'].source_address_prefix,
|
|
list):
|
|
raise ValueError(
|
|
'expected list for glusterfs network security rule')
|
|
if 'smb' in ns_conf:
|
|
sc_ns_inbound['smb'] = InboundNetworkSecurityRule(
|
|
destination_port_range='445',
|
|
source_address_prefix=_kv_read_checked(ns_conf, 'smb'),
|
|
protocol='tcp',
|
|
)
|
|
if not isinstance(sc_ns_inbound['smb'].source_address_prefix, list):
|
|
raise ValueError('expected list for smb network security rule')
|
|
if 'custom_inbound_rules' in ns_conf:
|
|
# reserve keywords (current and expected possible future support)
|
|
_reserved = frozenset([
|
|
'ssh', 'nfs', 'glusterfs', 'smb', 'cifs', 'samba', 'zfs',
|
|
'beegfs', 'cephfs',
|
|
])
|
|
for key in ns_conf['custom_inbound_rules']:
|
|
# ensure key is not reserved
|
|
if key.lower() in _reserved:
|
|
raise ValueError(
|
|
('custom inbound rule of name {} conflicts with a '
|
|
'reserved name {}').format(key, _reserved))
|
|
sc_ns_inbound[key] = InboundNetworkSecurityRule(
|
|
destination_port_range=_kv_read_checked(
|
|
ns_conf['custom_inbound_rules'][key],
|
|
'destination_port_range'),
|
|
source_address_prefix=_kv_read_checked(
|
|
ns_conf['custom_inbound_rules'][key],
|
|
'source_address_prefix'),
|
|
protocol=_kv_read_checked(
|
|
ns_conf['custom_inbound_rules'][key], 'protocol'),
|
|
)
|
|
if not isinstance(sc_ns_inbound[key].source_address_prefix, list):
|
|
raise ValueError(
|
|
'expected list for network security rule {} '
|
|
'source_address_prefix'.format(key))
|
|
# sc file server settings
|
|
file_server = fileserver_settings(sc_conf, sc_vm_count)
|
|
# sc ssh settings
|
|
ssh_conf = sc_conf['ssh']
|
|
sc_ssh_username = _kv_read_checked(ssh_conf, 'username')
|
|
sc_ssh_public_key = _kv_read_checked(ssh_conf, 'ssh_public_key')
|
|
if util.is_not_empty(sc_ssh_public_key):
|
|
sc_ssh_public_key = pathlib.Path(sc_ssh_public_key)
|
|
sc_ssh_public_key_data = _kv_read_checked(ssh_conf, 'ssh_public_key_data')
|
|
sc_ssh_private_key = _kv_read_checked(ssh_conf, 'ssh_private_key')
|
|
if util.is_not_empty(sc_ssh_private_key):
|
|
sc_ssh_private_key = pathlib.Path(sc_ssh_private_key)
|
|
if (sc_ssh_public_key is not None and
|
|
util.is_not_empty(sc_ssh_public_key_data)):
|
|
raise ValueError('cannot specify both an SSH public key file and data')
|
|
if (sc_ssh_public_key is None and
|
|
util.is_none_or_empty(sc_ssh_public_key_data) and
|
|
sc_ssh_private_key is not None):
|
|
raise ValueError(
|
|
'cannot specify an SSH private key with no public key specified')
|
|
sc_ssh_gen_file_path = _kv_read_checked(
|
|
ssh_conf, 'generated_file_export_path', '.')
|
|
# ensure ssh username and samba username are not the same
|
|
if file_server.samba.account.username == sc_ssh_username:
|
|
raise ValueError(
|
|
'SSH username and samba account username cannot be the same')
|
|
# sc vm disk map settings
|
|
vmd_conf = sc_conf['vm_disk_map']
|
|
_disk_set = frozenset(md_disk_names)
|
|
disk_map = {}
|
|
for vmkey in vmd_conf:
|
|
# ensure all disks in disk array are specified in managed disks
|
|
disk_array = vmd_conf[vmkey]['disk_array']
|
|
if not _disk_set.issuperset(set(disk_array)):
|
|
raise ValueError(
|
|
('All disks {} for vm {} are not specified in '
|
|
'managed_disks:disk_names ({})').format(
|
|
disk_array, vmkey, _disk_set))
|
|
raid_level = _kv_read(vmd_conf[vmkey], 'raid_level', -1)
|
|
if len(disk_array) == 1 and raid_level != -1:
|
|
raise ValueError(
|
|
'Cannot specify a RAID-level with 1 disk in array')
|
|
else:
|
|
if raid_level == 0 and len(disk_array) < 2:
|
|
raise ValueError('RAID-0 arrays require at least two disks')
|
|
if raid_level != 0:
|
|
raise ValueError('Unsupported RAID level {}'.format(
|
|
raid_level))
|
|
filesystem = vmd_conf[vmkey]['filesystem']
|
|
if filesystem != 'btrfs' and not filesystem.startswith('ext'):
|
|
raise ValueError('Unsupported filesystem type {}'.format(
|
|
filesystem))
|
|
disk_map[int(vmkey)] = MappedVmDiskSettings(
|
|
disk_array=disk_array,
|
|
filesystem=vmd_conf[vmkey]['filesystem'],
|
|
raid_level=raid_level,
|
|
)
|
|
# check disk map against vm_count
|
|
if len(disk_map) != sc_vm_count:
|
|
raise ValueError(
|
|
('Number of entries in vm_disk_map {} inconsistent with '
|
|
'vm_count {}').format(len(disk_map), sc_vm_count))
|
|
return RemoteFsSettings(
|
|
location=location,
|
|
managed_disks=ManagedDisksSettings(
|
|
resource_group=md_rg,
|
|
premium=md_premium,
|
|
disk_size_gb=md_disk_size_gb,
|
|
disk_names=md_disk_names,
|
|
),
|
|
storage_cluster=StorageClusterSettings(
|
|
id=sc_id,
|
|
resource_group=sc_rg,
|
|
virtual_network=virtual_network_settings(
|
|
sc_conf,
|
|
default_resource_group=sc_rg,
|
|
default_existing_ok=False,
|
|
default_create_nonexistant=True,
|
|
),
|
|
network_security=NetworkSecuritySettings(
|
|
inbound=sc_ns_inbound,
|
|
),
|
|
file_server=file_server,
|
|
vm_count=sc_vm_count,
|
|
vm_size=sc_vm_size,
|
|
fault_domains=sc_fault_domains,
|
|
public_ip=PublicIpSettings(
|
|
enabled=sc_pip_enabled,
|
|
static=sc_pip_static,
|
|
),
|
|
hostname_prefix=sc_hostname_prefix,
|
|
ssh=SSHSettings(
|
|
username=sc_ssh_username,
|
|
expiry_days=9999,
|
|
ssh_public_key=sc_ssh_public_key,
|
|
ssh_public_key_data=sc_ssh_public_key_data,
|
|
ssh_private_key=sc_ssh_private_key,
|
|
generate_docker_tunnel_script=False,
|
|
generated_file_export_path=sc_ssh_gen_file_path,
|
|
hpn_server_swap=False,
|
|
),
|
|
vm_disk_map=disk_map,
|
|
),
|
|
)
|
|
|
|
|
|
def generate_availability_set_name(sc):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Generate an availabilty set name
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: availability set name
|
|
"""
|
|
return '{}-as'.format(sc.hostname_prefix)
|
|
|
|
|
|
def generate_virtual_machine_name(sc, i):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Generate a virtual machine name
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: vm name
|
|
"""
|
|
return '{}-vm{}'.format(sc.hostname_prefix, i)
|
|
|
|
|
|
def get_offset_from_virtual_machine_name(vm_name):
|
|
# type: (StorageClusterSettings) -> int
|
|
"""Gets the virtual machine offset given a vm name
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: int
|
|
:return: vm offset
|
|
"""
|
|
return int(vm_name.split('-vm')[-1])
|
|
|
|
|
|
def generate_virtual_machine_extension_name(sc, i):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Generate a virtual machine extension name
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: vm extension name
|
|
"""
|
|
return '{}-vmext{}'.format(sc.hostname_prefix, i)
|
|
|
|
|
|
def generate_network_security_group_name(sc):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Generate a network security group name
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: nsg name
|
|
"""
|
|
return '{}-nsg'.format(sc.hostname_prefix)
|
|
|
|
|
|
def generate_network_security_inbound_rule_name(rule_name, i):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Generate a network security inbound rule name
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: inbound rule name
|
|
"""
|
|
return '{}_in-{}'.format(rule_name, i)
|
|
|
|
|
|
def generate_network_security_inbound_rule_description(rule_name, i):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Generate a network security inbound rule description
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: inbound description
|
|
"""
|
|
return '{} inbound ({})'.format(rule_name, i)
|
|
|
|
|
|
def generate_public_ip_name(sc, i):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Generate a public ip name
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: public ip name
|
|
"""
|
|
return '{}-pip{}'.format(sc.hostname_prefix, i)
|
|
|
|
|
|
def generate_hostname(sc, i):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Generate a hostname (dns label prefix)
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: hostname
|
|
"""
|
|
return '{}{}'.format(sc.hostname_prefix, i)
|
|
|
|
|
|
def generate_network_interface_name(sc, i):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Generate a network inetrface name
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: network interface name
|
|
"""
|
|
return '{}-ni{}'.format(sc.hostname_prefix, i)
|
|
|
|
|
|
def get_file_server_glusterfs_volume_name(sc):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Get the glusterfs volume name
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: glusterfs volume name
|
|
"""
|
|
try:
|
|
volname = sc.file_server.server_options['glusterfs']['volume_name']
|
|
except KeyError:
|
|
volname = get_gluster_default_volume_name()
|
|
return volname
|
|
|
|
|
|
def get_file_server_glusterfs_volume_type(sc):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Get the glusterfs volume type
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: glusterfs volume type
|
|
"""
|
|
try:
|
|
voltype = sc.file_server.server_options[
|
|
'glusterfs']['volume_type'].lower()
|
|
except KeyError:
|
|
voltype = 'distributed'
|
|
return voltype
|
|
|
|
|
|
def get_file_server_glusterfs_transport(sc):
|
|
# type: (StorageClusterSettings) -> str
|
|
"""Get the glusterfs transport
|
|
:param StorageClusterSettings sc: storage cluster settings
|
|
:rtype: str
|
|
:return: glusterfs transport
|
|
"""
|
|
try:
|
|
transport = sc.file_server.server_options[
|
|
'glusterfs']['transport'].lower()
|
|
if transport != 'tcp':
|
|
raise ValueError('Only tcp is supported as transport')
|
|
except KeyError:
|
|
transport = 'tcp'
|
|
return transport
|