batch-shipyard/convoy/settings.py

5651 строка
197 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
2016-11-12 02:11:13 +03:00
import collections
import datetime
try:
import pathlib2 as pathlib
except ImportError:
import pathlib
import re
# non-stdlib imports
import azure.batch.models as batchmodels
2017-08-08 05:42:09 +03:00
import dateutil.parser
# local imports
from . import task_factory
from . import util
2016-11-12 02:11:13 +03:00
# global defines
_METADATA_VERSION_NAME = 'batch_shipyard_version'
_GLUSTER_DEFAULT_VOLNAME = 'gv0'
_GLUSTER_ON_COMPUTE_VOLUME = 'gluster_on_compute/{}'.format(
_GLUSTER_DEFAULT_VOLNAME)
_HOST_MOUNTS_DIR = '$AZ_BATCH_NODE_ROOT_DIR/mounts'
_HOST_MOUNTS_DIR_WINDOWS = '%AZ_BATCH_NODE_ROOT_DIR%\\mounts'
2017-04-28 20:56:03 +03:00
_TENSORBOARD_DOCKER_IMAGE = (
'gcr.io/tensorflow/tensorflow',
2017-11-13 02:11:26 +03:00
'/usr/local/lib/python2.7/dist-packages/tensorboard/main.py',
6006
2017-04-28 20:56:03 +03:00
)
_GPU_CC37_INSTANCES = re.compile(
# standard nc
2019-07-16 18:49:37 +03:00
r'^standard_nc[\d]+r?(_promo)?$',
re.IGNORECASE
)
_GPU_COMPUTE_INSTANCES = re.compile(
2018-11-29 20:39:34 +03:00
# standard nc, ncv2, ncv3, nd, ndv2
2019-07-16 18:49:37 +03:00
r'^standard_n[cd][\d]+r?s?(_v[\d])?(_promo)?$',
re.IGNORECASE
)
_GPU_VISUALIZATION_INSTANCES = re.compile(
# standard nv, nvv2
2019-07-16 18:49:37 +03:00
r'^standard_nv[\d]+s?(_v2)?(_promo)?$',
re.IGNORECASE
)
_SRIOV_RDMA_INSTANCES = re.compile(
# standard hb/hc
r'^standard_((hb|hc)[\d]+m?rs?(_v[\d])?)$',
re.IGNORECASE
)
_NETWORKDIRECT_RDMA_INSTANCES = re.compile(
# standard a8/a9, h+r, nc+r, nd+r
2019-07-16 18:49:37 +03:00
r'^standard_((a8|a9)|((h|nc|nd)[\d]+m?rs?(_v[1-3])?))(_promo)?$',
re.IGNORECASE
)
_PREMIUM_STORAGE_INSTANCES = re.compile(
r'^standard_(([a-z]+[\d]+.*s(_v[\d])?)|([dg]s[\d]+(_v2)?))$',
re.IGNORECASE
)
_NESTED_VIRTUALIZATION_INSTANCES = re.compile(
# standard dv3/ev3, fv2, m
r'^standard_(([de][\d]+s?_v3)|(f[\d]+s_v2)|(m[\d]+[lmst]*))$',
re.IGNORECASE
)
_VM_TCP_NO_TUNE = frozenset((
# basic
'basic_a0', 'basic_a1', 'basic_a2', 'basic_a3', 'basic_a4',
# standard_a
'standard_a0', 'standard_a1', 'standard_a2', 'standard_a3', 'standard_a5',
'standard_a6',
# standard_a_v2
'standard_a1_v2', 'standard_a2_v2', 'standard_a4_v2', 'standard_a2m_v2',
'standard_a4m_v2',
# standard_d
'standard_d1', 'standard_ds1', 'standard_d2', 'standard_ds2',
# standard_d_v2
'standard_d1_v2', 'standard_ds1_v2',
# standard_d_v3
'standard_d2_v3', 'standard_d2s_v3',
# standard_e_v3
'standard_e2_v3', 'standard_e2s_v3',
# standard_f
'standard_f1', 'standard_f1s',
# standard_f_v2
'standard_f2s_v2',
# standard_b
'standard_b1s', 'standard_b1ms', 'standard_b2s', 'standard_b2ms',
'standard_b4ms', 'standard_b8ms',
))
_VM_GPU_COUNT = {
2019-07-16 18:49:37 +03:00
1: re.compile(r'^standard_n[cdv]6r?s?(_v[\d])?(_promo)?$', re.IGNORECASE),
2: re.compile(r'^standard_n[cdv]12r?s?(_v[\d])?(_promo)?$', re.IGNORECASE),
4: re.compile(r'^standard_n[cdv]24r?s?(_v[\d])?(_promo)?$', re.IGNORECASE),
8: re.compile(r'^standard_nd40s_v2$', re.IGNORECASE),
}
_VM_GPU_CLASS = {
2019-07-16 18:49:37 +03:00
'tesla_k80': re.compile(r'^standard_n[c][\d]+r?(_promo)?$', re.IGNORECASE),
'tesla_p40': re.compile(r'^standard_n[d][\d]+r?s?$', re.IGNORECASE),
'tesla_p100': re.compile(r'^standard_n[c][\d]+r?s_v2$', re.IGNORECASE),
'tesla_v100': re.compile(
r'^standard_n(([c][\d]+r?s_v3)|(d40s_v2))$', re.IGNORECASE),
2019-07-16 18:49:37 +03:00
'tesla_m60': re.compile(
r'^standard_nv[\d]+s?(_v2)?(_promo)?$', re.IGNORECASE),
}
_VM_IB_CLASS = {
'qdr_ib': re.compile(r'^standard_(a8|a9)$', re.IGNORECASE),
'fdr_ib': re.compile(
2019-07-16 18:49:37 +03:00
r'^standard_(((h|nc|nd)+[\d]+m?rs?(_v[1-3])?))(_promo)?$',
re.IGNORECASE),
'edr_ib': re.compile(r'^standard_(hc|hb)+[\d]+rs$', re.IGNORECASE),
}
_VALID_PUBLISHERS = frozenset((
'canonical', 'credativ', 'microsoft-azure-batch',
'microsoftwindowsserver', 'openlogic'
))
_SINGULARITY_COMMANDS = frozenset(('exec', 'run'))
_FORBIDDEN_MERGE_TASK_PROPERTIES = frozenset((
'depends_on', 'depends_on_range', 'multi_instance', 'task_factory'
))
2016-11-12 02:11:13 +03:00
# named tuples
PoolVmCountSettings = collections.namedtuple(
'PoolVmCountSettings', [
'dedicated',
'low_priority',
]
)
2017-06-06 18:29:44 +03:00
PoolVmPlatformImageSettings = collections.namedtuple(
'PoolVmPlatformImageSettings', [
'publisher',
'offer',
'sku',
'version',
'native',
2018-02-28 19:38:57 +03:00
'license_type',
2017-06-06 18:29:44 +03:00
]
)
PoolVmCustomImageSettings = collections.namedtuple(
'PoolVmCustomImageSettings', [
'arm_image_id',
2017-06-06 18:29:44 +03:00
'node_agent',
'native',
2018-02-28 19:38:57 +03:00
'license_type',
2017-06-06 18:29:44 +03:00
]
)
PoolAutoscaleScenarioSettings = collections.namedtuple(
'PoolAutoscaleScenarioSettings', [
'name',
'maximum_vm_count',
'maximum_vm_increment_per_evaluation',
'node_deallocation_option',
'sample_lookback_interval',
'required_sample_percentage',
'rebalance_preemption_percentage',
'bias_last_sample',
'bias_node_type',
'weekday_start',
'weekday_end',
'workhour_start',
'workhour_end',
]
)
PoolAutoscaleSettings = collections.namedtuple(
'PoolAutoscaleSettings', [
'evaluation_interval',
'formula',
'scenario',
]
)
PoolAutopoolSettings = collections.namedtuple(
'PoolAutopoolSettings', [
'pool_lifetime',
'keep_alive',
]
)
PrometheusSettings = collections.namedtuple(
'PrometheusSettings', [
'ne_enabled', 'ne_port', 'ne_options', 'ca_enabled', 'ca_port',
'ca_options',
]
)
AdditionalNodePrepSettings = collections.namedtuple(
'AdditionalNodePrepSettings', [
'commands_pre',
'commands_post',
'environment_variables',
'environment_variables_keyvault_secret_id',
]
)
2016-11-12 02:11:13 +03:00
PoolSettings = collections.namedtuple(
'PoolSettings', [
'id', 'vm_size', 'vm_count', 'resize_timeout', 'max_tasks_per_node',
2017-06-06 18:29:44 +03:00
'inter_node_communication_enabled', 'vm_configuration',
2017-10-04 18:59:03 +03:00
'reboot_on_start_task_failed', 'attempt_recovery_on_unusable',
2016-11-12 02:11:13 +03:00
'block_until_all_global_resources_loaded',
2017-05-01 19:43:41 +03:00
'transfer_files_on_pool_creation', 'input_data', 'resource_files',
'gpu_driver', 'ssh', 'rdp', 'additional_node_prep', 'virtual_network',
2018-03-14 22:21:13 +03:00
'autoscale', 'node_fill_type', 'remote_access_control',
'certificates', 'prometheus', 'upload_diagnostics_logs_on_unusable',
'container_runtimes_install', 'container_runtimes_default',
2019-08-14 01:06:50 +03:00
'per_job_auto_scratch', 'batch_insights_enabled', 'public_ips',
2016-11-12 02:11:13 +03:00
]
)
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',
'allow_docker_access',
]
)
RDPSettings = collections.namedtuple(
'RDPSettings', [
'username', 'expiry_days', 'password',
]
)
2018-03-14 22:21:13 +03:00
RemoteAccessControl = collections.namedtuple(
'RemoteAccessControl', [
'starting_port', 'backend_port', 'protocol', 'allow', 'deny',
]
)
AADSettings = collections.namedtuple(
'AADSettings', [
'directory_id', 'application_id', 'auth_key', 'rsa_private_key_pem',
'x509_cert_sha1_thumbprint', 'user', 'password', 'endpoint',
'token_cache_file', 'authority_url',
]
)
2017-03-03 07:17:35 +03:00
KeyVaultCredentialsSettings = collections.namedtuple(
'KeyVaultCredentialsSettings', [
'aad', 'keyvault_uri', 'keyvault_credentials_secret_id',
2017-03-03 07:17:35 +03:00
]
)
ManagementCredentialsSettings = collections.namedtuple(
'ManagementCredentialsSettings', [
'aad', 'subscription_id',
2017-03-03 07:17:35 +03:00
]
)
BatchCredentialsSettings = collections.namedtuple(
'BatchCredentialsSettings', [
2017-03-08 22:13:09 +03:00
'aad', 'account', 'account_key', 'account_service_url',
'resource_group', 'subscription_id', 'location',
'app_insights_instrumentation_key',
'app_insights_application_id',
]
)
StorageCredentialsSettings = collections.namedtuple(
'StorageCredentialsSettings', [
'account', 'account_key', 'endpoint', 'resource_group',
]
)
BatchShipyardSettings = collections.namedtuple(
'BatchShipyardSettings', [
'storage_account_settings', 'storage_entity_prefix',
'generated_sas_expiry_days', 'use_shipyard_docker_image',
'store_timing_metrics', 'fallback_registry',
'delay_docker_image_preload',
]
)
DataReplicationSettings = collections.namedtuple(
'DataReplicationSettings', [
'concurrent_source_downloads',
]
)
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', 'is_file_share',
'remote_path', 'blobxfer_extra_options',
]
)
2017-08-08 05:42:09 +03:00
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',
2017-08-08 05:42:09 +03:00
]
)
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', 'remote_path',
'is_file_share', 'include', 'exclude',
]
)
TaskExitOptions = collections.namedtuple(
'TaskExitOptions', [
'job_action', 'dependency_action',
]
)
TaskSettings = collections.namedtuple(
'TaskSettings', [
'id', 'docker_image', 'singularity_image', 'name', 'run_options',
'docker_exec_options', 'singularity_cmd', 'run_elevated',
'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', 'multi_instance', 'default_exit_options',
'working_dir',
]
)
MultiInstanceSettings = collections.namedtuple(
'MultiInstanceSettings', [
'num_instances', 'coordination_command', 'resource_files',
'pre_execution_command', 'mpi',
]
)
MpiSettings = collections.namedtuple(
'MpiSettings', [
Fixes and update of recipes (#290) * Fix multi-instance tasks that are not a MPI task * Add setup task script for CNTK-CPU-Infiniband-IntelMPI * Update CNTK-CPU-Infiniband-IntelMPI recipe * Add MPI executable path option * Update CNTK-CPU-OpenMPI recipe * Change the default MPI executable_path to mpirun * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-Infiniband-IntelMPI * Update CNTK-GPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-OpenMPI * Add setup task script for NAMD-Infiniband-IntelMPI * Update NAMD-Infiniband-IntelMPI recipe * Add setup task script for OpenFOAM-Infiniband-IntelMPI * Update OpenFOAM-Infiniband-IntelMPI recipe * Update TensorFlow-GPU Singularity recipe * Add setup task script for OpenFOAM-TCP-OpenMPI * Update OpenFOAM-TCP-OpenMPI recipe * Add support for arbitrary commands with the MPI processes_per_node option * Fix MPI with native images * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Modify CNTK-GPU-Infiniband-IntelMPI recipe * Modify NAMD-Infiniband-IntelMPI recipe * Update processes_per_node documentation * Fix `pool images list` with Singularity images * Modify OpenFOAM-Infiniband-IntelMPI set up script * Add check for mpi setting with Windows * Add auto scratch support with OpenFOAM-Infiniband-IntelMPI recipe * Modify OpenFOAM-TCP-OpenMPI set up script * Add auto scratch support with OpenFOAM-TCP-OpenMPI recipe * Add mpiBench-IntelMPI recipe * Add mpiBench-MPICH recipe * Add mpiBench-OpenMPI recipe * Resolve PR comments * Resolve PR comments
2019-07-18 04:57:06 +03:00
'runtime', 'executable_path', 'options', 'processes_per_node',
]
)
ResourceFileSettings = collections.namedtuple(
'ResourceFileSettings', [
'file_path', 'blob_source', 'file_mode',
]
)
CustomMountFstabSettings = collections.namedtuple(
'CustomMountFstabSettings', [
'fs_spec', 'fs_vfstype', 'fs_mntops', 'fs_freq', 'fs_passno',
]
)
FederationPoolConstraintSettings = collections.namedtuple(
'FederationPoolConstraintSettings', [
'native', 'windows', 'location', 'custom_image_arm_id',
'virtual_network_arm_id', 'low_priority_nodes_allow',
'low_priority_nodes_exclusive', 'autoscale_allow',
'autoscale_exclusive', 'container_registries_private_docker_hub',
'container_registries_public', 'max_active_task_backlog_ratio',
'max_active_task_backlog_autoscale_exempt',
]
)
FederationComputeNodeConstraintSettings = collections.namedtuple(
'FederationComputeNodeConstraintSettings', [
'vm_size', 'cores', 'core_variance', 'memory', 'memory_variance',
'exclusive', 'gpu', 'infiniband',
]
)
FederationConstraintSettings = collections.namedtuple(
'FederationConstraintSettings', [
'pool', 'compute_node',
]
)
ManagedDisksSettings = collections.namedtuple(
'ManagedDisksSettings', [
'location', 'resource_group', 'zone', 'sku', 'disk_size_gb',
'disk_provisioned_perf_iops_rw', 'disk_provisioned_perf_mbps_rw',
'disk_names',
]
)
VirtualNetworkSettings = collections.namedtuple(
'VirtualNetworkSettings', [
'arm_subnet_id', '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',
]
)
RemoteFsSettings = collections.namedtuple(
'RemoteFsSettings', [
'managed_disks', 'storage_cluster',
]
)
PrometheusMonitoringSettings = collections.namedtuple(
'PrometheusMonitoringSettings', [
'port', 'scrape_interval',
]
)
GrafanaMonitoringSettings = collections.namedtuple(
'GrafanaMonitoringSettings', [
'admin_user', 'admin_password', 'additional_dashboards',
]
)
MonitoringServicesSettings = collections.namedtuple(
'MonitoringServicesSettings', [
'resource_polling_interval', 'lets_encrypt_enabled',
'lets_encrypt_staging', 'prometheus', 'grafana',
]
)
MonitoringVmSettings = collections.namedtuple(
'MonitoringVmSettings', [
'resource_group', 'virtual_network', 'network_security',
'vm_size', 'public_ip', 'hostname_prefix', 'ssh',
'accelerated_networking',
]
)
FederationProxyOptionsSettings = collections.namedtuple(
'FederationProxyOptionsSettings', [
'federations_polling_interval', 'actions_polling_interval',
'log_persistence', 'log_level', 'log_filename',
'scheduling_after_success_blackout_interval',
'scheduling_after_success_evaluate_autoscale',
]
)
SlurmBatchPoolSettings = collections.namedtuple(
'SlurmBatchPoolSettings', [
'batch_service_url', 'compute_node_type', 'max_compute_nodes',
'weight', 'features', 'reclaim_exclude_num_nodes',
]
)
SlurmPartitionSettings = collections.namedtuple(
'SlurmPartitionSettings', [
'batch_pools', 'max_runtime_limit', 'default', 'preempt_type',
'preempt_mode', 'over_subscribe', 'priority_tier', 'other_options',
]
)
SlurmUnmanagedPartitionSettings = collections.namedtuple(
'SlurmUnmanagedPartitionSettings', [
'partition', 'nodes',
]
)
SlurmOptionsSettings = collections.namedtuple(
'SlurmOptionsSettings', [
'cluster_id', 'idle_reclaim_time', 'max_nodes', 'elastic_partitions',
'unmanaged_partitions',
]
)
SlurmSharedDataVolumesSettings = collections.namedtuple(
'SlurmSharedDataVolumesSettings', [
'id', 'host_mount_path', 'store_slurmctld_state',
]
)
SlurmCredentialsSettings = collections.namedtuple(
'SlurmCredentialsSettings', [
'db_password',
]
)
SingularitySignedImageSettings = collections.namedtuple(
'SingularitySignedImageSettings', [
'image', 'key_fingerprint', 'key_file'
]
)
class VmResource(object):
def __init__(
self, location, resource_group, hostname_prefix, vm_size,
public_ip, virtual_network, network_security, ssh,
accelerated_networking, zone):
# type: (VmResource, str, str, str, str, PublicIpSettings,
# VirtualNetworkSettings, NetworkSecuritySettings, SSHSettings,
# bool, int) -> None
if location is None or ' ' in location:
raise ValueError('invalid location specified')
self.location = location
self.resource_group = resource_group
self.hostname_prefix = hostname_prefix
self.vm_size = vm_size
self.public_ip = public_ip
self.virtual_network = virtual_network
self.network_security = network_security
self.ssh = ssh
self.accelerated_networking = accelerated_networking
self.zone = zone
class StorageClusterSettings(VmResource):
def __init__(
self, id, file_server, vm_count, fault_domains, vm_disk_map,
location, resource_group, hostname_prefix, vm_size,
public_ip, virtual_network, network_security, ssh,
accelerated_networking, zone, prometheus):
# type: (StorageClusterSettings, str, FileServerSettings, int, int,
# Dict, str, str, str, str, PublicIpSettings,
# VirtualNetworkSettings, NetworkSecuritySettings, SSHSettings,
# bool, int, PrometheusSettings) -> None
super(StorageClusterSettings, self).__init__(
location, resource_group, hostname_prefix, vm_size, public_ip,
virtual_network, network_security, ssh, accelerated_networking,
zone)
self.id = id
self.file_server = file_server
self.vm_count = vm_count
self.fault_domains = fault_domains
self.vm_disk_map = vm_disk_map
self.prometheus = prometheus
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):
2017-04-28 20:56:03 +03:00
# 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_valid_publishers():
# type: (None) -> str
"""Get valid publishers
:rtype: str
:return: publisher set
"""
return _VALID_PUBLISHERS
2017-04-28 20:56:03 +03:00
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)
2017-04-28 20:56:03 +03:00
"""
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
2016-11-12 02:11:13 +03:00
def get_host_mounts_path(is_windows):
# type: (bool) -> str
"""Get host mounts path
:param bool is_windows: is windows pool
:rtype: str
:return: host mounts dir
"""
return _HOST_MOUNTS_DIR_WINDOWS if is_windows else _HOST_MOUNTS_DIR
def get_singularity_tmpdir(config):
# type: (dict) -> str
"""Get Singularity tmpdir var
:param dict config: configuration dict
:rtype: str
:return: singularity tmpdir
"""
if is_windows_pool(config):
sep = '\\'
else:
sep = '/'
return sep.join((temp_disk_mountpoint(config), 'singularity', 'tmp'))
def get_singularity_cachedir(config):
# type: (dict) -> str
"""Get Singularity cachedir var
:param dict config: configuration dict
:rtype: str
:return: singularity cachedir
"""
if is_windows_pool(config):
sep = '\\'
else:
sep = '/'
return sep.join((temp_disk_mountpoint(config), 'singularity', 'cache'))
def get_singularity_sypgpdir(config):
# type: (dict) -> str
"""Get Singularity sypgpdir var
:param dict config: configuration dict
:rtype: str
:return: singularity sypgpdir
"""
if is_windows_pool(config):
sep = '\\'
else:
sep = '/'
return sep.join((temp_disk_mountpoint(config), 'singularity', 'sypgp'))
2016-11-12 02:11:13 +03:00
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
"""
return (
is_gpu_compute_pool(vm_size) or is_gpu_visualization_pool(vm_size)
)
2016-11-12 02:11:13 +03:00
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
"""
2018-11-29 20:39:34 +03:00
return _GPU_COMPUTE_INSTANCES.match(vm_size) is not None
2016-11-12 02:11:13 +03:00
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
"""
2018-11-29 20:39:34 +03:00
return _GPU_VISUALIZATION_INSTANCES.match(vm_size) is not None
2016-11-12 02:11:13 +03:00
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
2018-03-22 00:39:58 +03:00
:return: type of gpu and compute capability
"""
if is_gpu_compute_pool(vm_size):
if _GPU_CC37_INSTANCES.match(vm_size):
2018-03-22 00:39:58 +03:00
return 'compute_cc37'
else:
return 'compute_cc6-7'
elif is_gpu_visualization_pool(vm_size):
2018-03-22 00:39:58 +03:00
return 'viz_cc52'
else:
return None
def get_num_gpus_from_vm_size(vm_size):
# type: (str) -> int
"""Get number of GPUs from VM size
:param str vm_size: vm size
:rtype: int
:return: number of GPUs
"""
for vm in _VM_GPU_COUNT:
if _VM_GPU_COUNT[vm].match(vm_size):
return vm
raise RuntimeError('vm_size {} has no mapping to number of GPUs'.format(
vm_size))
def get_gpu_class_from_vm_size(vm_size):
# type: (str) -> str
"""Get GPU class from VM size
:param str vm_size: vm size
:rtype: str
:return: GPU class
"""
for c in _VM_GPU_CLASS:
if _VM_GPU_CLASS[c].match(vm_size):
return c
raise RuntimeError('vm_size {} has no mapping to GPU class'.format(
vm_size))
2017-07-06 21:12:05 +03:00
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_vm_size(config)
2017-07-06 21:12:05 +03:00
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 == 'microsoft-azure-batch':
return True
elif (publisher == 'canonical' and offer == 'ubuntuserver' and
2017-07-06 21:12:05 +03:00
sku > '16.04'):
return True
elif publisher == 'openlogic':
if offer == 'centos-hpc' and sku >= '7.3':
return True
elif offer == 'centos' and sku >= '7.3':
return True
return False
def is_lis_install_required(config, vm_size=None):
# type: (dict, str) -> bool
"""Check if the pool requires installing LIS
:param dict config: configuration dict
:param str vm_size: vm size
:rtype: bool
:return: if pool requires lis install
"""
# LIS is linux only
if is_windows_pool(config):
return False
if util.is_none_or_empty(vm_size):
vm_size = _pool_vm_size(config)
# currently lis is only required for GPU pool setup for certain distros
if is_gpu_pool(vm_size):
publisher = pool_publisher(config, lower=True)
offer = pool_offer(config, lower=True)
sku = pool_sku(config, lower=True)
if publisher == 'openlogic' and offer == 'centos' and sku > '7.3':
return True
return False
2017-07-06 21:12:05 +03:00
def is_native_docker_pool(config, vm_config=None):
# type: (dict, any) -> bool
"""Check if vm configuration has native docker support
:param dict config: configuration dict
:param any vm_config: vm configuration
:rtype: bool
:return: if vm configuration has native docker support
"""
if vm_config is None:
vm_config = _populate_pool_vm_configuration(config)
return vm_config.native
def is_windows_pool(config, vm_config=None):
# type: (dict, any) -> bool
"""Check if pool is Windows
:param dict config: configuration dict
:param any vm_config: vm configuration
:rtype: bool
:return: pool is Windows
"""
if vm_config is None:
vm_config = _populate_pool_vm_configuration(config)
if is_platform_image(config, vm_config=vm_config):
return vm_config.publisher == 'microsoftwindowsserver'
else:
return vm_config.node_agent.startswith('batch.node.windows')
def is_sriov_rdma_pool(vm_size):
# type: (str) -> bool
"""Check if pool is SRIOV IB/RDMA capable
:param str vm_size: vm size
:rtype: bool
:return: if sriov rdma is present
"""
return _SRIOV_RDMA_INSTANCES.match(vm_size) is not None
def is_networkdirect_rdma_pool(vm_size):
# type: (str) -> bool
"""Check if pool is NetworkDirect IB/RDMA capable
:param str vm_size: vm size
:rtype: bool
:return: if network direct rdma is present
"""
return _NETWORKDIRECT_RDMA_INSTANCES.match(vm_size) is not None
2016-11-12 02:11:13 +03:00
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
"""
return is_sriov_rdma_pool(vm_size) or is_networkdirect_rdma_pool(vm_size)
2016-11-12 02:11:13 +03:00
def get_ib_class_from_vm_size(vm_size):
# type: (str) -> str
"""Get IB class from VM size
:param str vm_size: vm size
:rtype: str
:return: IB class
"""
for c in _VM_IB_CLASS:
if _VM_IB_CLASS[c].match(vm_size):
return c
raise RuntimeError('vm_size {} has no mapping to IB class'.format(
vm_size))
2017-03-13 12:10:08 +03:00
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
"""
2018-11-29 20:39:34 +03:00
return _PREMIUM_STORAGE_INSTANCES.match(vm_size) is not None
def is_nested_virtualization_capable(vm_size):
# type: (str) -> bool
"""Check if vm size is nested virtualization capable
:pararm str vm_size: vm size
:rtype: bool
:return: if vm size is nested virtualization capable
"""
2018-11-29 20:39:34 +03:00
return _NESTED_VIRTUALIZATION_INSTANCES.match(vm_size) is not None
2017-03-13 12:10:08 +03:00
2017-10-04 18:59:03 +03:00
def is_platform_image(config, vm_config=None):
# type (dict, PoolVmConfiguration) -> bool
"""If pool is on a platform image
:param dict config: configuration object
:param bool vm_config: vm configuration
:rtype: bool
:return: if on platform image
"""
if vm_config is None:
vm_config = _populate_pool_vm_configuration(config)
return isinstance(vm_config, PoolVmPlatformImageSettings)
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:
2017-06-06 18:29:44 +03:00
vmconfig = _populate_pool_vm_configuration(config)
2017-10-04 18:59:03 +03:00
if is_platform_image(config, vm_config=vmconfig):
2017-06-06 18:29:44 +03:00
offer = pool_offer(config, lower=True)
else:
if vmconfig.node_agent.lower().startswith('batch.node.ubuntu'):
offer = 'ubuntu'
elif vmconfig.node_agent.lower().startswith('batch.node.windows'):
offer = 'windowsserver'
2017-06-06 18:29:44 +03:00
else:
offer = '!ubuntu'
else:
offer = offer.lower()
if offer.startswith('ubuntu'):
return '/mnt'
elif offer.startswith('windows'):
return 'D:\\batch'
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 raw(config):
# type: (dict) -> bool
"""Get raw setting
:param dict config: configuration object
:rtype: bool
:return: raw setting
"""
return config['_raw']
def get_auto_confirm(config):
# type: (dict) -> bool
"""Get autoconfirm setting
:param dict config: configuration object
:rtype: bool
:return: auto confirm setting
"""
return config['_auto_confirm']
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
2016-11-12 02:11:13 +03:00
# 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']
return PoolVmCountSettings(
dedicated=_kv_read(conf, 'dedicated', 0),
low_priority=_kv_read(conf, 'low_priority', 0),
)
def _pool_vm_size(config):
# type: (dict) -> str
"""Get Pool VM size
:param dict config: configuration object
:rtype: str
:return: Pool VM Size
"""
return config['pool_specification']['vm_size'].lower()
2017-06-06 18:29:44 +03:00
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:
publisher = conf['publisher'].lower()
offer = conf['offer'].lower()
sku = str(conf['sku']).lower()
# auto convert windows native if detected
if publisher == 'microsoftwindowsserver':
vm_config = PoolVmPlatformImageSettings(
publisher=publisher,
offer=offer,
sku=sku,
version=_kv_read_checked(conf, 'version', default='latest'),
native=True,
2018-02-28 19:38:57 +03:00
license_type=_kv_read_checked(conf, 'license_type'),
)
elif publisher == 'microsoft-azure-batch':
# auto convert linux native if detected
vm_config = PoolVmPlatformImageSettings(
publisher=publisher,
offer=offer,
sku=sku,
version=_kv_read_checked(conf, 'version', default='latest'),
native=True,
license_type=None,
)
else:
vm_config = PoolVmPlatformImageSettings(
publisher=publisher,
offer=offer,
sku=sku,
version=_kv_read_checked(conf, 'version', default='latest'),
native=False,
2018-02-28 19:38:57 +03:00
license_type=None,
)
# auto convert vm config to native if specified
if not vm_config.native and _kv_read(conf, 'native', default=False):
vm_size = _pool_vm_size(config)
if (vm_config.publisher == 'canonical' and
vm_config.offer == 'ubuntuserver' and
vm_config.sku == '16.04-lts'):
vm_config = PoolVmPlatformImageSettings(
publisher='microsoft-azure-batch',
offer='ubuntu-server-container{}'.format(
'-rdma' if is_rdma_pool(vm_size) else ''),
sku=vm_config.sku.replace('.', '-'),
version='latest',
native=True,
license_type=None,
)
elif (vm_config.publisher == 'openlogic' and
vm_config.offer.startswith('centos') and
(vm_config.sku == '7.4' or vm_config.sku == '7.5' or
vm_config.sku == '7.6' or vm_config.sku == '7.7')):
vm_config = PoolVmPlatformImageSettings(
publisher='microsoft-azure-batch',
offer='centos-container{}'.format(
'-rdma' if is_rdma_pool(vm_size) else ''),
sku=vm_config.sku.replace('.', '-'),
version='latest',
native=True,
2018-02-28 19:38:57 +03:00
license_type=None,
)
return vm_config
2017-06-06 18:29:44 +03:00
else:
conf = pool_vm_configuration(config, 'custom_image')
node_agent = conf['node_agent'].lower()
if node_agent == 'batch.node.windows amd64':
native = True
2018-02-28 19:38:57 +03:00
license_type = _kv_read_checked(conf, 'license_type')
else:
native = _kv_read(conf, 'native', default=False)
2018-02-28 19:38:57 +03:00
license_type = None
2017-06-06 18:29:44 +03:00
return PoolVmCustomImageSettings(
arm_image_id=_kv_read_checked(conf, 'arm_image_id'),
node_agent=node_agent,
native=native,
2018-02-28 19:38:57 +03:00
license_type=license_type,
2017-06-06 18:29:44 +03:00
)
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:
2017-07-20 22:55:49 +03:00
mvc = _kv_read_checked(scenconf, 'maximum_vm_count')
if mvc is None:
raise ValueError('maximum_vm_count must be specified')
mvipe = _kv_read_checked(
scenconf, 'maximum_vm_increment_per_evaluation', default={})
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)
tr = _kv_read_checked(scenconf, 'time_ranges', default={})
trweekday = _kv_read_checked(tr, 'weekdays', default={})
trworkhour = _kv_read_checked(tr, 'work_hours', default={})
scenario = PoolAutoscaleScenarioSettings(
name=_kv_read_checked(scenconf, 'name').lower(),
maximum_vm_count=_pool_vm_count(config, conf=mvc),
maximum_vm_increment_per_evaluation=_pool_vm_count(
config, conf=mvipe),
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(),
weekday_start=_kv_read(trweekday, 'start', default=1),
weekday_end=_kv_read(trweekday, 'end', default=5),
workhour_start=_kv_read(trworkhour, 'start', default=8),
workhour_end=_kv_read(trworkhour, 'end', default=17),
)
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 prometheus_settings(config):
# type: (dict) -> PrometheusSettings
"""Get Prometheus Settings
:param dict config: configuration object
:rtype: PrometheusSettings
:return Prometheus settings from specification
"""
conf = _kv_read_checked(config, 'prometheus', default={})
ne_conf = _kv_read_checked(conf, 'node_exporter', default={})
ca_conf = _kv_read_checked(conf, 'cadvisor', default={})
ca_options = _kv_read_checked(ca_conf, 'options')
# do not allow docker_only, which interferes with metric gathering
if util.is_not_empty(ca_options) and '-docker_only' in ca_options:
ca_options.remove('-docker_only')
return PrometheusSettings(
ne_enabled=_kv_read(ne_conf, 'enabled', default=False),
ne_port=_kv_read(ne_conf, 'port', default=9100),
ne_options=_kv_read_checked(ne_conf, 'options'),
ca_enabled=_kv_read(ca_conf, 'enabled', default=False),
ca_port=_kv_read(ca_conf, 'port', default=8080),
ca_options=ca_options,
)
2016-11-12 02:11:13 +03:00
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)
2017-10-04 18:59:03 +03:00
max_tasks_per_node = _kv_read(conf, 'max_tasks_per_node', default=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
2017-10-04 18:59:03 +03:00
inter_node_communication_enabled = _kv_read(
conf, 'inter_node_communication_enabled', default=False)
reboot_on_start_task_failed = _kv_read(
conf, 'reboot_on_start_task_failed', default=False)
attempt_recovery_on_unusable = _kv_read(
conf, 'attempt_recovery_on_unusable', default=False)
block_until_all_gr = _kv_read(
conf, 'block_until_all_global_resources_loaded', default=True)
transfer_files_on_pool_creation = _kv_read(
conf, 'transfer_files_on_pool_creation', default=False)
2016-11-12 02:11:13 +03:00
try:
input_data = conf['input_data']
if util.is_none_or_empty(input_data):
raise KeyError()
except KeyError:
input_data = None
2017-05-01 19:43:41 +03:00
# 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
2017-04-19 04:46:57 +03:00
# ssh settings
2016-11-12 02:11:13 +03:00
try:
2017-04-19 04:46:57 +03:00
sshconf = conf['ssh']
ssh_username = _kv_read_checked(sshconf, 'username')
2016-11-12 02:11:13 +03:00
if util.is_none_or_empty(ssh_username):
raise KeyError()
except KeyError:
ssh_username = None
2017-04-19 04:46:57 +03:00
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 = '.'
2017-04-19 04:46:57 +03:00
ssh_hpn = None
ssh_ada = None
2017-04-19 04:46:57 +03:00
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)
ssh_ada = _kv_read(sshconf, 'allow_docker_access', False)
# rdp settings
try:
rdpconf = conf['rdp']
rdp_username = _kv_read_checked(rdpconf, 'username')
if util.is_none_or_empty(rdp_username):
raise KeyError()
except KeyError:
rdp_username = None
rdp_expiry_days = None
rdp_password = None
else:
rdp_expiry_days = _kv_read(rdpconf, 'expiry_days', 30)
if rdp_expiry_days <= 0:
rdp_expiry_days = 30
rdp_password = _kv_read_checked(rdpconf, 'password')
2018-03-14 22:21:13 +03:00
# remote access control
rac = _kv_read_checked(conf, 'remote_access_control', default={})
rac = RemoteAccessControl(
starting_port=_kv_read(rac, 'starting_port', default=49000),
backend_port='22' if ssh_username is not None else '3389',
protocol='tcp',
allow=_kv_read_checked(rac, 'allow'),
deny=_kv_read_checked(rac, 'deny'),
)
if (rac.starting_port < 1 or
(rac.starting_port > 49000 and rac.starting_port <= 55000) or
rac.starting_port > 64536):
raise ValueError('starting_port is invalid or in a reserved range')
2018-03-14 22:21:13 +03:00
# gpu driver
2016-11-12 02:11:13 +03:00
try:
2018-03-14 22:21:13 +03:00
gpu_driver = _kv_read_checked(conf['gpu']['nvidia_driver'], 'source')
2016-11-12 02:11:13 +03:00
except KeyError:
gpu_driver = None
2018-03-14 22:21:13 +03:00
# additional node prep
addl_np = _kv_read_checked(
conf, 'additional_node_prep', default={})
addl_np_cmds = _kv_read_checked(addl_np, 'commands', default={})
additional_node_prep = AdditionalNodePrepSettings(
commands_pre=_kv_read_checked(addl_np_cmds, 'pre', default=[]),
commands_post=_kv_read_checked(addl_np_cmds, 'post', default=[]),
environment_variables=_kv_read_checked(
addl_np, 'environment_variables', default={}),
environment_variables_keyvault_secret_id=_kv_read_checked(
addl_np, 'environment_variables_keyvault_secret_id'),
)
del addl_np_cmds
del addl_np
# certificates
certdict = _kv_read_checked(conf, 'certificates', default={})
certs = []
for tp in certdict:
visibility = []
for vis in certdict[tp]['visibility']:
if vis == 'remote_user':
visibility.append(
batchmodels.CertificateVisibility.remote_user)
elif vis == 'start_task':
visibility.append(batchmodels.CertificateVisibility.start_task)
elif vis == 'task':
visibility.append(batchmodels.CertificateVisibility.task)
certs.append(batchmodels.CertificateReference(
thumbprint=tp, thumbprint_algorithm='sha1',
visibility=visibility
))
# container runtimes
try:
cr_install = _kv_read_checked(
conf['container_runtimes'], 'install', default=[])
except KeyError:
cr_install = []
try:
cr_default = _kv_read_checked(
conf['container_runtimes'], 'default', default='runc')
except KeyError:
cr_default = 'runc'
2016-11-12 02:11:13 +03:00
return PoolSettings(
id=conf['id'],
vm_size=_pool_vm_size(config),
vm_count=_pool_vm_count(config),
resize_timeout=resize_timeout,
2016-11-12 02:11:13 +03:00
max_tasks_per_node=max_tasks_per_node,
inter_node_communication_enabled=inter_node_communication_enabled,
2017-06-06 18:29:44 +03:00
vm_configuration=_populate_pool_vm_configuration(config),
2016-11-12 02:11:13 +03:00
reboot_on_start_task_failed=reboot_on_start_task_failed,
2017-10-04 18:59:03 +03:00
attempt_recovery_on_unusable=attempt_recovery_on_unusable,
2016-11-12 02:11:13 +03:00
block_until_all_global_resources_loaded=block_until_all_gr,
transfer_files_on_pool_creation=transfer_files_on_pool_creation,
input_data=input_data,
2017-05-01 19:43:41 +03:00
resource_files=resource_files,
2016-11-12 02:11:13 +03:00
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,
2016-11-12 02:11:13 +03:00
generate_docker_tunnel_script=ssh_gen_docker_tunnel,
generated_file_export_path=ssh_gen_file_path,
hpn_server_swap=ssh_hpn,
allow_docker_access=ssh_ada,
2016-11-12 02:11:13 +03:00
),
rdp=RDPSettings(
username=rdp_username,
expiry_days=rdp_expiry_days,
password=rdp_password,
),
2016-11-12 02:11:13 +03:00
gpu_driver=gpu_driver,
additional_node_prep=additional_node_prep,
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'),
2018-03-14 22:21:13 +03:00
remote_access_control=rac,
certificates=certs,
prometheus=prometheus_settings(conf),
upload_diagnostics_logs_on_unusable=_kv_read(
conf, 'upload_diagnostics_logs_on_unusable', default=True),
container_runtimes_install=cr_install,
container_runtimes_default=cr_default,
2018-11-01 02:06:00 +03:00
per_job_auto_scratch=_kv_read(
conf, 'per_job_auto_scratch', default=False),
batch_insights_enabled=_kv_read(
conf, 'batch_insights_enabled', default=False),
2019-08-14 01:06:50 +03:00
public_ips=_kv_read_checked(conf, 'public_ips'),
2016-11-12 02:11:13 +03:00
)
2017-10-04 18:59:03 +03:00
def set_attempt_recovery_on_unusable(config, flag):
# type: (dict, bool) -> None
"""Set attempt recovery on unusable setting
:param dict config: configuration object
:param bool flag: flag to set
"""
config['pool_specification']['attempt_recovery_on_unusable'] = flag
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
"""
2017-06-06 18:29:44 +03:00
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
"""
2017-06-06 18:29:44 +03:00
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
"""
2017-06-06 18:29:44 +03:00
conf = pool_vm_configuration(config, 'platform_image')
try:
sku = str(conf['sku'])
if util.is_none_or_empty(sku):
raise KeyError()
except (KeyError, TypeError):
sku = None
2017-06-06 18:29:44 +03:00
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
2017-01-05 21:20:13 +03:00
"""Get raw credentials dictionary
:param dict config: configuration object
:param bool omit_keyvault: omit keyvault settings if present
2017-01-05 21:20:13 +03:00
:rtype: dict
:return: credentials dict
"""
conf = config['credentials']
if omit_keyvault:
conf.pop('keyvault', None)
return conf
def determine_cloud_type_from_aad(config):
# type: (dict) -> str
"""Determine cloud type from aad settings
:param dict config: configuration object
:rtype: str
:return: cloud type string
"""
auth_url = credentials_management(
config).aad.authority_url.rstrip('/').lower()
if auth_url.endswith('.com'):
cloud_type = 'public'
elif auth_url.endswith('.cn'):
cloud_type = 'china'
elif auth_url.endswith('.de'):
cloud_type = 'germany'
elif auth_url.endswith('.us'):
cloud_type = 'usgov'
else:
raise ValueError('unknown sovereign cloud authority url: {}'.format(
auth_url))
return cloud_type
def _aad_credentials(
conf, service, default_endpoint=None, default_token_cache_file=None):
# type: (dict, str) -> AADSettings
"""Retrieve AAD Settings
:param dict conf: credentials configuration object
:param str service: credentials section service name
:param str default_endpoint: default endpoint
:param str default_token_cache_file: default token cache file
:rtype: AADSettings
:return: AAD settings
"""
super_aad = _kv_read_checked(conf, 'aad', default={})
if service in conf:
service_aad = _kv_read_checked(conf[service], 'aad', default={})
else:
service_aad = {}
if util.is_not_empty(super_aad) or util.is_not_empty(service_aad):
aad_directory_id = (
_kv_read_checked(service_aad, 'directory_id') or
_kv_read_checked(super_aad, 'directory_id')
)
aad_application_id = (
_kv_read_checked(service_aad, 'application_id') or
_kv_read_checked(super_aad, 'application_id')
)
aad_auth_key = (
_kv_read_checked(service_aad, 'auth_key') or
_kv_read_checked(super_aad, 'auth_key')
)
aad_user = (
_kv_read_checked(service_aad, 'user') or
_kv_read_checked(super_aad, 'user')
)
aad_password = (
_kv_read_checked(service_aad, 'password') or
_kv_read_checked(super_aad, 'password')
)
aad_cert_private_key = (
_kv_read_checked(service_aad, 'rsa_private_key_pem') or
_kv_read_checked(super_aad, 'rsa_private_key_pem')
)
aad_cert_thumbprint = (
_kv_read_checked(service_aad, 'x509_cert_sha1_thumbprint') or
_kv_read_checked(super_aad, 'x509_cert_sha1_thumbprint')
)
aad_authority_url = (
_kv_read_checked(service_aad, 'authority_url') or
_kv_read_checked(
super_aad, 'authority_url',
default='https://login.microsoftonline.com')
)
aad_endpoint = _kv_read_checked(
service_aad, 'endpoint', default=default_endpoint)
token_cache = _kv_read_checked(service_aad, 'token_cache', default={})
if _kv_read(token_cache, 'enabled', default=True):
token_cache_file = _kv_read_checked(
token_cache, 'filename', default=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,
authority_url=aad_authority_url,
)
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,
authority_url=None,
)
def credentials_keyvault(config):
2017-03-03 07:17:35 +03:00
# type: (dict) -> KeyVaultCredentialsSettings
"""Get KeyVault settings
:param dict config: configuration object
2017-03-03 07:17:35 +03:00
:rtype: KeyVaultCredentialsSettings
:return: Key Vault settings
"""
try:
creds = config['credentials']
except (KeyError, TypeError):
creds = {}
conf = _kv_read_checked(creds, 'keyvault', default={})
2017-03-03 07:17:35 +03:00
return KeyVaultCredentialsSettings(
aad=_aad_credentials(
creds,
'keyvault',
default_endpoint='https://vault.azure.net',
default_token_cache_file=(
'.batch_shipyard_aad_keyvault_token.json'
),
),
keyvault_uri=_kv_read_checked(conf, 'uri'),
keyvault_credentials_secret_id=_kv_read_checked(
conf, 'credentials_secret_id'),
)
2017-01-05 21:20:13 +03:00
2017-03-03 07:17:35 +03:00
def credentials_management(config):
# type: (dict) -> ManagementCredentialsSettings
"""Get Management settings
:param dict config: configuration object
:rtype: ManagementCredentialsSettings
:return: Management settings
"""
try:
creds = config['credentials']
2017-03-03 07:17:35 +03:00
except (KeyError, TypeError):
creds = {}
conf = _kv_read_checked(creds, 'management', default={})
2017-03-03 07:17:35 +03:00
return ManagementCredentialsSettings(
aad=_aad_credentials(
creds,
'management',
default_endpoint='https://management.azure.com/',
default_token_cache_file=(
'.batch_shipyard_aad_management_token.json'
),
),
subscription_id=_kv_read_checked(conf, 'subscription_id'),
2017-03-03 07:17:35 +03:00
)
def parse_batch_service_url(account_service_url, test_cluster=False):
# type: (str, bool) -> Tuple[str, str]
"""Parse batch service url into account name and location
:param str account_service_url: account url
:param bool test_cluster: test cluster
:rtype: tuple
:return: account, location
"""
# parse location from url
tmp = account_service_url.split('.')
location = tmp[1].lower()
# parse account name from url
if test_cluster:
account = account_service_url.split('/')[-1]
else:
account = tmp[0].split('/')[-1]
return account, location
def credentials_batch(config):
# type: (dict) -> BatchCredentialsSettings
"""Get Batch credentials
:param dict config: configuration object
:rtype: BatchCredentialsSettings
:return: batch creds
"""
try:
creds = config['credentials']
except (KeyError, TypeError):
raise ValueError('credentials not specified')
# allow conf to not exist due to later keyvault load
conf = _kv_read_checked(creds, 'batch', default={})
2017-03-08 22:13:09 +03:00
account_key = _kv_read_checked(conf, 'account_key')
account_service_url = conf['account_service_url']
resource_group = _kv_read_checked(conf, 'resource_group')
test_cluster = _kv_read(conf, 'test_cluster', default=False)
# get subscription id from management section
try:
subscription_id = _kv_read_checked(
creds['management'], 'subscription_id')
except (KeyError, TypeError):
subscription_id = None
account, location = parse_batch_service_url(
account_service_url, test_cluster=test_cluster)
aad = _aad_credentials(
creds,
'batch',
default_endpoint='https://batch.core.windows.net/',
default_token_cache_file=(
'.batch_shipyard_aad_batch_token.json'
),
)
if util.is_not_empty(account_key) and util.is_not_empty(aad.directory_id):
raise ValueError(
'Both account_key and aad settings specified for batch '
'credentials. If using Azure Active Directory, then do not '
'specify an account_key.')
app_insights = _kv_read_checked(conf, 'application_insights', default={})
return BatchCredentialsSettings(
aad=aad,
account=account,
2017-03-08 22:13:09 +03:00
account_key=account_key,
account_service_url=conf['account_service_url'],
resource_group=resource_group,
location=location,
subscription_id=subscription_id,
app_insights_instrumentation_key=_kv_read_checked(
app_insights, 'instrumentation_key'),
app_insights_application_id=_kv_read_checked(
app_insights, 'application_id'),
)
2017-01-05 21:20:13 +03:00
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_aad(config):
# type: (dict) -> AADSettings
"""Get storage AAD credentials
:param dict config: configuration object
:rtype: AADSettings
:return: storage aad settings
"""
if 'aad' in config['credentials']['storage']:
return _aad_credentials(
config['credentials'],
'storage',
default_endpoint='https://management.azure.com/',
default_token_cache_file=(
'.batch_shipyard_aad_storage_token.json'
),
)
else:
return _aad_credentials(
config['credentials'],
'management',
default_endpoint='https://management.azure.com/',
default_token_cache_file=(
'.batch_shipyard_aad_storage_token.json'
),
)
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
"""
try:
conf = config['credentials']['storage'][ssel]
except KeyError:
raise ValueError(
('Could not find storage account alias {} in credentials:storage '
'configuration. Please ensure the storage account alias '
'exists.').format(ssel))
return StorageCredentialsSettings(
account=conf['account'],
account_key=_kv_read_checked(conf, 'account_key'),
endpoint=_kv_read_checked(
conf, 'endpoint', default='core.windows.net'),
resource_group=_kv_read_checked(conf, 'resource_group'),
)
2017-01-05 21:20:13 +03:00
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']:
if conf == 'aad':
continue
2017-01-05 21:20:13 +03:00
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(config, ssel, sakey, ep=None):
# type: (dict, str, str, str) -> None
"""Set Storage account key and endpoint
2017-01-05 21:20:13 +03:00
:param dict config: configuration object
:param str ssel: storage selector link
:param str sakey: storage account key
:param str ep: endpoint
2017-01-05 21:20:13 +03:00
"""
config['credentials']['storage'][ssel]['account_key'] = sakey
if util.is_not_empty(ep):
config['credentials']['storage'][ssel]['endpoint'] = ep
2017-01-05 21:20:13 +03:00
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 singularity_registry_login(config, server):
# type: (dict, str) -> tuple
"""Get singularity 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']['singularity_registry'][
server]['username']
pw = config['credentials']['singularity_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 credentials_iterate_registry_servers(config, is_docker):
# type: (dict, bool) -> str
"""Iterate registry servers
2017-01-05 21:20:13 +03:00
:param dict config: configuration object
:param bool is_docker: is a docker registry
2017-01-05 21:20:13 +03:00
:rtype: str
:return: registry server name
2017-01-05 21:20:13 +03:00
"""
if is_docker:
kind = 'docker_registry'
else:
kind = 'singularity_registry'
2017-01-05 21:20:13 +03:00
try:
for conf in config['credentials'][kind]:
2017-01-05 21:20:13 +03:00
yield conf
except KeyError:
pass
def credentials_registry_password_secret_id(config, link, is_docker):
# type: (dict, str, bool) -> str
"""Get registry password KeyVault Secret Id
2017-01-05 21:20:13 +03:00
:param dict config: configuration object
:param str link: registry link
:param bool is_docker: is docker registry
2017-01-05 21:20:13 +03:00
:rtype: str
:return: keyvault secret id
"""
if is_docker:
kind = 'docker_registry'
else:
kind = 'singularity_registry'
2017-01-05 21:20:13 +03:00
try:
secid = config['credentials'][kind][link][
'password_keyvault_secret_id']
2017-01-05 21:20:13 +03:00
if util.is_none_or_empty(secid):
raise KeyError()
except KeyError:
return None
return secid
def set_credentials_registry_password(config, link, is_docker, password):
# type: (dict, str, bool, str) -> None
"""Set registry password
2017-01-05 21:20:13 +03:00
:param dict config: configuration object
:param str link: registry link
:param bool is_docker: is docker registry
2017-01-05 21:20:13 +03:00
:param str password: password
"""
if is_docker:
kind = 'docker_registry'
else:
kind = 'singularity_registry'
config['credentials'][kind][link]['password'] = password
2017-01-05 21:20:13 +03:00
def credentials_slurm(config):
# type: (dict) -> SlurmCredentialsSettings
"""Get slurm settings
:param dict config: configuration object
:rtype: SlurmCredentialsSettings
:return: Slurm settings
"""
try:
creds = config['credentials']
except (KeyError, TypeError):
creds = {}
conf = _kv_read_checked(creds, 'slurm', default={})
return SlurmCredentialsSettings(
db_password=_kv_read_checked(conf, 'db_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'
return BatchShipyardSettings(
storage_account_settings=stlink,
storage_entity_prefix=sep,
generated_sas_expiry_days=_kv_read(conf, 'generated_sas_expiry_days'),
use_shipyard_docker_image=_kv_read(
conf, 'use_shipyard_docker_image', default=True),
store_timing_metrics=_kv_read(
conf, 'store_timing_metrics', default=False),
fallback_registry=_kv_read_checked(conf, 'fallback_registry'),
delay_docker_image_preload=_kv_read(
conf, 'delay_docker_image_preload', default=False),
)
def requires_populate_global_resources_storage(config):
# type: (dict) -> bool
"""Requires populating global resources in storage
:param dict config: configuration object
:rtype: bool
:return: if populating gr is required
"""
pool = pool_settings(config)
native = is_native_docker_pool(config, vm_config=pool.vm_configuration)
if not native:
return True
bs = batch_shipyard_settings(config)
is_windows = is_windows_pool(config, vm_config=pool.vm_configuration)
return bs.delay_docker_image_preload and not is_windows
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
2016-11-12 02:11:13 +03:00
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_registries(config, images=None):
# type: (dict, List[str]) -> list
"""Get Docker registries specified
:param dict config: configuration object
:param list images: list of images to base return
:rtype: list
:return: list of batchmodels.ContainerRegistry objects
"""
servers = []
if images is None:
# get fallback docker registry
bs = batch_shipyard_settings(config)
if util.is_not_empty(bs.fallback_registry):
servers.append(bs.fallback_registry)
# get additional docker registries
try:
servers.extend(
config['global_resources']['additional_registries']['docker'])
except KeyError:
pass
images = global_resources_docker_images(config)
# parse images for servers
for image in images:
tmp = image.split('/')
if len(tmp) > 1:
if '.' in tmp[0] or ':' in tmp[0] and tmp[0] != 'localhost':
servers.append(tmp[0])
# create unique set
servers = set(servers)
# get login info for each registry
registries = []
# add docker hub if found
hubuser, hubpw = docker_registry_login(config, 'hub')
if util.is_not_empty(hubuser) or util.is_not_empty(hubpw):
registries.append(
batchmodels.ContainerRegistry(
registry_server=None,
user_name=hubuser,
password=hubpw,
)
)
del hubuser
del hubpw
for server in servers:
user, pw = docker_registry_login(config, server)
if util.is_none_or_empty(user) or util.is_none_or_empty(pw):
# registries can be public with a specified server
continue
registries.append(
batchmodels.ContainerRegistry(
registry_server=server,
user_name=user,
password=pw,
)
)
return registries
def singularity_registries(config, images=None):
# type: (dict, List[str]) -> list
"""Get Singularity registries specified
:param dict config: configuration object
:param list images: list of images to base return
:rtype: list
:return: list of batchmodels.ContainerRegistry objects
"""
servers = []
if images is None:
try:
servers.extend(
config['global_resources']['additional_registries'][
'singularity'])
except KeyError:
pass
images = global_resources_singularity_images(config)
# parse images for servers
for image in images:
_, _, image_name = (
image.partition('://'))
tmp = image_name.split('/')
if len(tmp) > 1:
if '.' in tmp[0] or ':' in tmp[0] and tmp[0] != 'localhost':
servers.append(tmp[0])
# create unique set
servers = set(servers)
# get login info for each registry
registries = []
# add docker hub if found and no servers are specified
if len(servers) == 0:
hubuser, hubpw = docker_registry_login(config, 'hub')
if util.is_not_empty(hubuser) or util.is_not_empty(hubpw):
registries.append(
batchmodels.ContainerRegistry(
registry_server=None,
user_name=hubuser,
password=hubpw,
)
)
del hubuser
del hubpw
for server in servers:
user, pw = singularity_registry_login(config, server)
if util.is_none_or_empty(user) or util.is_none_or_empty(pw):
# registries can be public with a specified server
continue
registries.append(
batchmodels.ContainerRegistry(
registry_server=server,
user_name=user,
password=pw,
)
)
return registries
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']
if concurrent_source_downloads is None:
raise KeyError()
except KeyError:
concurrent_source_downloads = 10
return DataReplicationSettings(
concurrent_source_downloads=concurrent_source_downloads,
)
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_singularity_images(config):
# type: (dict) -> list
"""Get list of all singularity images (signed and unsigned)
:param dict config: configuration object
:rtype: list
:return: all singularity images (signed and unsigned)
"""
global_resources = _kv_read_checked(config, 'global_resources', default={})
singularity_images = (
_kv_read_checked(global_resources, 'singularity_images', default={}))
singularity_unsigned_images = (
_kv_read_checked(singularity_images, 'unsigned', default=[]))
singularity_signed_images_settings = (
global_resources_singularity_signed_images_settings(config))
singularity_signed_images = (
[settings.image for settings in singularity_signed_images_settings])
images = singularity_unsigned_images + singularity_signed_images
singularity_signed_and_unsigned_images = (
set(singularity_unsigned_images).intersection(
singularity_signed_images))
if len(singularity_signed_and_unsigned_images):
raise ValueError(
'image(s) "{}" should not be both signed and unsigned'
.format('", "'.join(singularity_signed_and_unsigned_images)))
return images
def global_resources_singularity_signed_images_settings(config):
# type: (dict) -> list
"""Get list of singularity signed images settings
:param dict config: configuration object
:rtype: list
:return: singularity signed images settings
"""
global_resources = _kv_read_checked(config, 'global_resources', default={})
singularity_images = _kv_read_checked(
global_resources, 'singularity_images', default={})
singularity_signed_images = _kv_read_checked(
singularity_images, 'signed', default=[])
singularity_signed_images_settings = []
for settings in singularity_signed_images:
image = _kv_read_checked(settings, 'image')
if image is None:
raise ValueError('singularity signed image is invalid')
key_fingerprint = _kv_read_checked(settings, 'key_fingerprint')
if key_fingerprint is None:
raise ValueError('key_fingerprint for singularity signed image'
' "{}" is invalid'.format(image))
key_file = _kv_read_checked(settings, 'key_file')
key_file_path = None
if key_file is not None:
key_file_path = pathlib.Path(key_file)
if not key_file_path.is_file():
raise ValueError('invalid key file for image "{}"'
.format(image))
singularity_signed_images_settings.append(
SingularitySignedImageSettings(
image=image,
key_fingerprint=key_fingerprint,
key_file=key_file_path,
)
)
return singularity_signed_images_settings
def singularity_signed_images_key_fingerprint_dict(config):
# type: (dict) -> dict
"""Get dict of singularity signed images to key fingerprint
:param dict config: configuration object
:rtype: dict
:return: singularity signed images to key fingerprint
"""
images_settings = (
global_resources_singularity_signed_images_settings(config))
return dict((settings.image, settings.key_fingerprint)
for settings in images_settings)
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
"""
source = _kv_read_checked(conf, 'source', default={})
path = _kv_read_checked(source, 'path')
if util.is_none_or_empty(path):
raise ValueError('global resource files path is invalid')
return SourceSettings(
path=path,
include=_kv_read_checked(source, 'include'),
exclude=_kv_read_checked(source, '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
data_transfer = _kv_read_checked(conf, 'data_transfer', default={})
method = _kv_read_checked(data_transfer, 'method')
if util.is_none_or_empty(method):
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
else:
method = method.lower()
ssh_eo = _kv_read_checked(
data_transfer, 'scp_ssh_extra_options', default='')
rsync_eo = _kv_read_checked(
data_transfer, 'rsync_extra_options', default='')
try:
mpt = 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 = 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(data_transfer, 'ssh_private_key')
if util.is_not_empty(ssh_private_key):
ssh_private_key = pathlib.Path(ssh_private_key)
return DestinationSettings(
storage_account_settings=storage,
shared_data_volume=shared,
relative_destination_path=rdp,
data_transfer=DataTransferSettings(
is_file_share=data_is_file_share(data_transfer),
remote_path=data_remote_path(data_transfer),
blobxfer_extra_options=data_blobxfer_extra_options(data_transfer),
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,
)
)
2017-10-17 22:57:02 +03:00
def _global_resources_volumes(config):
# type: (dict) -> dict
"""Get global resources volumes dictionary
:param dict config: configuration object
:rtype: dict
:return: volumes
"""
try:
vols = config['global_resources']['volumes']
if util.is_none_or_empty(vols):
raise KeyError()
except KeyError:
vols = {}
2017-10-17 22:57:02 +03:00
return vols
def global_resources_data_volumes(config):
# type: (dict) -> dict
"""Get data volumes dictionary
:param dict config: configuration object
:rtype: dict
:return: data volumes
"""
try:
2017-10-17 22:57:02 +03:00
dv = _global_resources_volumes(config)['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:
2017-10-17 22:57:02 +03:00
sdv = _global_resources_volumes(config)['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:
if is_shared_data_volume_custom_linux_mount(sdv, sdvkey):
mo = sdv[sdvkey]['fstab_entry']['fs_mntops']
else:
mo = sdv[sdvkey]['mount_options']
except KeyError:
mo = None
return mo
def azure_storage_account_settings(sdv, sdvkey):
# type: (dict, str) -> str
"""Get azure 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 azure_blob_container_name(sdv, sdvkey):
# type: (dict, str) -> str
"""Get azure blob container name
:param dict sdv: shared_data_volume configuration object
:param str sdvkey: key to sdv
:rtype: str
:return: azure blob container name
"""
return sdv[sdvkey]['azure_blob_container_name']
def azure_file_host_mount_path(storage_account_name, share_name, is_windows):
# type: (str, str, bool) -> str
"""Get azure file share host mount path
:param str storage_account_name: storage account name
:param str share_name: file share name
:param bool is_windows: is windows
:rtype: str
:return: host mount path for azure file share
"""
return '{root}{sep}azfile-{sa}-{share}'.format(
root=get_host_mounts_path(is_windows),
sep='\\' if is_windows else '/',
sa=storage_account_name,
share=share_name)
def azure_blob_host_mount_path(storage_account_name, container_name):
# type: (str, str) -> str
"""Get azure blob container host mount path
:param str storage_account_name: storage account name
:param str container_name: container name
:rtype: str
:return: host mount path for azure file share
"""
return '{root}/azblob-{sa}-{cont}'.format(
root=get_host_mounts_path(False),
sa=storage_account_name,
cont=container_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 custom_linux_mount_fstab_options(sdv, sdvkey):
# type: (dict, str) -> str
"""Get custom mount fstab options
:param dict sdv: shared_data_volume configuration object
:param str sdvkey: key to sdv
:rtype: str
:return: custom mount fstab options
"""
try:
fstab = sdv[sdvkey]['fstab_entry']
if util.is_none_or_empty(fstab):
raise KeyError()
fs_spec = _kv_read_checked(fstab, 'fs_spec')
if util.is_none_or_empty(fs_spec):
raise ValueError(
('fs_spec for fstab_entry of custom mount {} is '
'invalid').format(sdvkey))
fs_vfstype = _kv_read_checked(fstab, 'fs_vfstype')
if util.is_none_or_empty(fs_vfstype):
raise ValueError(
('fs_vfstype for fstab_entry of custom mount {} is '
'invalid').format(sdvkey))
fs_mntops = _kv_read_checked(fstab, 'fs_mntops', default='defaults')
fs_freq = _kv_read(fstab, 'fs_freq', default=0)
fs_passno = _kv_read(fstab, 'fs_passno', default=0)
except KeyError:
return None
return CustomMountFstabSettings(
fs_spec=fs_spec,
fs_vfstype=fs_vfstype,
fs_mntops=fs_mntops,
fs_freq=fs_freq,
fs_passno=fs_passno,
)
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_azure_blob(sdv, sdvkey):
# type: (dict, str) -> bool
"""Determine if shared data volume is an azure blob container via fuse
:param dict sdv: shared_data_volume configuration object
:param str sdvkey: key to sdv
:rtype: bool
:return: if shared data volume is azure blob
"""
return shared_data_volume_driver(sdv, sdvkey).lower() == 'azureblob'
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'
def is_shared_data_volume_custom_linux_mount(sdv, sdvkey):
# type: (dict, str) -> bool
"""Determine if shared data volume is a custom linux mount
:param dict sdv: shared_data_volume configuration object
:param str sdvkey: key to sdv
:rtype: bool
:return: if shared data volume is a custom linux mount
"""
return (
shared_data_volume_driver(sdv, sdvkey).lower() == 'custom_linux_mount'
)
# 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_remote_path(conf):
# type: (dict) -> str
"""Retrieve remote path on Azure Storage for data transfer
:param dict conf: configuration object
:rtype: str
:return: remote path
"""
return _kv_read_checked(conf, 'remote_path')
def data_container_from_remote_path(conf, rp=None):
# type: (dict, str) -> str
"""Get Container or File share name from remote path
:param dict conf: configuration object
:param str rp: remote path
:rtype: str
:return: container/fshare name
"""
if rp is None:
rp = data_remote_path(conf)
if util.is_none_or_empty(rp):
raise ValueError(
2017-10-04 18:59:03 +03:00
'cannot derive container name from invalid remote_path')
return rp.split('/')[0]
def data_local_path(conf, on_task, task_wd=True):
# type: (dict, bool) -> str
"""Retrieve local path for data transfer
:param dict conf: configuration object
:param bool on_task: if input data is on the task spec
:param bool task_wd: if path is not specified use task working dir, else
use task dir
:rtype: str
:return: local path
"""
try:
dst = conf['local_path']
if util.is_none_or_empty(dst):
raise KeyError()
except KeyError:
if on_task:
if task_wd:
dst = '$AZ_BATCH_TASK_WORKING_DIR'
else:
dst = '$AZ_BATCH_TASK_DIR'
else:
raise ValueError(
'local_path not specified for input_data or output_data '
'for storage_account_settings {}'.format(
conf['storage_account_settings']))
return dst
def data_is_file_share(conf):
# type: (dict) -> bool
"""Retrieve if data transfer originates/destined for file share
:param dict conf: configuration object
:rtype: bool
:return: is Azure file share
"""
return _kv_read(conf, 'is_file_share', default=False)
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
"""
return _kv_read_checked(conf, 'blobxfer_extra_options', default='')
def data_include(conf):
# type: (dict) -> str
"""Retrieve input data include filters
:param dict conf: configuration object
:rtype: str
:return: include filters
"""
2017-11-15 08:26:52 +03:00
return _kv_read_checked(conf, 'include', [])
def data_exclude(conf):
# type: (dict) -> str
"""Retrieve input data exclude filters
:param dict conf: configuration object
:rtype: str
:return: exclude filters
"""
2017-11-15 08:26:52 +03:00
return _kv_read_checked(conf, 'exclude', [])
def data_condition(conf):
# type: (dict) -> str
"""Retrieve output data condition
:param dict conf: configuration object
:rtype: str
:return: condition
"""
return _kv_read_checked(conf, 'condition', default='tasksuccess')
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']
# JOBS SETTINGS
def job_specifications(config):
# type: (dict) -> dict
"""Get job specifications config block
:param dict config: configuration object
:rtype: dict
:return: job specifications
"""
try:
return config['job_specifications']
except KeyError:
raise ValueError(
'job_specifications is not found or invalid, did you specify a '
'jobs configuration file?')
2017-07-31 22:46:41 +03:00
def autogenerated_task_id_prefix(config):
# type: (dict) -> str
"""Get the autogenerated task id prefix to use
2017-07-31 22:46:41 +03:00
:param dict config: configuration object
:rtype: str
:return: auto-gen task id prefix
2017-07-31 22:46:41 +03:00
"""
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)
2017-07-31 22:46:41 +03:00
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']
drp = data_remote_path(az)
tfstorage = TaskFactoryStorageSettings(
storage_settings=credentials_storage(
config, data_storage_account_settings(az)),
storage_link_name=az['storage_account_settings'],
container=data_container_from_remote_path(None, drp),
remote_path=drp,
is_file_share=data_is_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):
task['##tfgen'] = True
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
2017-08-08 05:42:09 +03:00
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')
)
2017-08-08 05:42:09 +03:00
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),
2017-08-08 05:42:09 +03:00
)
)
else:
return None
2017-07-20 19:34:44 +03:00
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
2018-11-01 02:06:00 +03:00
def job_requires_auto_scratch(conf):
# type: (dict) -> bool
"""Get job auto scratch setting
:param dict conf: job configuration object
:rtype: bool
:return: job auto scratch
"""
return _kv_read(conf, 'auto_scratch', default=False)
def job_preparation_command(conf):
# type: (dict) -> str
"""Get arbitrary job preparation command
:param dict conf: job configuration object
:rtype: str
:return: job prep command
"""
return _kv_read_checked(
_kv_read_checked(conf, 'job_preparation', default={}),
'command')
def job_release_command(conf):
# type: (dict) -> str
"""Get arbitrary job release command
:param dict conf: job configuration object
:rtype: str
:return: job release command
"""
return _kv_read_checked(
_kv_read_checked(conf, 'job_release', default={}),
'command')
def job_federation_constraint_settings(conf, federation_id):
# type: (dict, str) -> dict
"""Gets federation constraints
:param dict conf: job configuration object
:param str federation_id: federation id
:rtype: dict
:return: federation constraints
"""
if util.is_none_or_empty(federation_id):
return None
fc_conf = _kv_read_checked(conf, 'federation_constraints', default={})
pool_conf = _kv_read_checked(fc_conf, 'pool', default={})
native = _kv_read(pool_conf, 'native')
windows = _kv_read(pool_conf, 'windows')
if windows and native is not None and not native:
raise ValueError(
'cannot set constraint windows as true and native as false')
pool_location = _kv_read_checked(pool_conf, 'location')
if pool_location is not None:
if ' ' in pool_location:
raise ValueError(
'pool:location "{}" is invalid, please ensure proper region '
'name and not its display name'.format(pool_location))
pool_location = pool_location.lower()
pool_custom_image_arm_id = _kv_read_checked(
pool_conf, 'custom_image_arm_id')
if pool_custom_image_arm_id is not None:
pool_custom_image_arm_id = pool_custom_image_arm_id.lower()
pool_virtual_network_arm_id = _kv_read_checked(
pool_conf, 'virtual_network_arm_id')
if pool_virtual_network_arm_id is not None:
pool_virtual_network_arm_id = pool_virtual_network_arm_id.lower()
pool_lp_conf = _kv_read_checked(
pool_conf, 'low_priority_nodes', default={})
lp_allow = _kv_read(pool_lp_conf, 'allow', default=True)
lp_exclusive = _kv_read(pool_lp_conf, 'exclusive', default=False)
if not lp_allow and lp_exclusive:
raise ValueError(
'cannot set constraint low_priority:allow to false and '
'low_priority:exclusive to true')
pool_as_conf = _kv_read_checked(
pool_conf, 'autoscale', default={})
autoscale_allow = _kv_read(pool_as_conf, 'allow', default=True)
autoscale_exclusive = _kv_read(pool_as_conf, 'exclusive', default=False)
if not autoscale_allow and autoscale_exclusive:
raise ValueError(
'cannot set constraint autoscale:allow to false and '
'autoscale:exclusive to true')
pool_reg_conf = _kv_read_checked(
pool_conf, 'container_registries', default={})
pool_mab_conf = _kv_read_checked(
pool_conf, 'max_active_task_backlog', default={})
matbr = _kv_read(pool_mab_conf, 'ratio')
if matbr is not None:
matbr = float(matbr)
if matbr < 0:
raise ValueError(
'cannot set constraint max_active_task_backlog:ratio to '
'a negative value')
matbae = _kv_read(
pool_mab_conf, 'autoscale_exempt', default=True)
node_conf = _kv_read_checked(fc_conf, 'compute_node', default={})
vm_size = _kv_read_checked(node_conf, 'vm_size')
if vm_size is not None:
vm_size = vm_size.lower()
core_conf = _kv_read_checked(node_conf, 'cores', default={})
node_cores = _kv_read(core_conf, 'amount')
if util.is_not_empty(vm_size) and node_cores is not None:
raise ValueError(
'cannot specify both vm_size and cores for compute_node '
'constraint')
node_core_variance = _kv_read(core_conf, 'schedulable_variance')
if node_core_variance is not None:
node_core_variance = float(node_core_variance)
if node_core_variance < 0:
raise ValueError(
'cannot specify a negative cores:schedulable_variance')
memory_conf = _kv_read(node_conf, 'memory', default={})
node_memory = _kv_read_checked(memory_conf, 'amount')
if util.is_not_empty(vm_size) and node_memory is not None:
raise ValueError(
'cannot specify both vm_size and memory for compute_node '
'constraint')
if node_memory is not None:
node_memory = node_memory.lower()
if node_memory[-1] not in ('b', 'k', 'm', 'g', 't'):
raise ValueError(
'federation_constraints:compute_node:memory has invalid '
'suffix')
if int(node_memory[:-1]) <= 0:
raise ValueError(
'federation_constraints:compute_node:memory is a '
'non-positive value')
node_memory_variance = _kv_read(memory_conf, 'schedulable_variance')
if node_memory_variance is not None:
node_memory_variance = float(node_memory_variance)
if node_memory_variance < 0:
raise ValueError(
'cannot specify a negative memory:schedulable_variance')
node_gpu = _kv_read(node_conf, 'gpu')
if node_gpu and util.is_not_empty(vm_size) and not is_gpu_pool(vm_size):
raise ValueError(
'cannot specify gpu=True while vm_size does not have GPUs')
node_ib = _kv_read(node_conf, 'infiniband')
if node_ib and util.is_not_empty(vm_size) and not is_rdma_pool(vm_size):
raise ValueError(
'cannot specify infiniband=True while vm_size does not have '
'RDMA/IB')
return FederationConstraintSettings(
pool=FederationPoolConstraintSettings(
native=native,
windows=windows,
location=pool_location,
custom_image_arm_id=pool_custom_image_arm_id,
virtual_network_arm_id=pool_virtual_network_arm_id,
low_priority_nodes_allow=lp_allow,
low_priority_nodes_exclusive=lp_exclusive,
autoscale_allow=autoscale_allow,
autoscale_exclusive=autoscale_exclusive,
container_registries_private_docker_hub=_kv_read(
pool_reg_conf, 'private_docker_hub', default=False),
container_registries_public=_kv_read_checked(
pool_reg_conf, 'public'),
max_active_task_backlog_ratio=matbr,
max_active_task_backlog_autoscale_exempt=matbae,
),
compute_node=FederationComputeNodeConstraintSettings(
vm_size=vm_size,
cores=node_cores,
core_variance=node_core_variance,
memory=node_memory,
memory_variance=node_memory_variance,
exclusive=_kv_read(node_conf, 'exclusive', default=False),
gpu=node_gpu,
infiniband=node_ib,
),
)
def job_has_merge_task(conf):
# type: (dict) -> bool
"""Determines if job has a merge task
:param dict conf: job configuration object
:rtype: bool
:return: job has merge task
"""
try:
merge = conf['merge_task']
except KeyError:
return False
else:
if any(x in merge for x in _FORBIDDEN_MERGE_TASK_PROPERTIES):
raise ValueError(
'merge_task has one or more forbidden properties: {}'.format(
_FORBIDDEN_MERGE_TASK_PROPERTIES))
return True
def job_merge_task(conf):
# type: (dict) -> dict
"""Gets merge task
:param dict conf: job configuration object
:rtype: dict
:return: merge task
"""
return conf['merge_task']
def job_force_enable_task_dependencies(conf):
# type: (dict) -> bool
"""If task dependencies are force enabled
:param dict conf: task configuration object
:rtype: bool
:return: jobs with force enabled task dependencies
"""
return _kv_read(conf, 'force_enable_task_dependencies', default=False)
def has_depends_on_task(conf):
# type: (dict) -> bool
"""Determines if task has task dependencies
:param dict conf: task configuration object
:rtype: bool
:return: task has task dependencies
"""
2017-01-12 20:23:25 +03:00
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'])) and
('##tfgen' not in conf or not conf['##tfgen'])):
raise ValueError(
2017-01-12 20:23:25 +03:00
'task id is not specified, but depends_on or '
'depends_on_range is set')
return True
return False
def has_task_exit_condition_job_action(jobspec, conf):
# type: (dict, dict) -> bool
"""Determines if task has task exit condition job action
:param dict jobspec: job configuration object
:param dict conf: task configuration object
:rtype: bool
:return: task has exit condition job action
"""
try:
conf['exit_conditions']['default']['exit_options']['job_action']
except KeyError:
try:
jobspec['exit_conditions']['default']['exit_options']['job_action']
except KeyError:
return False
return True
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
"""
return (
_kv_read_checked(conf, 'docker_image') or
_kv_read_checked(conf, 'image')
)
def task_singularity_image(conf):
# type: (dict) -> str
"""Get singularity image used by task
:param dict conf: task configuration object
:rtype: str
:return: singularity image used by task
"""
return _kv_read_checked(conf, 'singularity_image')
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, federation_id=None):
# type: (azure.batch.models.CloudPool, dict, PoolSettings, dict,
# dict, str) -> 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 str federation_id: federation id
:rtype: TaskSettings
:return: task settings
"""
native = is_native_docker_pool(config, vm_config=poolconf.vm_configuration)
is_windows = is_windows_pool(config, vm_config=poolconf.vm_configuration)
# 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')
docker_image = task_docker_image(conf)
singularity_image = _kv_read_checked(conf, 'singularity_image')
if (util.is_none_or_empty(docker_image) and
util.is_none_or_empty(singularity_image)):
raise ValueError('Container image is unspecified or invalid')
if (util.is_not_empty(docker_image) and
util.is_not_empty(singularity_image)):
raise ValueError(
'Cannot specify both a Docker and Singularity image for a task')
if util.is_not_empty(singularity_image) and native:
raise ValueError(
'Cannot run Singularity containers on native container '
'support pools')
if is_windows and util.is_not_empty(singularity_image):
raise ValueError(
'Cannot run Singularity containers on windows pools')
# get some pool props
publisher = None
offer = None
node_agent = None
if cloud_pool is None:
pool_id = poolconf.id
vm_size = poolconf.vm_size
inter_node_comm = poolconf.inter_node_communication_enabled
2017-10-04 18:59:03 +03:00
is_custom_image = not is_platform_image(
config, vm_config=poolconf.vm_configuration)
2017-06-06 18:29:44 +03:00
if is_custom_image:
node_agent = poolconf.vm_configuration.node_agent
else:
publisher = poolconf.vm_configuration.publisher.lower()
offer = poolconf.vm_configuration.offer.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 = util.is_not_empty(
cloud_pool.virtual_machine_configuration.image_reference.
virtual_machine_image_id)
2017-06-06 18:29:44 +03:00
if is_custom_image:
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()
# get federation job constraint overrides
if util.is_not_empty(federation_id):
fed_constraints = job_federation_constraint_settings(
jobspec, federation_id)
if fed_constraints.pool.native is not None:
native = fed_constraints.pool.native
if fed_constraints.pool.windows is not None:
is_windows = fed_constraints.pool.windows
is_custom_image = util.is_not_empty(
fed_constraints.pool.custom_image_arm_id)
if is_multi_instance_task(conf):
inter_node_comm = True
else:
fed_constraints = None
# get depends on
try:
depends_on = conf['depends_on']
if util.is_none_or_empty(depends_on):
raise KeyError()
except KeyError:
depends_on = None
2017-01-12 20:23:25 +03:00
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)):
2017-01-12 20:23:25 +03:00
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
docker_exec_options = []
singularity_cmd = None
run_elevated = True
if util.is_not_empty(docker_image):
run_opts = _kv_read_checked(
conf, 'additional_docker_run_options', default=[])
if '--privileged' in run_opts:
docker_exec_options.append('--privileged')
else:
run_opts = _kv_read_checked(
conf, 'additional_singularity_options', default=[])
singularity_execution = _kv_read_checked(
conf, 'singularity_execution', default={})
singularity_cmd = _kv_read_checked(
singularity_execution, 'cmd', default='exec')
run_elevated = _kv_read(
singularity_execution, 'elevated', default=False)
if singularity_cmd not in _SINGULARITY_COMMANDS:
raise ValueError('singularity_cmd is invalid: {}'.format(
singularity_cmd))
# docker specific options
name = None
if util.is_not_empty(docker_image):
# parse remove container option
rm_container = _kv_read(conf, 'remove_container_after_exit')
if rm_container is None:
rm_container = _kv_read(
jobspec, 'remove_container_after_exit', default=True)
if rm_container and '--rm' not in run_opts:
run_opts.append('--rm')
del rm_container
# parse /dev/shm option
shm_size = (
_kv_read(conf, 'shm_size') or
_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
name = _kv_read_checked(conf, 'name')
if util.is_none_or_empty(name):
name = task_id
set_task_name(conf, name)
run_opts.append('--name {}'.format(name))
# parse labels option
labels = _kv_read_checked(conf, 'labels')
if util.is_not_empty(labels):
for label in labels:
run_opts.append('-l {}'.format(label))
del labels
# parse ports option
ports = _kv_read_checked(conf, 'ports')
if util.is_not_empty(ports):
for port in ports:
run_opts.append('-p {}'.format(port))
del ports
# parse entrypoint
entrypoint = _kv_read_checked(conf, 'entrypoint')
if util.is_not_empty(entrypoint):
run_opts.append('--entrypoint {}'.format(entrypoint))
del entrypoint
# get user identity settings
if not is_windows:
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,
)
# append user identity options
uiopt = None
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
uiopt = '-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')
uiopt = '-u {}:{}'.format(
ui.specific_user_uid, ui.specific_user_gid)
attach_ui = True
if util.is_not_empty(uiopt):
run_opts.append(uiopt)
docker_exec_options.append(uiopt)
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
del uiopt
# get command
command = _kv_read_checked(conf, 'command')
# parse data volumes
data_volumes = _kv_read_checked(jobspec, 'data_volumes')
tdv = _kv_read_checked(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
# binding order matters for Singularity
bindparm = '-v' if util.is_not_empty(docker_image) else '-B'
# get working dir default
def_wd = _kv_read_checked(
conf, 'default_working_dir',
default=_kv_read_checked(jobspec, 'default_working_dir')
)
if util.is_none_or_empty(def_wd) or def_wd == 'batch':
if is_windows:
def_wd = '%AZ_BATCH_TASK_WORKING_DIR%'
else:
def_wd = '$AZ_BATCH_TASK_WORKING_DIR'
# set working directory if not already set
if def_wd != 'container':
if util.is_not_empty(docker_image):
if not any((x.startswith('-w ') or x.startswith('--workdir '))
for x in run_opts):
run_opts.append('-w {}'.format(def_wd))
else:
if not any(x.startswith('--pwd ') for x in run_opts):
run_opts.append('--pwd {}'.format(def_wd))
working_dir = (
batchmodels.ContainerWorkingDirectory.task_working_directory
)
else:
working_dir = (
batchmodels.ContainerWorkingDirectory.container_image_default
)
del def_wd
# bind root dir and set working dir
if not native:
restrict_bind = _kv_read(
jobspec, 'restrict_default_bind_mounts', default=False)
if restrict_bind:
# mount task directory only
if is_windows:
run_opts.append(
'{} %AZ_BATCH_TASK_DIR%:%AZ_BATCH_TASK_DIR%'.format(
bindparm))
else:
run_opts.append(
'{} $AZ_BATCH_TASK_DIR:$AZ_BATCH_TASK_DIR'.format(
bindparm))
else:
# mount batch root dir
if is_windows:
run_opts.append(
'{} %AZ_BATCH_NODE_ROOT_DIR%:'
'%AZ_BATCH_NODE_ROOT_DIR%'.format(bindparm))
else:
run_opts.append(
'{} $AZ_BATCH_NODE_ROOT_DIR:'
'$AZ_BATCH_NODE_ROOT_DIR'.format(bindparm))
if util.is_not_empty(data_volumes):
dv = global_resources_data_volumes(config)
for dvkey in data_volumes:
try:
hostpath = _kv_read_checked(dv[dvkey], 'host_path')
except KeyError:
raise ValueError(
('ensure that the {} data volume exists in the '
'global configuration').format(dvkey))
bindopt = _kv_read_checked(dv[dvkey], 'bind_options', default='')
if util.is_not_empty(bindopt):
bindopt = ':{}'.format(bindopt)
if util.is_not_empty(hostpath):
run_opts.append('{} {}:{}{}'.format(
bindparm, hostpath, dv[dvkey]['container_path'], bindopt))
else:
if util.is_not_empty(bindopt):
run_opts.append('{bp} {cp}:{cp}{bo}'.format(
bp=bindparm, cp=dv[dvkey]['container_path'],
bo=bindopt))
else:
run_opts.append('{} {}'.format(
bindparm, dv[dvkey]['container_path']))
del data_volumes
# parse shared data volumes
shared_data_volumes = _kv_read_checked(jobspec, 'shared_data_volumes')
tsdv = _kv_read_checked(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
2018-11-01 02:06:00 +03:00
if job_requires_auto_scratch(jobspec):
run_opts.append(
'{} {}/auto_scratch/{}:$AZ_BATCH_TASK_DIR/auto_scratch'.format(
bindparm,
_HOST_MOUNTS_DIR,
jobspec['id']))
if util.is_not_empty(shared_data_volumes):
sdv = global_resources_shared_data_volumes(config)
for sdvkey in shared_data_volumes:
try:
bindopt = _kv_read_checked(
sdv[sdvkey], 'bind_options', default='')
except KeyError:
raise ValueError(
('ensure that the {} shared data volume exists in the '
'global configuration').format(sdvkey))
if util.is_not_empty(bindopt):
bindopt = ':{}'.format(bindopt)
if is_shared_data_volume_gluster_on_compute(sdv, sdvkey):
run_opts.append('{} {}/{}:{}{}'.format(
bindparm,
_HOST_MOUNTS_DIR,
get_gluster_on_compute_volume(),
shared_data_volume_container_path(sdv, sdvkey),
bindopt))
2017-03-13 12:10:08 +03:00
elif is_shared_data_volume_storage_cluster(sdv, sdvkey):
run_opts.append('{} {}/{}:{}{}'.format(
bindparm,
_HOST_MOUNTS_DIR,
2017-03-13 12:10:08 +03:00
sdvkey,
shared_data_volume_container_path(sdv, sdvkey),
bindopt))
elif is_shared_data_volume_azure_blob(sdv, sdvkey):
sa = credentials_storage(
config,
azure_storage_account_settings(sdv, sdvkey))
cont_name = azure_blob_container_name(sdv, sdvkey)
hmp = azure_blob_host_mount_path(sa.account, cont_name)
run_opts.append('{} {}:{}{}'.format(
bindparm,
hmp,
shared_data_volume_container_path(sdv, sdvkey),
bindopt))
elif is_shared_data_volume_azure_file(sdv, sdvkey):
sa = credentials_storage(
config,
azure_storage_account_settings(sdv, sdvkey))
share_name = azure_file_share_name(sdv, sdvkey)
hmp = azure_file_host_mount_path(
sa.account, share_name, is_windows)
run_opts.append('{} {}:{}{}'.format(
bindparm,
hmp,
shared_data_volume_container_path(sdv, sdvkey),
bindopt))
elif is_shared_data_volume_custom_linux_mount(sdv, sdvkey):
run_opts.append('{} {}/{}:{}{}'.format(
bindparm,
_HOST_MOUNTS_DIR,
sdvkey,
shared_data_volume_container_path(sdv, sdvkey),
bindopt))
else:
raise RuntimeError(
'unknown shared data volume type: {}'.format(sdvkey))
del shared_data_volumes
# env vars
env_vars = _kv_read_checked(conf, 'environment_variables', default={})
ev_secid = _kv_read_checked(
conf, 'environment_variables_keyvault_secret_id')
# singularity login
if util.is_not_empty(singularity_image):
registry_type, _, image_name = singularity_image.partition('://')
if (registry_type.lower() == 'oras' or
registry_type.lower() == 'docker'):
registry = image_name.partition('/')[0]
username, password = singularity_registry_login(config, registry)
if username is not None and password is not None:
env_vars['SINGULARITY_DOCKER_USERNAME'] = username
env_vars['SINGULARITY_DOCKER_PASSWORD'] = password
# constraints
max_task_retries = _kv_read(conf, 'max_task_retries')
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)
retention_time = (
_kv_read_checked(conf, 'retention_time') or
_kv_read_checked(jobspec, 'retention_time')
)
if util.is_not_empty(retention_time):
retention_time = util.convert_string_to_timedelta(retention_time)
# exit conditions, right now specific exit codes/ranges are not supported
job_default_eo = _kv_read_checked(
_kv_read_checked(
_kv_read_checked(
jobspec,
'exit_conditions',
default={}
),
'default',
default={}
),
'exit_options',
default={}
)
task_default_eo = _kv_read_checked(
_kv_read_checked(
_kv_read_checked(
conf,
'exit_conditions',
default={}
),
'default',
default={}
),
'exit_options',
default={}
)
job_action = batchmodels.JobAction(
_kv_read_checked(
task_default_eo,
'job_action',
default=batchmodels.JobAction(
_kv_read_checked(job_default_eo, 'job_action', default='none')
)
)
)
dependency_action = batchmodels.DependencyAction(
_kv_read_checked(
task_default_eo,
'dependency_action',
default=batchmodels.DependencyAction(
_kv_read_checked(
job_default_eo, 'dependency_action', default='block')
)
)
)
# gpu
gpu = _kv_read(conf, 'gpus')
if gpu is None:
gpu = _kv_read(jobspec, 'gpus')
if gpu is not None:
gpu = str(gpu)
# if not specified check for gpu pool and implicitly enable
if util.is_none_or_empty(gpu):
if is_gpu_pool(vm_size) and not is_windows:
gpu = 'all'
else:
gpu = 'disable'
# adjust for gpu settings
if util.is_not_empty(gpu) and gpu != 'disable':
if util.is_not_empty(federation_id):
# ensure that the job-level constraint does not conflict with
# job/task level requirements
if (fed_constraints.compute_node.gpu is not None and
not fed_constraints.compute_node.gpu):
raise ValueError(
'job or task requirement of gpu conflicts with '
'compute_node:gpu federation constraint')
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))
if is_windows:
raise ValueError(
'cannot execute a gpu task on a windows pool: {}'.format(
pool_id))
# set docker commands with nvidia container runtime
if util.is_not_empty(singularity_image):
if gpu != 'all':
raise ValueError(
'cannot execute a singularity container with non-default '
'gpu options (i.e., "all")')
run_opts.append('--nv')
else:
# batch native mode will take care of using the proper runtime
# TODO once native mode supports new native docker gpu support
# this should change and support gpus options
if not native:
run_opts.append('--gpus={}'.format(gpu))
# infiniband
infiniband = _kv_read(conf, 'infiniband')
if infiniband is None:
_kv_read(jobspec, 'infiniband')
# if not specified, check for rdma pool and implicitly enable
if infiniband is None:
if is_rdma_pool(vm_size) and inter_node_comm and not is_windows:
infiniband = True
else:
infiniband = False
# adjust for infiniband
if infiniband:
# adjust ib with fed constraints (normalize to a known base config)
if util.is_not_empty(federation_id):
# ensure that the job-level constraint does not conflict with
# job/task level requirements
if (fed_constraints.compute_node.infiniband is not None and
not fed_constraints.compute_node.infiniband):
raise ValueError(
'job or task requirement of infiniband conflicts with '
'compute_node:infiniband federation constraint')
# set publisher and offer (or node agent)
if infiniband:
if is_custom_image:
node_agent = 'batch.node.centos'
else:
publisher = 'openlogic'
offer = 'centos-hpc'
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))
# mount /opt/intel for NetworkDirect RDMA
if is_networkdirect_rdma_pool(vm_size):
run_opts.append('{} /opt/intel:/opt/intel:ro'.format(bindparm))
# until Batch supports SRIOV natively for Docker, add additional
# IB devices and rdma dat regardless of pool native mode
if is_sriov_rdma_pool(vm_size):
if util.is_not_empty(docker_image):
run_opts.append('--device=/dev/infiniband/issm0')
run_opts.append('--device=/dev/infiniband/ucm0')
run_opts.append('--device=/dev/infiniband/umad0')
if (native and
(((publisher == 'openlogic' and offer == 'centos-hpc') or
(publisher == 'microsoft-azure-batch' and
offer == 'centos-container-rdma')) or
(is_custom_image and
node_agent.startswith('batch.node.centos')))):
run_opts.append('{} /etc/dat.conf:/etc/dat.conf:ro'.format(
bindparm))
if not native:
if util.is_not_empty(docker_image):
# 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')
else:
# ensure singularity opts do not have network namespace
# or contain options
try:
run_opts.remove('-c')
except ValueError:
pass
try:
run_opts.remove('--contain')
except ValueError:
pass
try:
run_opts.remove('-C')
except ValueError:
pass
try:
run_opts.remove('--containall')
except ValueError:
pass
try:
run_opts.remove('-n')
except ValueError:
pass
try:
run_opts.remove('--net')
except ValueError:
pass
# add rdma dat files into container space
if (((publisher == 'openlogic' and offer == 'centos-hpc') or
(publisher == 'microsoft-azure-batch' and
offer == 'centos-container-rdma')) or
(is_custom_image and
node_agent.startswith('batch.node.centos'))):
if is_networkdirect_rdma_pool(vm_size):
run_opts.append('{} /etc/rdma:/etc/rdma:ro'.format(
bindparm))
run_opts.append(
'{} /etc/rdma/dat.conf:/etc/dat.conf:ro'.format(
bindparm))
elif is_sriov_rdma_pool(vm_size):
run_opts.append('{} /etc/dat.conf:/etc/dat.conf:ro'.format(
bindparm))
elif ((publisher == 'microsoft-azure-batch' and
offer == 'ubuntu-server-container-rdma') or
(is_custom_image and
node_agent.startswith('batch.node.ubuntu'))):
if is_networkdirect_rdma_pool(vm_size):
run_opts.append('{} /etc/dat.conf:/etc/dat.conf:ro'.format(
bindparm))
run_opts.append(
'{} /etc/dat.conf:/etc/rdma/dat.conf:ro'.format(
bindparm))
elif is_sriov_rdma_pool(vm_size):
2019-10-16 18:49:41 +03:00
run_opts.append('{} /etc/dat.conf:/etc/dat.conf:ro'.format(
bindparm))
elif ((publisher == 'suse' and offer == 'sles-hpc') or
(is_custom_image and
node_agent.startswith('batch.node.opensuse'))):
run_opts.append('{} /etc/dat.conf:/etc/dat.conf:ro'.format(
bindparm))
run_opts.append(
'{} /etc/dat.conf:/etc/rdma/dat.conf:ro'.format(bindparm))
if util.is_not_empty(docker_image):
run_opts.append('--device=/dev/hvnd_rdma')
else:
raise ValueError(
('Unsupported infiniband VM config, publisher={} '
'offer={}').format(publisher, offer))
# always add option for envfile
envfile = '.shipyard.envlist'
if util.is_not_empty(docker_image) and not native:
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))
# Docker container must be named
if util.is_not_empty(docker_image):
if util.is_none_or_empty(name):
raise ValueError(
'multi-instance task with a Docker image must be invoked '
'with a named container')
# application command cannot be empty/None
if util.is_none_or_empty(command):
raise ValueError(
'multi-instance task must have an application command')
2017-09-27 20:56:51 +03:00
# set docker run options for coordination command
if util.is_not_empty(docker_image):
if not native:
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')
else:
# ensure network namespace is not enabled
2017-09-27 20:56:51 +03:00
try:
run_opts.remove('-n')
except ValueError:
pass
try:
run_opts.remove('--net')
2017-09-27 20:56:51 +03:00
except ValueError:
pass
# get coordination command
try:
coordination_command = conf[
'multi_instance']['coordination_command']
if util.is_none_or_empty(coordination_command):
raise KeyError()
coordination_command = '{}'.format(' ' + coordination_command)
except KeyError:
2017-09-27 20:56:51 +03:00
# manually set coordination command to ssh for native
# containers in daemon mode if not specified
if native:
coordination_command = '/usr/sbin/sshd -p 23'
else:
coordination_command = ''
if native or util.is_not_empty(singularity_image):
2017-09-27 20:56:51 +03:00
if util.is_not_empty(coordination_command):
cc_args = [coordination_command]
else:
cc_args = None
else:
cc_args = [
'docker run {} {}{}'.format(
' '.join(run_opts),
docker_image,
coordination_command),
]
del coordination_command
# get num instances
num_instances = conf['multi_instance']['num_instances']
if not isinstance(num_instances, int):
if util.is_not_empty(federation_id):
raise ValueError(
'cannot specify a non-integral value "{}" for '
'num_instances for a multi-instance task destined for '
'a federation'.format(num_instances))
# 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
pre_execution_command = _kv_read_checked(
conf['multi_instance'], 'pre_execution_command', None)
mpi = _kv_read(conf['multi_instance'], 'mpi', None)
if mpi is not None:
Fixes and update of recipes (#290) * Fix multi-instance tasks that are not a MPI task * Add setup task script for CNTK-CPU-Infiniband-IntelMPI * Update CNTK-CPU-Infiniband-IntelMPI recipe * Add MPI executable path option * Update CNTK-CPU-OpenMPI recipe * Change the default MPI executable_path to mpirun * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-Infiniband-IntelMPI * Update CNTK-GPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-OpenMPI * Add setup task script for NAMD-Infiniband-IntelMPI * Update NAMD-Infiniband-IntelMPI recipe * Add setup task script for OpenFOAM-Infiniband-IntelMPI * Update OpenFOAM-Infiniband-IntelMPI recipe * Update TensorFlow-GPU Singularity recipe * Add setup task script for OpenFOAM-TCP-OpenMPI * Update OpenFOAM-TCP-OpenMPI recipe * Add support for arbitrary commands with the MPI processes_per_node option * Fix MPI with native images * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Modify CNTK-GPU-Infiniband-IntelMPI recipe * Modify NAMD-Infiniband-IntelMPI recipe * Update processes_per_node documentation * Fix `pool images list` with Singularity images * Modify OpenFOAM-Infiniband-IntelMPI set up script * Add check for mpi setting with Windows * Add auto scratch support with OpenFOAM-Infiniband-IntelMPI recipe * Modify OpenFOAM-TCP-OpenMPI set up script * Add auto scratch support with OpenFOAM-TCP-OpenMPI recipe * Add mpiBench-IntelMPI recipe * Add mpiBench-MPICH recipe * Add mpiBench-OpenMPI recipe * Resolve PR comments * Resolve PR comments
2019-07-18 04:57:06 +03:00
if is_windows:
raise ValueError(
'The mpi setting is not supported on windows pools')
mpi_runtime = _kv_read_checked(mpi, 'runtime', '').lower()
Fixes and update of recipes (#290) * Fix multi-instance tasks that are not a MPI task * Add setup task script for CNTK-CPU-Infiniband-IntelMPI * Update CNTK-CPU-Infiniband-IntelMPI recipe * Add MPI executable path option * Update CNTK-CPU-OpenMPI recipe * Change the default MPI executable_path to mpirun * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-Infiniband-IntelMPI * Update CNTK-GPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-OpenMPI * Add setup task script for NAMD-Infiniband-IntelMPI * Update NAMD-Infiniband-IntelMPI recipe * Add setup task script for OpenFOAM-Infiniband-IntelMPI * Update OpenFOAM-Infiniband-IntelMPI recipe * Update TensorFlow-GPU Singularity recipe * Add setup task script for OpenFOAM-TCP-OpenMPI * Update OpenFOAM-TCP-OpenMPI recipe * Add support for arbitrary commands with the MPI processes_per_node option * Fix MPI with native images * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Modify CNTK-GPU-Infiniband-IntelMPI recipe * Modify NAMD-Infiniband-IntelMPI recipe * Update processes_per_node documentation * Fix `pool images list` with Singularity images * Modify OpenFOAM-Infiniband-IntelMPI set up script * Add check for mpi setting with Windows * Add auto scratch support with OpenFOAM-Infiniband-IntelMPI recipe * Modify OpenFOAM-TCP-OpenMPI set up script * Add auto scratch support with OpenFOAM-TCP-OpenMPI recipe * Add mpiBench-IntelMPI recipe * Add mpiBench-MPICH recipe * Add mpiBench-OpenMPI recipe * Resolve PR comments * Resolve PR comments
2019-07-18 04:57:06 +03:00
mpi_executable_path = _kv_read_checked(
mpi, 'executable_path', 'mpirun').lower()
mpi_options = _kv_read_checked(mpi, 'options', [])
mpi_ppn = _kv_read(mpi, 'processes_per_node', None)
else:
num_instances = 0
cc_args = None
mi_resource_files = None
pre_execution_command = None
mpi = None
return TaskSettings(
id=task_id,
docker_image=docker_image,
singularity_image=singularity_image,
name=name,
run_options=run_opts,
docker_exec_options=docker_exec_options,
working_dir=working_dir,
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,
2017-01-12 20:23:25 +03:00
depends_on_range=depends_on_range,
singularity_cmd=singularity_cmd,
run_elevated=run_elevated,
multi_instance=MultiInstanceSettings(
num_instances=num_instances,
coordination_command=cc_args,
resource_files=mi_resource_files,
pre_execution_command=pre_execution_command,
mpi=None if mpi is None else MpiSettings(
runtime=mpi_runtime,
Fixes and update of recipes (#290) * Fix multi-instance tasks that are not a MPI task * Add setup task script for CNTK-CPU-Infiniband-IntelMPI * Update CNTK-CPU-Infiniband-IntelMPI recipe * Add MPI executable path option * Update CNTK-CPU-OpenMPI recipe * Change the default MPI executable_path to mpirun * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-Infiniband-IntelMPI * Update CNTK-GPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-OpenMPI * Add setup task script for NAMD-Infiniband-IntelMPI * Update NAMD-Infiniband-IntelMPI recipe * Add setup task script for OpenFOAM-Infiniband-IntelMPI * Update OpenFOAM-Infiniband-IntelMPI recipe * Update TensorFlow-GPU Singularity recipe * Add setup task script for OpenFOAM-TCP-OpenMPI * Update OpenFOAM-TCP-OpenMPI recipe * Add support for arbitrary commands with the MPI processes_per_node option * Fix MPI with native images * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Modify CNTK-GPU-Infiniband-IntelMPI recipe * Modify NAMD-Infiniband-IntelMPI recipe * Update processes_per_node documentation * Fix `pool images list` with Singularity images * Modify OpenFOAM-Infiniband-IntelMPI set up script * Add check for mpi setting with Windows * Add auto scratch support with OpenFOAM-Infiniband-IntelMPI recipe * Modify OpenFOAM-TCP-OpenMPI set up script * Add auto scratch support with OpenFOAM-TCP-OpenMPI recipe * Add mpiBench-IntelMPI recipe * Add mpiBench-MPICH recipe * Add mpiBench-OpenMPI recipe * Resolve PR comments * Resolve PR comments
2019-07-18 04:57:06 +03:00
executable_path=mpi_executable_path,
options=mpi_options,
Fixes and update of recipes (#290) * Fix multi-instance tasks that are not a MPI task * Add setup task script for CNTK-CPU-Infiniband-IntelMPI * Update CNTK-CPU-Infiniband-IntelMPI recipe * Add MPI executable path option * Update CNTK-CPU-OpenMPI recipe * Change the default MPI executable_path to mpirun * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-Infiniband-IntelMPI * Update CNTK-GPU-Infiniband-IntelMPI recipe * Add setup task script for CNTK-GPU-OpenMPI * Add setup task script for NAMD-Infiniband-IntelMPI * Update NAMD-Infiniband-IntelMPI recipe * Add setup task script for OpenFOAM-Infiniband-IntelMPI * Update OpenFOAM-Infiniband-IntelMPI recipe * Update TensorFlow-GPU Singularity recipe * Add setup task script for OpenFOAM-TCP-OpenMPI * Update OpenFOAM-TCP-OpenMPI recipe * Add support for arbitrary commands with the MPI processes_per_node option * Fix MPI with native images * Modify CNTK-CPU-Infiniband-IntelMPI recipe * Modify CNTK-GPU-Infiniband-IntelMPI recipe * Modify NAMD-Infiniband-IntelMPI recipe * Update processes_per_node documentation * Fix `pool images list` with Singularity images * Modify OpenFOAM-Infiniband-IntelMPI set up script * Add check for mpi setting with Windows * Add auto scratch support with OpenFOAM-Infiniband-IntelMPI recipe * Modify OpenFOAM-TCP-OpenMPI set up script * Add auto scratch support with OpenFOAM-TCP-OpenMPI recipe * Add mpiBench-IntelMPI recipe * Add mpiBench-MPICH recipe * Add mpiBench-OpenMPI recipe * Resolve PR comments * Resolve PR comments
2019-07-18 04:57:06 +03:00
processes_per_node=mpi_ppn
),
),
default_exit_options=TaskExitOptions(
job_action=job_action,
dependency_action=dependency_action,
),
)
# 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', {})
arm_subnet_id = _kv_read_checked(conf, 'arm_subnet_id')
name = _kv_read_checked(conf, 'name')
if util.is_not_empty(arm_subnet_id) and util.is_not_empty(name):
raise ValueError(
'cannot specify both arm_subnet_id and virtual_network.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')
if util.is_not_empty(name) and util.is_none_or_empty(subnet_name):
raise ValueError(
'subnet name not specified on virtual_network: {}'.format(name))
subnet_address_prefix = _kv_read_checked(sub_conf, 'address_prefix')
return VirtualNetworkSettings(
arm_subnet_id=arm_subnet_id,
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):
2017-03-17 01:12:22 +03:00
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):
2017-03-17 01:12:22 +03:00
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,
),
)
2017-03-17 01:12:22 +03:00
def remotefs_settings(config, sc_id=None):
# type: (dict, str) -> RemoteFsSettings
"""Get remote fs settings
:param dict config: configuration dict
2017-03-17 01:12:22 +03:00
:param str sc_id: storage cluster id
:rtype: RemoteFsSettings
:return: remote fs settings
"""
# general settings
try:
conf = config['remote_fs']
if util.is_none_or_empty(conf):
raise KeyError
except KeyError:
raise ValueError(
'remote_fs settings are invalid or missing. Did you specify an '
'fs configuration file?')
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')
zone = _kv_read(conf, 'zone')
# 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_sku = _kv_read(md_conf, 'sku')
if util.is_none_or_empty(md_sku):
raise ValueError('invalid managed_disks:sku in remote_fs')
md_disk_size_gb = _kv_read(md_conf, 'disk_size_gb')
md_disk_pp = _kv_read(md_conf, 'disk_provisioned_performance', default={})
md_disk_pp_iops = _kv_read(md_disk_pp, 'iops_read_write')
md_disk_pp_mbps = _kv_read(md_disk_pp, 'mbps_read_write')
md_disk_names = _kv_read_checked(md_conf, 'disk_names')
2017-03-17 01:12:22 +03:00
md = ManagedDisksSettings(
location=location,
2017-03-17 01:12:22 +03:00
resource_group=md_rg,
zone=zone,
sku=md_sku,
2017-03-17 01:12:22 +03:00
disk_size_gb=md_disk_size_gb,
disk_provisioned_perf_iops_rw=md_disk_pp_iops,
disk_provisioned_perf_mbps_rw=md_disk_pp_mbps,
2017-03-17 01:12:22 +03:00
disk_names=md_disk_names,
)
if util.is_none_or_empty(sc_id):
2017-03-17 01:12:22 +03:00
return RemoteFsSettings(
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(sc_rg):
2017-03-17 01:12:22 +03:00
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')
sc_accel_net = _kv_read(sc_conf, 'accelerated_networking', False)
# 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')
2017-03-31 05:48:17 +03:00
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)
2017-03-31 05:48:17 +03:00
_reserved = frozenset([
'ssh', 'nfs', 'glusterfs', 'smb', 'cifs', 'samba', 'zfs',
'beegfs', 'cephfs',
2017-03-31 05:48:17 +03:00
])
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', default=-1)
if len(disk_array) == 1:
if 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:
2017-03-06 11:35:24 +03:00
raise ValueError('RAID-0 arrays require at least two disks')
if raid_level != 0:
raise ValueError('Unsupported RAID level {}'.format(
raid_level))
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(
2017-03-17 01:12:22 +03:00
('Number of entries in vm_disk_map {} inconsistent with '
'vm_count {}').format(len(disk_map), sc_vm_count))
return RemoteFsSettings(
managed_disks=md,
storage_cluster=StorageClusterSettings(
id=sc_id,
location=location,
resource_group=sc_rg,
zone=zone,
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,
accelerated_networking=sc_accel_net,
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,
allow_docker_access=False,
),
vm_disk_map=disk_map,
prometheus=prometheus_settings(sc_conf),
),
)
def monitoring_prometheus_settings(config):
# type: (dict) -> PrometheusMonitoringSettings
"""Get prometheus monitoring settings
:param dict config: configuration dict
:rtype: PrometheusMonitoringSettings
:return: Prometheus monitoring settings
"""
try:
conf = config['monitoring']['services']['prometheus']
except KeyError:
conf = {}
port = None
else:
port = str(_kv_read(conf, 'port'))
return PrometheusMonitoringSettings(
port=port,
scrape_interval=_kv_read_checked(
conf, 'scrape_interval', default='10s'),
)
def credentials_grafana_admin_password_secret_id(config):
# type: (dict) -> str
"""Get Grafana admin password KeyVault Secret Id
:param dict config: configuration object
:rtype: str
:return: keyvault secret id
"""
try:
secid = config[
'credentials']['monitoring']['grafana']['admin'][
'password_keyvault_secret_id']
if util.is_none_or_empty(secid):
raise KeyError()
except KeyError:
return None
return secid
def set_credentials_grafana_admin_password(config, pw):
# type: (dict, str) -> None
"""Set Grafana admin password
:param dict config: configuration object
:param str pw: password
"""
config['credentials']['monitoring']['grafana']['admin']['password'] = pw
def monitoring_grafana_settings(config):
# type: (dict) -> GrafanaMonitoringSettings
"""Get grafana monitoring settings
:param dict config: configuration dict
:rtype: GrafanaMonitoringSettings
:return: Grafana monitoring settings
"""
try:
conf = config['monitoring']['services']['grafana']
except KeyError:
conf = {}
try:
gaconf = config['credentials']['monitoring']['grafana']
except KeyError:
gaconf = {}
admin = _kv_read_checked(gaconf, 'admin', default={})
admin_user = _kv_read_checked(admin, 'username')
if util.is_none_or_empty(admin_user):
raise ValueError('Grafana admin user is invalid')
admin_password = _kv_read_checked(admin, 'password')
if util.is_none_or_empty(admin_password):
raise ValueError('Grafana admin password is invalid')
return GrafanaMonitoringSettings(
admin_user=admin_user,
admin_password=admin_password,
additional_dashboards=_kv_read_checked(conf, 'additional_dashboards'),
)
def monitoring_services_settings(config):
# type: (dict) -> MonitoringServicesSettings
"""Get services monitoring settings
:param dict config: configuration dict
:rtype: MonitoringServicesSettings
:return: Services monitoring settings
"""
try:
conf = config['monitoring']['services']
except KeyError:
conf = {}
le = _kv_read_checked(conf, 'lets_encrypt', default={})
return MonitoringServicesSettings(
resource_polling_interval=str(_kv_read(
conf, 'resource_polling_interval', 15)),
lets_encrypt_enabled=_kv_read(le, 'enabled', default=True),
lets_encrypt_staging=_kv_read(
le, 'use_staging_environment', default=True),
prometheus=monitoring_prometheus_settings(config),
grafana=monitoring_grafana_settings(config),
)
def monitoring_settings(config):
# type: (dict) -> VmResource
"""Get monitoring settings
:param dict config: configuration dict
:rtype: VmResource
:return: VM resource settings
"""
# general settings
try:
conf = config['monitoring']
if util.is_none_or_empty(conf):
raise KeyError
except KeyError:
raise ValueError('monitoring settings are invalid')
location = conf['location']
if util.is_none_or_empty(location):
raise ValueError('invalid location in monitoring')
# monitoring vm settings
rg = _kv_read_checked(conf, 'resource_group')
if util.is_none_or_empty(rg):
raise ValueError('invalid resource_group in monitoring')
vm_size = _kv_read_checked(conf, 'vm_size')
hostname_prefix = _kv_read_checked(conf, 'hostname_prefix')
accel_net = _kv_read(conf, 'accelerated_networking', False)
# public ip settings
pip_conf = _kv_read_checked(conf, 'public_ip', {})
pip_enabled = _kv_read(pip_conf, 'enabled', True)
pip_static = _kv_read(pip_conf, 'static', False)
# sc network security settings
ns_conf = conf['network_security']
ns_inbound = {
'ssh': InboundNetworkSecurityRule(
destination_port_range='22',
source_address_prefix=_kv_read_checked(ns_conf, 'ssh', ['*']),
protocol='tcp',
),
}
if not isinstance(ns_inbound['ssh'].source_address_prefix, list):
raise ValueError('expected list for ssh network security rule')
if 'grafana' in ns_conf:
ns_inbound['grafana'] = InboundNetworkSecurityRule(
# grafana is reverse proxied through nginx on the HTTPS port
destination_port_range='443' if pip_enabled else '3000',
source_address_prefix=_kv_read_checked(ns_conf, 'grafana'),
protocol='tcp',
)
if not isinstance(ns_inbound['grafana'].source_address_prefix, list):
raise ValueError('expected list for grafana network security rule')
promconf = monitoring_prometheus_settings(config)
if promconf.port is not None:
if 'prometheus' not in ns_conf:
raise ValueError(
'prometheus port specified, but no network security '
'rule exists')
else:
ns_inbound['prometheus'] = InboundNetworkSecurityRule(
destination_port_range=promconf.port,
source_address_prefix=_kv_read_checked(ns_conf, 'prometheus'),
protocol='tcp',
)
if not isinstance(
ns_inbound['prometheus'].source_address_prefix, list):
raise ValueError(
'expected list for prometheus network security rule')
if 'custom_inbound_rules' in ns_conf:
for key in ns_conf['custom_inbound_rules']:
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(ns_inbound[key].source_address_prefix, list):
raise ValueError(
'expected list for network security rule {} '
'source_address_prefix'.format(key))
# ssh settings
ssh_conf = conf['ssh']
ssh_username = _kv_read_checked(ssh_conf, 'username')
ssh_public_key = _kv_read_checked(ssh_conf, '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(ssh_conf, 'ssh_public_key_data')
ssh_private_key = _kv_read_checked(ssh_conf, '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_file_path = _kv_read_checked(
ssh_conf, 'generated_file_export_path', '.')
return VmResource(
location=location,
resource_group=rg,
zone=_kv_read(conf, 'zone'),
hostname_prefix=hostname_prefix,
virtual_network=virtual_network_settings(
conf,
default_resource_group=rg,
default_existing_ok=False,
default_create_nonexistant=True,
),
network_security=NetworkSecuritySettings(
inbound=ns_inbound,
),
vm_size=vm_size,
accelerated_networking=accel_net,
public_ip=PublicIpSettings(
enabled=pip_enabled,
static=pip_static,
),
ssh=SSHSettings(
username=ssh_username,
expiry_days=9999,
ssh_public_key=ssh_public_key,
ssh_public_key_data=ssh_public_key_data,
ssh_private_key=ssh_private_key,
generate_docker_tunnel_script=False,
generated_file_export_path=ssh_gen_file_path,
hpn_server_swap=False,
allow_docker_access=False,
),
)
def federation_proxy_options_settings(config):
# type: (dict) -> FederationProxyOptionsSettings
"""Get federation proxy options settings
:param dict config: configuration dict
:rtype: FederationProxyOptionsSettings
:return: federation proxy options settings
"""
try:
conf = config['federation']['proxy_options']
except KeyError:
conf = {}
pi_conf = _kv_read_checked(conf, 'polling_interval', default={})
fpi = _kv_read(pi_conf, 'federations', 30)
if fpi < 5:
raise ValueError(
'the polling_interval:federations value can not be less than 5')
api = _kv_read(pi_conf, 'actions', 5)
if api < 5:
raise ValueError(
'the polling_interval:actions value can not be less than 5')
log_conf = _kv_read_checked(conf, 'logging', default={})
sched_conf = _kv_read_checked(conf, 'scheduling', default={})
as_conf = _kv_read_checked(sched_conf, 'after_success', default={})
sasbi = _kv_read(as_conf, 'blackout_interval', 15)
if sasbi < 2:
raise ValueError(
'the scheduling:after_success:blackout_interval value can not '
'be less than 2')
return FederationProxyOptionsSettings(
federations_polling_interval=str(fpi),
actions_polling_interval=str(api),
log_persistence=_kv_read(log_conf, 'persistence', True),
log_level=_kv_read_checked(log_conf, 'level', 'debug'),
log_filename=_kv_read_checked(log_conf, 'filename', 'fedproxy.log'),
scheduling_after_success_blackout_interval=sasbi,
scheduling_after_success_evaluate_autoscale=_kv_read(
as_conf, 'evaluate_autoscale', True),
)
def federation_settings(config):
# type: (dict) -> VmResource
"""Get federation settings
:param dict config: configuration dict
:rtype: VmResource
:return: VM resource settings
"""
# general settings
try:
conf = config['federation']
if util.is_none_or_empty(conf):
raise KeyError
except KeyError:
raise ValueError('federation settings are invalid or missing')
location = conf['location']
if util.is_none_or_empty(location):
raise ValueError('invalid location in federation')
# vm settings
rg = _kv_read_checked(conf, 'resource_group')
if util.is_none_or_empty(rg):
raise ValueError('invalid resource_group in federation')
vm_size = _kv_read_checked(conf, 'vm_size')
hostname_prefix = _kv_read_checked(conf, 'hostname_prefix')
accel_net = _kv_read(conf, 'accelerated_networking', False)
# public ip settings
pip_conf = _kv_read_checked(conf, 'public_ip', {})
pip_enabled = _kv_read(pip_conf, 'enabled', True)
pip_static = _kv_read(pip_conf, 'static', False)
# sc network security settings
ns_conf = conf['network_security']
ns_inbound = {
'ssh': InboundNetworkSecurityRule(
destination_port_range='22',
source_address_prefix=_kv_read_checked(ns_conf, 'ssh', ['*']),
protocol='tcp',
),
}
if not isinstance(ns_inbound['ssh'].source_address_prefix, list):
raise ValueError('expected list for ssh network security rule')
if 'custom_inbound_rules' in ns_conf:
for key in ns_conf['custom_inbound_rules']:
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(ns_inbound[key].source_address_prefix, list):
raise ValueError(
'expected list for network security rule {} '
'source_address_prefix'.format(key))
# ssh settings
ssh_conf = conf['ssh']
ssh_username = _kv_read_checked(ssh_conf, 'username')
ssh_public_key = _kv_read_checked(ssh_conf, '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(ssh_conf, 'ssh_public_key_data')
ssh_private_key = _kv_read_checked(ssh_conf, '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_file_path = _kv_read_checked(
ssh_conf, 'generated_file_export_path', '.')
return VmResource(
location=location,
resource_group=rg,
zone=_kv_read(conf, 'zone'),
hostname_prefix=hostname_prefix,
virtual_network=virtual_network_settings(
conf,
default_resource_group=rg,
default_existing_ok=False,
default_create_nonexistant=True,
),
network_security=NetworkSecuritySettings(
inbound=ns_inbound,
),
vm_size=vm_size,
accelerated_networking=accel_net,
public_ip=PublicIpSettings(
enabled=pip_enabled,
static=pip_static,
),
ssh=SSHSettings(
username=ssh_username,
expiry_days=9999,
ssh_public_key=ssh_public_key,
ssh_public_key_data=ssh_public_key_data,
ssh_private_key=ssh_private_key,
generate_docker_tunnel_script=False,
generated_file_export_path=ssh_gen_file_path,
hpn_server_swap=False,
allow_docker_access=False,
),
)
def slurm_options_settings(config):
# type: (dict) -> SlurmOptionsSettings
"""Get slurm options settings
:param dict config: configuration dict
:rtype: SlurmOptionsSettings
:return: slurm options settings
"""
try:
conf = config['slurm']['slurm_options']
except KeyError:
conf = {}
cluster_id = config['slurm']['cluster_id']
if util.is_none_or_empty(cluster_id) or len(cluster_id) > 22:
raise ValueError(
'cluster_id is invalid. Must be between 1 and 22 '
'characters in length')
bc = credentials_batch(config)
idle_reclaim_time = _kv_read(conf, 'idle_reclaim_time', default='00:15:00')
idle_reclaim_time = util.convert_string_to_timedelta(idle_reclaim_time)
if idle_reclaim_time.total_seconds == 0:
raise ValueError('idle_reclaim_time must be positive')
max_nodes = 0
partitions = {}
part_conf = _kv_read_checked(conf, 'elastic_partitions')
for key in part_conf:
part = _kv_read_checked(part_conf, key)
batch_pools = {}
pool_conf = _kv_read_checked(part, 'batch_pools', default={})
for pkey in pool_conf:
bpool = _kv_read_checked(pool_conf, pkey)
batch_service_url = _kv_read_checked(bpool, 'account_service_url')
if util.is_none_or_empty(batch_service_url):
batch_service_url = bc.account_service_url
max_compute_nodes = _kv_read(bpool, 'max_compute_nodes')
reclaim_exclude_num_nodes = _kv_read(
bpool, 'reclaim_exclude_num_nodes', default=0)
if reclaim_exclude_num_nodes > max_compute_nodes:
raise ValueError(
'reclaim_exclude_num_nodes {} > '
'max_compute_nodes {}'.format(
reclaim_exclude_num_nodes, max_compute_nodes))
batch_pools[pkey] = SlurmBatchPoolSettings(
batch_service_url=batch_service_url,
compute_node_type=_kv_read_checked(bpool, 'compute_node_type'),
max_compute_nodes=max_compute_nodes,
weight=_kv_read(bpool, 'weight'),
features=_kv_read_checked(bpool, 'features', default=[]),
reclaim_exclude_num_nodes=reclaim_exclude_num_nodes,
)
max_nodes = max(max_nodes, batch_pools[pkey].max_compute_nodes)
max_runtime_limit = _kv_read_checked(part, 'max_runtime_limit')
if util.is_not_empty(max_runtime_limit):
max_runtime_limit = max_runtime_limit.replace('.', '-')
else:
max_runtime_limit = 'UNLIMITED'
partition = SlurmPartitionSettings(
batch_pools=batch_pools,
max_runtime_limit=max_runtime_limit,
default=_kv_read(part, 'default'),
preempt_type=_kv_read_checked(part, 'preempt_type'),
preempt_mode=_kv_read_checked(part, 'preempt_mode'),
over_subscribe=_kv_read_checked(part, 'over_subscribe'),
priority_tier=_kv_read(part, 'priority_tier'),
other_options=_kv_read_checked(part, 'other_options', default=[]),
)
partitions[key] = partition
unmanaged_partitions = []
upart_conf = _kv_read_checked(conf, 'unmanaged_partitions', default=[])
for upart in upart_conf:
unmanaged_partitions.append(SlurmUnmanagedPartitionSettings(
partition=_kv_read_checked(upart, 'partition'),
nodes=_kv_read_checked(upart, 'nodes'),
))
return SlurmOptionsSettings(
cluster_id=cluster_id,
idle_reclaim_time=idle_reclaim_time,
max_nodes=max_nodes,
elastic_partitions=partitions,
unmanaged_partitions=unmanaged_partitions,
)
def slurm_settings(config, kind):
# type: (dict) -> VmResource
"""Get slurm settings
:param dict config: configuration dict
:rtype: VmResource
:return: VM resource settings
"""
# general settings
try:
conf = config['slurm']
if util.is_none_or_empty(conf):
raise KeyError
except KeyError:
raise ValueError('slurm settings are invalid or missing')
location = conf['location']
if util.is_none_or_empty(location):
raise ValueError('invalid location in slurm')
rg = _kv_read_checked(conf, 'resource_group')
if util.is_none_or_empty(rg):
raise ValueError('invalid resource_group in slurm')
zone = _kv_read(conf, 'zone')
hostname_prefix = '{}-{}'.format(
_kv_read_checked(conf, 'cluster_id'),
# Azure doesn't like "login" for DNS
'gateway' if kind == 'login' else kind
)
# get controller settings
try:
conf = conf[kind]
if util.is_none_or_empty(conf):
raise KeyError
except KeyError:
raise ValueError(
'slurm:{} settings are invalid or missing'.format(kind))
# vm settings
vm_size = _kv_read_checked(conf, 'vm_size')
accel_net = _kv_read(conf, 'accelerated_networking', False)
# public ip settings
pip_conf = _kv_read_checked(conf, 'public_ip', {})
pip_enabled = _kv_read(pip_conf, 'enabled', True)
pip_static = _kv_read(pip_conf, 'static', False)
# sc network security settings
ns_conf = conf['network_security']
ns_inbound = {
'ssh': InboundNetworkSecurityRule(
destination_port_range='22',
source_address_prefix=_kv_read_checked(ns_conf, 'ssh', ['*']),
protocol='tcp',
),
}
if not isinstance(ns_inbound['ssh'].source_address_prefix, list):
raise ValueError('expected list for ssh network security rule')
if 'custom_inbound_rules' in ns_conf:
for key in ns_conf['custom_inbound_rules']:
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(ns_inbound[key].source_address_prefix, list):
raise ValueError(
'expected list for network security rule {} '
'source_address_prefix'.format(key))
# ssh settings
ssh_conf = conf['ssh']
ssh_username = _kv_read_checked(ssh_conf, 'username')
ssh_public_key = _kv_read_checked(ssh_conf, '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(ssh_conf, 'ssh_public_key_data')
ssh_private_key = _kv_read_checked(ssh_conf, '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_file_path = _kv_read_checked(
ssh_conf, 'generated_file_export_path', '.')
return VmResource(
location=location,
resource_group=rg,
zone=zone,
hostname_prefix=hostname_prefix,
virtual_network=virtual_network_settings(
conf,
default_resource_group=rg,
default_existing_ok=False,
default_create_nonexistant=True,
),
network_security=NetworkSecuritySettings(
inbound=ns_inbound,
),
vm_size=vm_size,
accelerated_networking=accel_net,
public_ip=PublicIpSettings(
enabled=pip_enabled,
static=pip_static,
),
ssh=SSHSettings(
username=ssh_username,
expiry_days=9999,
ssh_public_key=ssh_public_key,
ssh_public_key_data=ssh_public_key_data,
ssh_private_key=ssh_private_key,
generate_docker_tunnel_script=False,
generated_file_export_path=ssh_gen_file_path,
hpn_server_swap=False,
allow_docker_access=False,
),
)
def slurm_vm_count(config, kind):
# type: (dict, str) -> int
"""Get Slurm controller vm count
:param dict config: configuration dict
:param str kind: kind
:rtype: int
:return: vm count
"""
conf = _kv_read_checked(_kv_read_checked(config, 'slurm'), kind)
return _kv_read(conf, 'vm_count')
def slurm_additional_prep_script(config, kind):
# type: (dict, str) -> int
"""Get Slurm additional prep script
:param dict config: configuration dict
:param str kind: kind
:rtype: str
:return: prep script location
"""
conf = _kv_read_checked(_kv_read_checked(config, 'slurm'), kind)
return _kv_read(conf, 'additional_prep_script')
def slurm_shared_data_volumes(config):
# type: (dict) -> List[str]
"""Get Slurm shared data volumes
:param dict config: configuration dict
:rtype: List[str]
:return: list of SlurmSharedDataVolumesSettings
"""
conf = _kv_read_checked(config, 'slurm')
sdv = _kv_read_checked(conf, 'shared_data_volumes', default={})
vols = []
state = False
for sdkey in sdv:
host_mount_path = _kv_read_checked(sdv[sdkey], 'host_mount_path')
if host_mount_path == '/home' or host_mount_path == '/home/':
raise ValueError(
'/home host_mount_path for {} is currently not '
'supported'.format(sdkey))
store_slurmctld_state = _kv_read(sdv[sdkey], 'store_slurmctld_state')
if store_slurmctld_state:
if state:
raise ValueError(
'only one shared data volume should be designated as '
'store_slurmctld_state')
state = True
vols.append(SlurmSharedDataVolumesSettings(
id=sdkey,
host_mount_path=host_mount_path,
store_slurmctld_state=store_slurmctld_state,
))
return vols
def other_storage_account_settings(config, key):
# type: (dict, str) ->str
"""Get other storage account settings selector
:param dict config: configuration dict
:param str key: config key
:rtype: str
:return: other storage settings link
"""
try:
conf = config[key]
if util.is_none_or_empty(conf):
raise KeyError
except KeyError:
raise ValueError('{} settings are invalid or missing'.format(key))
ssel = _kv_read_checked(conf, 'storage_account_settings')
if util.is_none_or_empty(ssel):
raise ValueError(
'{} storage_account_settings are invalid or missing'.format(key))
return ssel
def federation_credentials_storage(config):
# type: (dict) -> StorageCredentialsSettings
"""Get federation storage account settings
:param dict config: configuration dict
:rtype: StorageCredentialsSettings
:return: federation storage cred settings
"""
return credentials_storage(
config, other_storage_account_settings(config, 'federation'))
def slurm_credentials_storage(config):
# type: (dict) -> StorageCredentialsSettings
"""Get slurm storage account settings
:param dict config: configuration dict
:rtype: StorageCredentialsSettings
:return: slurm storage cred settings
"""
return credentials_storage(
config, other_storage_account_settings(config, 'slurm'))
def generate_availability_set_name(vr):
# type: (VmResource) -> str
"""Generate an availabilty set name
:param VmResource vr: vm resource
:rtype: str
:return: availability set name
"""
return '{}-as'.format(vr.hostname_prefix)
def generate_virtual_machine_name(vr, i):
# type: (VmResource, int) -> str
"""Generate a virtual machine name
:param VmResource vr: vm resource
:param int i: resource number
:rtype: str
:return: vm name
"""
return '{}-vm{}'.format(vr.hostname_prefix, str(i).zfill(3))
def get_offset_from_virtual_machine_name(vm_name):
# type: (str) -> int
"""Gets the virtual machine offset given a vm name
:param str vm_name: vm name
:rtype: int
:return: vm offset
"""
return int(vm_name.split('-vm')[-1])
def generate_virtual_machine_extension_name(vr, i):
# type: (VmResource, int) -> str
"""Generate a virtual machine extension name
:param VmResource vr: vm resource
:param int i: resource number
:rtype: str
:return: vm extension name
"""
return '{}-vmext{}'.format(vr.hostname_prefix, str(i).zfill(3))
def generate_virtual_machine_msi_extension_name(vr, i):
# type: (VmResource, int) -> str
"""Generate a virtual machine msi extension name
:param VmResource vr: vm resource
:param int i: resource number
:rtype: str
:return: vm extension msi name
"""
return '{}-vmextmsi{}'.format(vr.hostname_prefix, str(i).zfill(3))
def generate_network_security_group_name(vr):
# type: (VmResource) -> str
"""Generate a network security group name
:param VmResource vr: vm resource
:rtype: str
:return: nsg name
"""
return '{}-nsg'.format(vr.hostname_prefix)
def generate_network_security_inbound_rule_name(rule_name, i):
# type: (str, int) -> str
"""Generate a network security inbound rule name
:param str rule_name: rule name
:parm int i: rule number
:rtype: str
:return: inbound rule name
"""
return '{}_in-{}'.format(rule_name, str(i).zfill(3))
def generate_network_security_inbound_rule_description(rule_name, i):
# type: (str, int) -> str
"""Generate a network security inbound rule description
:param str rule_name: rule name
:parm int i: rule number
:rtype: str
:return: inbound description
"""
return '{} inbound ({})'.format(rule_name, str(i).zfill(3))
def generate_public_ip_name(vr, i):
# type: (VmResource, int) -> str
"""Generate a public ip name
:param VmResource vr: vm resource
:parm int i: pip number
:rtype: str
:return: public ip name
"""
return '{}-pip{}'.format(vr.hostname_prefix, str(i).zfill(3))
def generate_hostname(vr, i):
# type: (VmResource, int) -> str
"""Generate a hostname (dns label prefix)
:param VmResource vr: vm resource
:parm int i: hostname number
:rtype: str
:return: hostname
"""
return '{}{}'.format(vr.hostname_prefix, str(i).zfill(3))
def generate_network_interface_name(vr, i):
# type: (VmResource, int) -> str
"""Generate a network inetrface name
:param VmResource vr: vm resource
:parm int i: network interface number
:rtype: str
:return: network interface name
"""
return '{}-ni{}'.format(vr.hostname_prefix, str(i).zfill(3))
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