config.py 48 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468
  1. from __future__ import absolute_import
  2. from __future__ import unicode_literals
  3. import functools
  4. import io
  5. import logging
  6. import os
  7. import string
  8. import sys
  9. from collections import namedtuple
  10. import six
  11. import yaml
  12. from cached_property import cached_property
  13. from . import types
  14. from .. import const
  15. from ..const import COMPOSEFILE_V1 as V1
  16. from ..const import COMPOSEFILE_V2_1 as V2_1
  17. from ..const import COMPOSEFILE_V2_3 as V2_3
  18. from ..const import COMPOSEFILE_V3_0 as V3_0
  19. from ..const import COMPOSEFILE_V3_4 as V3_4
  20. from ..utils import build_string_dict
  21. from ..utils import json_hash
  22. from ..utils import parse_bytes
  23. from ..utils import parse_nanoseconds_int
  24. from ..utils import splitdrive
  25. from ..version import ComposeVersion
  26. from .environment import env_vars_from_file
  27. from .environment import Environment
  28. from .environment import split_env
  29. from .errors import CircularReference
  30. from .errors import ComposeFileNotFound
  31. from .errors import ConfigurationError
  32. from .errors import DuplicateOverrideFileFound
  33. from .errors import VERSION_EXPLANATION
  34. from .interpolation import interpolate_environment_variables
  35. from .sort_services import get_container_name_from_network_mode
  36. from .sort_services import get_service_name_from_network_mode
  37. from .sort_services import sort_service_dicts
  38. from .types import MountSpec
  39. from .types import parse_extra_hosts
  40. from .types import parse_restart_spec
  41. from .types import SecurityOpt
  42. from .types import ServiceLink
  43. from .types import ServicePort
  44. from .types import VolumeFromSpec
  45. from .types import VolumeSpec
  46. from .validation import match_named_volumes
  47. from .validation import validate_against_config_schema
  48. from .validation import validate_config_section
  49. from .validation import validate_cpu
  50. from .validation import validate_depends_on
  51. from .validation import validate_extends_file_path
  52. from .validation import validate_healthcheck
  53. from .validation import validate_links
  54. from .validation import validate_network_mode
  55. from .validation import validate_pid_mode
  56. from .validation import validate_service_constraints
  57. from .validation import validate_top_level_object
  58. from .validation import validate_ulimits
  59. DOCKER_CONFIG_KEYS = [
  60. 'cap_add',
  61. 'cap_drop',
  62. 'cgroup_parent',
  63. 'command',
  64. 'cpu_count',
  65. 'cpu_percent',
  66. 'cpu_period',
  67. 'cpu_quota',
  68. 'cpu_rt_period',
  69. 'cpu_rt_runtime',
  70. 'cpu_shares',
  71. 'cpus',
  72. 'cpuset',
  73. 'detach',
  74. 'device_cgroup_rules',
  75. 'devices',
  76. 'dns',
  77. 'dns_search',
  78. 'dns_opt',
  79. 'domainname',
  80. 'entrypoint',
  81. 'env_file',
  82. 'environment',
  83. 'extra_hosts',
  84. 'group_add',
  85. 'hostname',
  86. 'healthcheck',
  87. 'image',
  88. 'ipc',
  89. 'isolation',
  90. 'labels',
  91. 'links',
  92. 'mac_address',
  93. 'mem_limit',
  94. 'mem_reservation',
  95. 'memswap_limit',
  96. 'mem_swappiness',
  97. 'net',
  98. 'oom_score_adj',
  99. 'oom_kill_disable',
  100. 'pid',
  101. 'ports',
  102. 'privileged',
  103. 'read_only',
  104. 'restart',
  105. 'runtime',
  106. 'secrets',
  107. 'security_opt',
  108. 'shm_size',
  109. 'pids_limit',
  110. 'stdin_open',
  111. 'stop_signal',
  112. 'sysctls',
  113. 'tty',
  114. 'user',
  115. 'userns_mode',
  116. 'volume_driver',
  117. 'volumes',
  118. 'volumes_from',
  119. 'working_dir',
  120. ]
  121. ALLOWED_KEYS = DOCKER_CONFIG_KEYS + [
  122. 'blkio_config',
  123. 'build',
  124. 'container_name',
  125. 'credential_spec',
  126. 'dockerfile',
  127. 'init',
  128. 'log_driver',
  129. 'log_opt',
  130. 'logging',
  131. 'network_mode',
  132. 'platform',
  133. 'scale',
  134. 'stop_grace_period',
  135. ]
  136. DOCKER_VALID_URL_PREFIXES = (
  137. 'http://',
  138. 'https://',
  139. 'git://',
  140. 'github.com/',
  141. 'git@',
  142. )
  143. SUPPORTED_FILENAMES = [
  144. 'docker-compose.yml',
  145. 'docker-compose.yaml',
  146. ]
  147. DEFAULT_OVERRIDE_FILENAMES = ('docker-compose.override.yml', 'docker-compose.override.yaml')
  148. log = logging.getLogger(__name__)
  149. class ConfigDetails(namedtuple('_ConfigDetails', 'working_dir config_files environment')):
  150. """
  151. :param working_dir: the directory to use for relative paths in the config
  152. :type working_dir: string
  153. :param config_files: list of configuration files to load
  154. :type config_files: list of :class:`ConfigFile`
  155. :param environment: computed environment values for this project
  156. :type environment: :class:`environment.Environment`
  157. """
  158. def __new__(cls, working_dir, config_files, environment=None):
  159. if environment is None:
  160. environment = Environment.from_env_file(working_dir)
  161. return super(ConfigDetails, cls).__new__(
  162. cls, working_dir, config_files, environment
  163. )
  164. class ConfigFile(namedtuple('_ConfigFile', 'filename config')):
  165. """
  166. :param filename: filename of the config file
  167. :type filename: string
  168. :param config: contents of the config file
  169. :type config: :class:`dict`
  170. """
  171. @classmethod
  172. def from_filename(cls, filename):
  173. return cls(filename, load_yaml(filename))
  174. @cached_property
  175. def version(self):
  176. if 'version' not in self.config:
  177. return V1
  178. version = self.config['version']
  179. if isinstance(version, dict):
  180. log.warn('Unexpected type for "version" key in "{}". Assuming '
  181. '"version" is the name of a service, and defaulting to '
  182. 'Compose file version 1.'.format(self.filename))
  183. return V1
  184. if not isinstance(version, six.string_types):
  185. raise ConfigurationError(
  186. 'Version in "{}" is invalid - it should be a string.'
  187. .format(self.filename))
  188. if version == '1':
  189. raise ConfigurationError(
  190. 'Version in "{}" is invalid. {}'
  191. .format(self.filename, VERSION_EXPLANATION)
  192. )
  193. if version == '2':
  194. return const.COMPOSEFILE_V2_0
  195. if version == '3':
  196. return const.COMPOSEFILE_V3_0
  197. return ComposeVersion(version)
  198. def get_service(self, name):
  199. return self.get_service_dicts()[name]
  200. def get_service_dicts(self):
  201. return self.config if self.version == V1 else self.config.get('services', {})
  202. def get_volumes(self):
  203. return {} if self.version == V1 else self.config.get('volumes', {})
  204. def get_networks(self):
  205. return {} if self.version == V1 else self.config.get('networks', {})
  206. def get_secrets(self):
  207. return {} if self.version < const.COMPOSEFILE_V3_1 else self.config.get('secrets', {})
  208. def get_configs(self):
  209. return {} if self.version < const.COMPOSEFILE_V3_3 else self.config.get('configs', {})
  210. class Config(namedtuple('_Config', 'version services volumes networks secrets configs')):
  211. """
  212. :param version: configuration version
  213. :type version: int
  214. :param services: List of service description dictionaries
  215. :type services: :class:`list`
  216. :param volumes: Dictionary mapping volume names to description dictionaries
  217. :type volumes: :class:`dict`
  218. :param networks: Dictionary mapping network names to description dictionaries
  219. :type networks: :class:`dict`
  220. :param secrets: Dictionary mapping secret names to description dictionaries
  221. :type secrets: :class:`dict`
  222. :param configs: Dictionary mapping config names to description dictionaries
  223. :type configs: :class:`dict`
  224. """
  225. class ServiceConfig(namedtuple('_ServiceConfig', 'working_dir filename name config')):
  226. @classmethod
  227. def with_abs_paths(cls, working_dir, filename, name, config):
  228. if not working_dir:
  229. raise ValueError("No working_dir for ServiceConfig.")
  230. return cls(
  231. os.path.abspath(working_dir),
  232. os.path.abspath(filename) if filename else filename,
  233. name,
  234. config)
  235. def find(base_dir, filenames, environment, override_dir=None):
  236. if filenames == ['-']:
  237. return ConfigDetails(
  238. os.path.abspath(override_dir) if override_dir else os.getcwd(),
  239. [ConfigFile(None, yaml.safe_load(sys.stdin))],
  240. environment
  241. )
  242. if filenames:
  243. filenames = [os.path.join(base_dir, f) for f in filenames]
  244. else:
  245. filenames = get_default_config_files(base_dir)
  246. log.debug("Using configuration files: {}".format(",".join(filenames)))
  247. return ConfigDetails(
  248. override_dir if override_dir else os.path.dirname(filenames[0]),
  249. [ConfigFile.from_filename(f) for f in filenames],
  250. environment
  251. )
  252. def validate_config_version(config_files):
  253. main_file = config_files[0]
  254. validate_top_level_object(main_file)
  255. for next_file in config_files[1:]:
  256. validate_top_level_object(next_file)
  257. if main_file.version != next_file.version:
  258. raise ConfigurationError(
  259. "Version mismatch: file {0} specifies version {1} but "
  260. "extension file {2} uses version {3}".format(
  261. main_file.filename,
  262. main_file.version,
  263. next_file.filename,
  264. next_file.version))
  265. def get_default_config_files(base_dir):
  266. (candidates, path) = find_candidates_in_parent_dirs(SUPPORTED_FILENAMES, base_dir)
  267. if not candidates:
  268. raise ComposeFileNotFound(SUPPORTED_FILENAMES)
  269. winner = candidates[0]
  270. if len(candidates) > 1:
  271. log.warn("Found multiple config files with supported names: %s", ", ".join(candidates))
  272. log.warn("Using %s\n", winner)
  273. return [os.path.join(path, winner)] + get_default_override_file(path)
  274. def get_default_override_file(path):
  275. override_files_in_path = [os.path.join(path, override_filename) for override_filename
  276. in DEFAULT_OVERRIDE_FILENAMES
  277. if os.path.exists(os.path.join(path, override_filename))]
  278. if len(override_files_in_path) > 1:
  279. raise DuplicateOverrideFileFound(override_files_in_path)
  280. return override_files_in_path
  281. def find_candidates_in_parent_dirs(filenames, path):
  282. """
  283. Given a directory path to start, looks for filenames in the
  284. directory, and then each parent directory successively,
  285. until found.
  286. Returns tuple (candidates, path).
  287. """
  288. candidates = [filename for filename in filenames
  289. if os.path.exists(os.path.join(path, filename))]
  290. if not candidates:
  291. parent_dir = os.path.join(path, '..')
  292. if os.path.abspath(parent_dir) != os.path.abspath(path):
  293. return find_candidates_in_parent_dirs(filenames, parent_dir)
  294. return (candidates, path)
  295. def check_swarm_only_config(service_dicts, compatibility=False):
  296. warning_template = (
  297. "Some services ({services}) use the '{key}' key, which will be ignored. "
  298. "Compose does not support '{key}' configuration - use "
  299. "`docker stack deploy` to deploy to a swarm."
  300. )
  301. def check_swarm_only_key(service_dicts, key):
  302. services = [s for s in service_dicts if s.get(key)]
  303. if services:
  304. log.warn(
  305. warning_template.format(
  306. services=", ".join(sorted(s['name'] for s in services)),
  307. key=key
  308. )
  309. )
  310. if not compatibility:
  311. check_swarm_only_key(service_dicts, 'deploy')
  312. check_swarm_only_key(service_dicts, 'credential_spec')
  313. check_swarm_only_key(service_dicts, 'configs')
  314. def load(config_details, compatibility=False):
  315. """Load the configuration from a working directory and a list of
  316. configuration files. Files are loaded in order, and merged on top
  317. of each other to create the final configuration.
  318. Return a fully interpolated, extended and validated configuration.
  319. """
  320. validate_config_version(config_details.config_files)
  321. processed_files = [
  322. process_config_file(config_file, config_details.environment)
  323. for config_file in config_details.config_files
  324. ]
  325. config_details = config_details._replace(config_files=processed_files)
  326. main_file = config_details.config_files[0]
  327. volumes = load_mapping(
  328. config_details.config_files, 'get_volumes', 'Volume'
  329. )
  330. networks = load_mapping(
  331. config_details.config_files, 'get_networks', 'Network'
  332. )
  333. secrets = load_mapping(
  334. config_details.config_files, 'get_secrets', 'Secret', config_details.working_dir
  335. )
  336. configs = load_mapping(
  337. config_details.config_files, 'get_configs', 'Config', config_details.working_dir
  338. )
  339. service_dicts = load_services(config_details, main_file, compatibility)
  340. if main_file.version != V1:
  341. for service_dict in service_dicts:
  342. match_named_volumes(service_dict, volumes)
  343. check_swarm_only_config(service_dicts, compatibility)
  344. version = V2_3 if compatibility and main_file.version >= V3_0 else main_file.version
  345. return Config(version, service_dicts, volumes, networks, secrets, configs)
  346. def load_mapping(config_files, get_func, entity_type, working_dir=None):
  347. mapping = {}
  348. for config_file in config_files:
  349. for name, config in getattr(config_file, get_func)().items():
  350. mapping[name] = config or {}
  351. if not config:
  352. continue
  353. external = config.get('external')
  354. if external:
  355. validate_external(entity_type, name, config, config_file.version)
  356. if isinstance(external, dict):
  357. config['name'] = external.get('name')
  358. elif not config.get('name'):
  359. config['name'] = name
  360. if 'driver_opts' in config:
  361. config['driver_opts'] = build_string_dict(
  362. config['driver_opts']
  363. )
  364. if 'labels' in config:
  365. config['labels'] = parse_labels(config['labels'])
  366. if 'file' in config:
  367. config['file'] = expand_path(working_dir, config['file'])
  368. return mapping
  369. def validate_external(entity_type, name, config, version):
  370. if (version < V2_1 or (version >= V3_0 and version < V3_4)) and len(config.keys()) > 1:
  371. raise ConfigurationError(
  372. "{} {} declared as external but specifies additional attributes "
  373. "({}).".format(
  374. entity_type, name, ', '.join(k for k in config if k != 'external')))
  375. def load_services(config_details, config_file, compatibility=False):
  376. def build_service(service_name, service_dict, service_names):
  377. service_config = ServiceConfig.with_abs_paths(
  378. config_details.working_dir,
  379. config_file.filename,
  380. service_name,
  381. service_dict)
  382. resolver = ServiceExtendsResolver(
  383. service_config, config_file, environment=config_details.environment
  384. )
  385. service_dict = process_service(resolver.run())
  386. service_config = service_config._replace(config=service_dict)
  387. validate_service(service_config, service_names, config_file)
  388. service_dict = finalize_service(
  389. service_config,
  390. service_names,
  391. config_file.version,
  392. config_details.environment,
  393. compatibility
  394. )
  395. return service_dict
  396. def build_services(service_config):
  397. service_names = service_config.keys()
  398. return sort_service_dicts([
  399. build_service(name, service_dict, service_names)
  400. for name, service_dict in service_config.items()
  401. ])
  402. def merge_services(base, override):
  403. all_service_names = set(base) | set(override)
  404. return {
  405. name: merge_service_dicts_from_files(
  406. base.get(name, {}),
  407. override.get(name, {}),
  408. config_file.version)
  409. for name in all_service_names
  410. }
  411. service_configs = [
  412. file.get_service_dicts() for file in config_details.config_files
  413. ]
  414. service_config = service_configs[0]
  415. for next_config in service_configs[1:]:
  416. service_config = merge_services(service_config, next_config)
  417. return build_services(service_config)
  418. def interpolate_config_section(config_file, config, section, environment):
  419. validate_config_section(config_file.filename, config, section)
  420. return interpolate_environment_variables(
  421. config_file.version,
  422. config,
  423. section,
  424. environment
  425. )
  426. def process_config_file(config_file, environment, service_name=None):
  427. services = interpolate_config_section(
  428. config_file,
  429. config_file.get_service_dicts(),
  430. 'service',
  431. environment)
  432. if config_file.version > V1:
  433. processed_config = dict(config_file.config)
  434. processed_config['services'] = services
  435. processed_config['volumes'] = interpolate_config_section(
  436. config_file,
  437. config_file.get_volumes(),
  438. 'volume',
  439. environment)
  440. processed_config['networks'] = interpolate_config_section(
  441. config_file,
  442. config_file.get_networks(),
  443. 'network',
  444. environment)
  445. if config_file.version >= const.COMPOSEFILE_V3_1:
  446. processed_config['secrets'] = interpolate_config_section(
  447. config_file,
  448. config_file.get_secrets(),
  449. 'secret',
  450. environment)
  451. if config_file.version >= const.COMPOSEFILE_V3_3:
  452. processed_config['configs'] = interpolate_config_section(
  453. config_file,
  454. config_file.get_configs(),
  455. 'config',
  456. environment
  457. )
  458. else:
  459. processed_config = services
  460. config_file = config_file._replace(config=processed_config)
  461. validate_against_config_schema(config_file)
  462. if service_name and service_name not in services:
  463. raise ConfigurationError(
  464. "Cannot extend service '{}' in {}: Service not found".format(
  465. service_name, config_file.filename))
  466. return config_file
  467. class ServiceExtendsResolver(object):
  468. def __init__(self, service_config, config_file, environment, already_seen=None):
  469. self.service_config = service_config
  470. self.working_dir = service_config.working_dir
  471. self.already_seen = already_seen or []
  472. self.config_file = config_file
  473. self.environment = environment
  474. @property
  475. def signature(self):
  476. return self.service_config.filename, self.service_config.name
  477. def detect_cycle(self):
  478. if self.signature in self.already_seen:
  479. raise CircularReference(self.already_seen + [self.signature])
  480. def run(self):
  481. self.detect_cycle()
  482. if 'extends' in self.service_config.config:
  483. service_dict = self.resolve_extends(*self.validate_and_construct_extends())
  484. return self.service_config._replace(config=service_dict)
  485. return self.service_config
  486. def validate_and_construct_extends(self):
  487. extends = self.service_config.config['extends']
  488. if not isinstance(extends, dict):
  489. extends = {'service': extends}
  490. config_path = self.get_extended_config_path(extends)
  491. service_name = extends['service']
  492. if config_path == self.config_file.filename:
  493. try:
  494. service_config = self.config_file.get_service(service_name)
  495. except KeyError:
  496. raise ConfigurationError(
  497. "Cannot extend service '{}' in {}: Service not found".format(
  498. service_name, config_path)
  499. )
  500. else:
  501. extends_file = ConfigFile.from_filename(config_path)
  502. validate_config_version([self.config_file, extends_file])
  503. extended_file = process_config_file(
  504. extends_file, self.environment, service_name=service_name
  505. )
  506. service_config = extended_file.get_service(service_name)
  507. return config_path, service_config, service_name
  508. def resolve_extends(self, extended_config_path, service_dict, service_name):
  509. resolver = ServiceExtendsResolver(
  510. ServiceConfig.with_abs_paths(
  511. os.path.dirname(extended_config_path),
  512. extended_config_path,
  513. service_name,
  514. service_dict),
  515. self.config_file,
  516. already_seen=self.already_seen + [self.signature],
  517. environment=self.environment
  518. )
  519. service_config = resolver.run()
  520. other_service_dict = process_service(service_config)
  521. validate_extended_service_dict(
  522. other_service_dict,
  523. extended_config_path,
  524. service_name)
  525. return merge_service_dicts(
  526. other_service_dict,
  527. self.service_config.config,
  528. self.config_file.version)
  529. def get_extended_config_path(self, extends_options):
  530. """Service we are extending either has a value for 'file' set, which we
  531. need to obtain a full path too or we are extending from a service
  532. defined in our own file.
  533. """
  534. filename = self.service_config.filename
  535. validate_extends_file_path(
  536. self.service_config.name,
  537. extends_options,
  538. filename)
  539. if 'file' in extends_options:
  540. return expand_path(self.working_dir, extends_options['file'])
  541. return filename
  542. def resolve_environment(service_dict, environment=None):
  543. """Unpack any environment variables from an env_file, if set.
  544. Interpolate environment values if set.
  545. """
  546. env = {}
  547. for env_file in service_dict.get('env_file', []):
  548. env.update(env_vars_from_file(env_file))
  549. env.update(parse_environment(service_dict.get('environment')))
  550. return dict(resolve_env_var(k, v, environment) for k, v in six.iteritems(env))
  551. def resolve_build_args(buildargs, environment):
  552. args = parse_build_arguments(buildargs)
  553. return dict(resolve_env_var(k, v, environment) for k, v in six.iteritems(args))
  554. def validate_extended_service_dict(service_dict, filename, service):
  555. error_prefix = "Cannot extend service '%s' in %s:" % (service, filename)
  556. if 'links' in service_dict:
  557. raise ConfigurationError(
  558. "%s services with 'links' cannot be extended" % error_prefix)
  559. if 'volumes_from' in service_dict:
  560. raise ConfigurationError(
  561. "%s services with 'volumes_from' cannot be extended" % error_prefix)
  562. if 'net' in service_dict:
  563. if get_container_name_from_network_mode(service_dict['net']):
  564. raise ConfigurationError(
  565. "%s services with 'net: container' cannot be extended" % error_prefix)
  566. if 'network_mode' in service_dict:
  567. if get_service_name_from_network_mode(service_dict['network_mode']):
  568. raise ConfigurationError(
  569. "%s services with 'network_mode: service' cannot be extended" % error_prefix)
  570. if 'depends_on' in service_dict:
  571. raise ConfigurationError(
  572. "%s services with 'depends_on' cannot be extended" % error_prefix)
  573. def validate_service(service_config, service_names, config_file):
  574. service_dict, service_name = service_config.config, service_config.name
  575. validate_service_constraints(service_dict, service_name, config_file)
  576. validate_paths(service_dict)
  577. validate_cpu(service_config)
  578. validate_ulimits(service_config)
  579. validate_network_mode(service_config, service_names)
  580. validate_pid_mode(service_config, service_names)
  581. validate_depends_on(service_config, service_names)
  582. validate_links(service_config, service_names)
  583. validate_healthcheck(service_config)
  584. if not service_dict.get('image') and has_uppercase(service_name):
  585. raise ConfigurationError(
  586. "Service '{name}' contains uppercase characters which are not valid "
  587. "as part of an image name. Either use a lowercase service name or "
  588. "use the `image` field to set a custom name for the service image."
  589. .format(name=service_name))
  590. def process_service(service_config):
  591. working_dir = service_config.working_dir
  592. service_dict = dict(service_config.config)
  593. if 'env_file' in service_dict:
  594. service_dict['env_file'] = [
  595. expand_path(working_dir, path)
  596. for path in to_list(service_dict['env_file'])
  597. ]
  598. if 'build' in service_dict:
  599. process_build_section(service_dict, working_dir)
  600. if 'volumes' in service_dict and service_dict.get('volume_driver') is None:
  601. service_dict['volumes'] = resolve_volume_paths(working_dir, service_dict)
  602. if 'sysctls' in service_dict:
  603. service_dict['sysctls'] = build_string_dict(parse_sysctls(service_dict['sysctls']))
  604. if 'labels' in service_dict:
  605. service_dict['labels'] = parse_labels(service_dict['labels'])
  606. service_dict = process_depends_on(service_dict)
  607. for field in ['dns', 'dns_search', 'tmpfs']:
  608. if field in service_dict:
  609. service_dict[field] = to_list(service_dict[field])
  610. service_dict = process_security_opt(process_blkio_config(process_ports(
  611. process_healthcheck(service_dict)
  612. )))
  613. return service_dict
  614. def process_build_section(service_dict, working_dir):
  615. if isinstance(service_dict['build'], six.string_types):
  616. service_dict['build'] = resolve_build_path(working_dir, service_dict['build'])
  617. elif isinstance(service_dict['build'], dict):
  618. if 'context' in service_dict['build']:
  619. path = service_dict['build']['context']
  620. service_dict['build']['context'] = resolve_build_path(working_dir, path)
  621. if 'labels' in service_dict['build']:
  622. service_dict['build']['labels'] = parse_labels(service_dict['build']['labels'])
  623. def process_ports(service_dict):
  624. if 'ports' not in service_dict:
  625. return service_dict
  626. ports = []
  627. for port_definition in service_dict['ports']:
  628. if isinstance(port_definition, ServicePort):
  629. ports.append(port_definition)
  630. else:
  631. ports.extend(ServicePort.parse(port_definition))
  632. service_dict['ports'] = ports
  633. return service_dict
  634. def process_depends_on(service_dict):
  635. if 'depends_on' in service_dict and not isinstance(service_dict['depends_on'], dict):
  636. service_dict['depends_on'] = dict([
  637. (svc, {'condition': 'service_started'}) for svc in service_dict['depends_on']
  638. ])
  639. return service_dict
  640. def process_blkio_config(service_dict):
  641. if not service_dict.get('blkio_config'):
  642. return service_dict
  643. for field in ['device_read_bps', 'device_write_bps']:
  644. if field in service_dict['blkio_config']:
  645. for v in service_dict['blkio_config'].get(field, []):
  646. rate = v.get('rate', 0)
  647. v['rate'] = parse_bytes(rate)
  648. if v['rate'] is None:
  649. raise ConfigurationError('Invalid format for bytes value: "{}"'.format(rate))
  650. for field in ['device_read_iops', 'device_write_iops']:
  651. if field in service_dict['blkio_config']:
  652. for v in service_dict['blkio_config'].get(field, []):
  653. try:
  654. v['rate'] = int(v.get('rate', 0))
  655. except ValueError:
  656. raise ConfigurationError(
  657. 'Invalid IOPS value: "{}". Must be a positive integer.'.format(v.get('rate'))
  658. )
  659. return service_dict
  660. def process_healthcheck(service_dict):
  661. if 'healthcheck' not in service_dict:
  662. return service_dict
  663. hc = service_dict['healthcheck']
  664. if 'disable' in hc:
  665. del hc['disable']
  666. hc['test'] = ['NONE']
  667. for field in ['interval', 'timeout', 'start_period']:
  668. if field not in hc or isinstance(hc[field], six.integer_types):
  669. continue
  670. hc[field] = parse_nanoseconds_int(hc[field])
  671. return service_dict
  672. def finalize_service_volumes(service_dict, environment):
  673. if 'volumes' in service_dict:
  674. finalized_volumes = []
  675. normalize = environment.get_boolean('COMPOSE_CONVERT_WINDOWS_PATHS')
  676. win_host = environment.get_boolean('COMPOSE_FORCE_WINDOWS_HOST')
  677. for v in service_dict['volumes']:
  678. if isinstance(v, dict):
  679. finalized_volumes.append(MountSpec.parse(v, normalize, win_host))
  680. else:
  681. finalized_volumes.append(VolumeSpec.parse(v, normalize, win_host))
  682. service_dict['volumes'] = finalized_volumes
  683. return service_dict
  684. def finalize_service(service_config, service_names, version, environment, compatibility):
  685. service_dict = dict(service_config.config)
  686. if 'environment' in service_dict or 'env_file' in service_dict:
  687. service_dict['environment'] = resolve_environment(service_dict, environment)
  688. service_dict.pop('env_file', None)
  689. if 'volumes_from' in service_dict:
  690. service_dict['volumes_from'] = [
  691. VolumeFromSpec.parse(vf, service_names, version)
  692. for vf in service_dict['volumes_from']
  693. ]
  694. service_dict = finalize_service_volumes(service_dict, environment)
  695. if 'net' in service_dict:
  696. network_mode = service_dict.pop('net')
  697. container_name = get_container_name_from_network_mode(network_mode)
  698. if container_name and container_name in service_names:
  699. service_dict['network_mode'] = 'service:{}'.format(container_name)
  700. else:
  701. service_dict['network_mode'] = network_mode
  702. if 'networks' in service_dict:
  703. service_dict['networks'] = parse_networks(service_dict['networks'])
  704. if 'restart' in service_dict:
  705. service_dict['restart'] = parse_restart_spec(service_dict['restart'])
  706. if 'secrets' in service_dict:
  707. service_dict['secrets'] = [
  708. types.ServiceSecret.parse(s) for s in service_dict['secrets']
  709. ]
  710. if 'configs' in service_dict:
  711. service_dict['configs'] = [
  712. types.ServiceConfig.parse(c) for c in service_dict['configs']
  713. ]
  714. normalize_build(service_dict, service_config.working_dir, environment)
  715. if compatibility:
  716. service_dict, ignored_keys = translate_deploy_keys_to_container_config(
  717. service_dict
  718. )
  719. if ignored_keys:
  720. log.warn(
  721. 'The following deploy sub-keys are not supported in compatibility mode and have'
  722. ' been ignored: {}'.format(', '.join(ignored_keys))
  723. )
  724. service_dict['name'] = service_config.name
  725. return normalize_v1_service_format(service_dict)
  726. def translate_resource_keys_to_container_config(resources_dict, service_dict):
  727. if 'limits' in resources_dict:
  728. service_dict['mem_limit'] = resources_dict['limits'].get('memory')
  729. if 'cpus' in resources_dict['limits']:
  730. service_dict['cpus'] = float(resources_dict['limits']['cpus'])
  731. if 'reservations' in resources_dict:
  732. service_dict['mem_reservation'] = resources_dict['reservations'].get('memory')
  733. if 'cpus' in resources_dict['reservations']:
  734. return ['resources.reservations.cpus']
  735. return []
  736. def convert_restart_policy(name):
  737. try:
  738. return {
  739. 'any': 'always',
  740. 'none': 'no',
  741. 'on-failure': 'on-failure'
  742. }[name]
  743. except KeyError:
  744. raise ConfigurationError('Invalid restart policy "{}"'.format(name))
  745. def translate_deploy_keys_to_container_config(service_dict):
  746. if 'credential_spec' in service_dict:
  747. del service_dict['credential_spec']
  748. if 'configs' in service_dict:
  749. del service_dict['configs']
  750. if 'deploy' not in service_dict:
  751. return service_dict, []
  752. deploy_dict = service_dict['deploy']
  753. ignored_keys = [
  754. k for k in ['endpoint_mode', 'labels', 'update_config', 'rollback_config', 'placement']
  755. if k in deploy_dict
  756. ]
  757. if 'replicas' in deploy_dict and deploy_dict.get('mode', 'replicated') == 'replicated':
  758. service_dict['scale'] = deploy_dict['replicas']
  759. if 'restart_policy' in deploy_dict:
  760. service_dict['restart'] = {
  761. 'Name': convert_restart_policy(deploy_dict['restart_policy'].get('condition', 'any')),
  762. 'MaximumRetryCount': deploy_dict['restart_policy'].get('max_attempts', 0)
  763. }
  764. for k in deploy_dict['restart_policy'].keys():
  765. if k != 'condition' and k != 'max_attempts':
  766. ignored_keys.append('restart_policy.{}'.format(k))
  767. ignored_keys.extend(
  768. translate_resource_keys_to_container_config(
  769. deploy_dict.get('resources', {}), service_dict
  770. )
  771. )
  772. del service_dict['deploy']
  773. return service_dict, ignored_keys
  774. def normalize_v1_service_format(service_dict):
  775. if 'log_driver' in service_dict or 'log_opt' in service_dict:
  776. if 'logging' not in service_dict:
  777. service_dict['logging'] = {}
  778. if 'log_driver' in service_dict:
  779. service_dict['logging']['driver'] = service_dict['log_driver']
  780. del service_dict['log_driver']
  781. if 'log_opt' in service_dict:
  782. service_dict['logging']['options'] = service_dict['log_opt']
  783. del service_dict['log_opt']
  784. if 'dockerfile' in service_dict:
  785. service_dict['build'] = service_dict.get('build', {})
  786. service_dict['build'].update({
  787. 'dockerfile': service_dict.pop('dockerfile')
  788. })
  789. return service_dict
  790. def merge_service_dicts_from_files(base, override, version):
  791. """When merging services from multiple files we need to merge the `extends`
  792. field. This is not handled by `merge_service_dicts()` which is used to
  793. perform the `extends`.
  794. """
  795. new_service = merge_service_dicts(base, override, version)
  796. if 'extends' in override:
  797. new_service['extends'] = override['extends']
  798. elif 'extends' in base:
  799. new_service['extends'] = base['extends']
  800. return new_service
  801. class MergeDict(dict):
  802. """A dict-like object responsible for merging two dicts into one."""
  803. def __init__(self, base, override):
  804. self.base = base
  805. self.override = override
  806. def needs_merge(self, field):
  807. return field in self.base or field in self.override
  808. def merge_field(self, field, merge_func, default=None):
  809. if not self.needs_merge(field):
  810. return
  811. self[field] = merge_func(
  812. self.base.get(field, default),
  813. self.override.get(field, default))
  814. def merge_mapping(self, field, parse_func=None):
  815. if not self.needs_merge(field):
  816. return
  817. if parse_func is None:
  818. def parse_func(m):
  819. return m or {}
  820. self[field] = parse_func(self.base.get(field))
  821. self[field].update(parse_func(self.override.get(field)))
  822. def merge_sequence(self, field, parse_func):
  823. def parse_sequence_func(seq):
  824. return to_mapping((parse_func(item) for item in seq), 'merge_field')
  825. if not self.needs_merge(field):
  826. return
  827. merged = parse_sequence_func(self.base.get(field, []))
  828. merged.update(parse_sequence_func(self.override.get(field, [])))
  829. self[field] = [item.repr() for item in sorted(merged.values())]
  830. def merge_scalar(self, field):
  831. if self.needs_merge(field):
  832. self[field] = self.override.get(field, self.base.get(field))
  833. def merge_service_dicts(base, override, version):
  834. md = MergeDict(base, override)
  835. md.merge_mapping('environment', parse_environment)
  836. md.merge_mapping('labels', parse_labels)
  837. md.merge_mapping('ulimits', parse_flat_dict)
  838. md.merge_mapping('sysctls', parse_sysctls)
  839. md.merge_mapping('depends_on', parse_depends_on)
  840. md.merge_mapping('storage_opt', parse_flat_dict)
  841. md.merge_sequence('links', ServiceLink.parse)
  842. md.merge_sequence('secrets', types.ServiceSecret.parse)
  843. md.merge_sequence('configs', types.ServiceConfig.parse)
  844. md.merge_sequence('security_opt', types.SecurityOpt.parse)
  845. md.merge_mapping('extra_hosts', parse_extra_hosts)
  846. md.merge_field('networks', merge_networks, default={})
  847. for field in ['volumes', 'devices']:
  848. md.merge_field(field, merge_path_mappings)
  849. for field in [
  850. 'cap_add', 'cap_drop', 'expose', 'external_links',
  851. 'volumes_from', 'device_cgroup_rules',
  852. ]:
  853. md.merge_field(field, merge_unique_items_lists, default=[])
  854. for field in ['dns', 'dns_search', 'env_file', 'tmpfs']:
  855. md.merge_field(field, merge_list_or_string)
  856. md.merge_field('logging', merge_logging, default={})
  857. merge_ports(md, base, override)
  858. md.merge_field('blkio_config', merge_blkio_config, default={})
  859. md.merge_field('healthcheck', merge_healthchecks, default={})
  860. md.merge_field('deploy', merge_deploy, default={})
  861. for field in set(ALLOWED_KEYS) - set(md):
  862. md.merge_scalar(field)
  863. if version == V1:
  864. legacy_v1_merge_image_or_build(md, base, override)
  865. elif md.needs_merge('build'):
  866. md['build'] = merge_build(md, base, override)
  867. return dict(md)
  868. def merge_unique_items_lists(base, override):
  869. override = [str(o) for o in override]
  870. base = [str(b) for b in base]
  871. return sorted(set().union(base, override))
  872. def merge_healthchecks(base, override):
  873. if override.get('disabled') is True:
  874. return override
  875. result = base.copy()
  876. result.update(override)
  877. return result
  878. def merge_ports(md, base, override):
  879. def parse_sequence_func(seq):
  880. acc = []
  881. for item in seq:
  882. acc.extend(ServicePort.parse(item))
  883. return to_mapping(acc, 'merge_field')
  884. field = 'ports'
  885. if not md.needs_merge(field):
  886. return
  887. merged = parse_sequence_func(md.base.get(field, []))
  888. merged.update(parse_sequence_func(md.override.get(field, [])))
  889. md[field] = [item for item in sorted(merged.values(), key=lambda x: x.target)]
  890. def merge_build(output, base, override):
  891. def to_dict(service):
  892. build_config = service.get('build', {})
  893. if isinstance(build_config, six.string_types):
  894. return {'context': build_config}
  895. return build_config
  896. md = MergeDict(to_dict(base), to_dict(override))
  897. md.merge_scalar('context')
  898. md.merge_scalar('dockerfile')
  899. md.merge_scalar('network')
  900. md.merge_scalar('target')
  901. md.merge_scalar('shm_size')
  902. md.merge_scalar('isolation')
  903. md.merge_mapping('args', parse_build_arguments)
  904. md.merge_field('cache_from', merge_unique_items_lists, default=[])
  905. md.merge_mapping('labels', parse_labels)
  906. md.merge_mapping('extra_hosts', parse_extra_hosts)
  907. return dict(md)
  908. def merge_deploy(base, override):
  909. md = MergeDict(base or {}, override or {})
  910. md.merge_scalar('mode')
  911. md.merge_scalar('endpoint_mode')
  912. md.merge_scalar('replicas')
  913. md.merge_mapping('labels', parse_labels)
  914. md.merge_mapping('update_config')
  915. md.merge_mapping('rollback_config')
  916. md.merge_mapping('restart_policy')
  917. if md.needs_merge('resources'):
  918. resources_md = MergeDict(md.base.get('resources') or {}, md.override.get('resources') or {})
  919. resources_md.merge_mapping('limits')
  920. resources_md.merge_field('reservations', merge_reservations, default={})
  921. md['resources'] = dict(resources_md)
  922. if md.needs_merge('placement'):
  923. placement_md = MergeDict(md.base.get('placement') or {}, md.override.get('placement') or {})
  924. placement_md.merge_field('constraints', merge_unique_items_lists, default=[])
  925. placement_md.merge_field('preferences', merge_unique_objects_lists, default=[])
  926. md['placement'] = dict(placement_md)
  927. return dict(md)
  928. def merge_networks(base, override):
  929. merged_networks = {}
  930. all_network_names = set(base) | set(override)
  931. base = {k: {} for k in base} if isinstance(base, list) else base
  932. override = {k: {} for k in override} if isinstance(override, list) else override
  933. for network_name in all_network_names:
  934. md = MergeDict(base.get(network_name, {}), override.get(network_name, {}))
  935. md.merge_field('aliases', merge_unique_items_lists, [])
  936. md.merge_field('link_local_ips', merge_unique_items_lists, [])
  937. md.merge_scalar('priority')
  938. md.merge_scalar('ipv4_address')
  939. md.merge_scalar('ipv6_address')
  940. merged_networks[network_name] = dict(md)
  941. return merged_networks
  942. def merge_reservations(base, override):
  943. md = MergeDict(base, override)
  944. md.merge_scalar('cpus')
  945. md.merge_scalar('memory')
  946. md.merge_sequence('generic_resources', types.GenericResource.parse)
  947. return dict(md)
  948. def merge_unique_objects_lists(base, override):
  949. result = dict((json_hash(i), i) for i in base + override)
  950. return [i[1] for i in sorted([(k, v) for k, v in result.items()], key=lambda x: x[0])]
  951. def merge_blkio_config(base, override):
  952. md = MergeDict(base, override)
  953. md.merge_scalar('weight')
  954. def merge_blkio_limits(base, override):
  955. index = dict((b['path'], b) for b in base)
  956. for o in override:
  957. index[o['path']] = o
  958. return sorted(list(index.values()), key=lambda x: x['path'])
  959. for field in [
  960. "device_read_bps", "device_read_iops", "device_write_bps",
  961. "device_write_iops", "weight_device",
  962. ]:
  963. md.merge_field(field, merge_blkio_limits, default=[])
  964. return dict(md)
  965. def merge_logging(base, override):
  966. md = MergeDict(base, override)
  967. md.merge_scalar('driver')
  968. if md.get('driver') == base.get('driver') or base.get('driver') is None:
  969. md.merge_mapping('options', lambda m: m or {})
  970. elif override.get('options'):
  971. md['options'] = override.get('options', {})
  972. return dict(md)
  973. def legacy_v1_merge_image_or_build(output, base, override):
  974. output.pop('image', None)
  975. output.pop('build', None)
  976. if 'image' in override:
  977. output['image'] = override['image']
  978. elif 'build' in override:
  979. output['build'] = override['build']
  980. elif 'image' in base:
  981. output['image'] = base['image']
  982. elif 'build' in base:
  983. output['build'] = base['build']
  984. def merge_environment(base, override):
  985. env = parse_environment(base)
  986. env.update(parse_environment(override))
  987. return env
  988. def merge_labels(base, override):
  989. labels = parse_labels(base)
  990. labels.update(parse_labels(override))
  991. return labels
  992. def split_kv(kvpair):
  993. if '=' in kvpair:
  994. return kvpair.split('=', 1)
  995. else:
  996. return kvpair, ''
  997. def parse_dict_or_list(split_func, type_name, arguments):
  998. if not arguments:
  999. return {}
  1000. if isinstance(arguments, list):
  1001. return dict(split_func(e) for e in arguments)
  1002. if isinstance(arguments, dict):
  1003. return dict(arguments)
  1004. raise ConfigurationError(
  1005. "%s \"%s\" must be a list or mapping," %
  1006. (type_name, arguments)
  1007. )
  1008. parse_build_arguments = functools.partial(parse_dict_or_list, split_env, 'build arguments')
  1009. parse_environment = functools.partial(parse_dict_or_list, split_env, 'environment')
  1010. parse_labels = functools.partial(parse_dict_or_list, split_kv, 'labels')
  1011. parse_networks = functools.partial(parse_dict_or_list, lambda k: (k, None), 'networks')
  1012. parse_sysctls = functools.partial(parse_dict_or_list, split_kv, 'sysctls')
  1013. parse_depends_on = functools.partial(
  1014. parse_dict_or_list, lambda k: (k, {'condition': 'service_started'}), 'depends_on'
  1015. )
  1016. def parse_flat_dict(d):
  1017. if not d:
  1018. return {}
  1019. if isinstance(d, dict):
  1020. return dict(d)
  1021. raise ConfigurationError("Invalid type: expected mapping")
  1022. def resolve_env_var(key, val, environment):
  1023. if val is not None:
  1024. return key, val
  1025. elif environment and key in environment:
  1026. return key, environment[key]
  1027. else:
  1028. return key, None
  1029. def resolve_volume_paths(working_dir, service_dict):
  1030. return [
  1031. resolve_volume_path(working_dir, volume)
  1032. for volume in service_dict['volumes']
  1033. ]
  1034. def resolve_volume_path(working_dir, volume):
  1035. if isinstance(volume, dict):
  1036. if volume.get('source', '').startswith(('.', '~')) and volume['type'] == 'bind':
  1037. volume['source'] = expand_path(working_dir, volume['source'])
  1038. return volume
  1039. mount_params = None
  1040. container_path, mount_params = split_path_mapping(volume)
  1041. if mount_params is not None:
  1042. host_path, mode = mount_params
  1043. if host_path is None:
  1044. return container_path
  1045. if host_path.startswith('.'):
  1046. host_path = expand_path(working_dir, host_path)
  1047. host_path = os.path.expanduser(host_path)
  1048. return u"{}:{}{}".format(host_path, container_path, (':' + mode if mode else ''))
  1049. return container_path
  1050. def normalize_build(service_dict, working_dir, environment):
  1051. if 'build' in service_dict:
  1052. build = {}
  1053. # Shortcut where specifying a string is treated as the build context
  1054. if isinstance(service_dict['build'], six.string_types):
  1055. build['context'] = service_dict.pop('build')
  1056. else:
  1057. build.update(service_dict['build'])
  1058. if 'args' in build:
  1059. build['args'] = build_string_dict(
  1060. resolve_build_args(build.get('args'), environment)
  1061. )
  1062. service_dict['build'] = build
  1063. def resolve_build_path(working_dir, build_path):
  1064. if is_url(build_path):
  1065. return build_path
  1066. return expand_path(working_dir, build_path)
  1067. def is_url(build_path):
  1068. return build_path.startswith(DOCKER_VALID_URL_PREFIXES)
  1069. def validate_paths(service_dict):
  1070. if 'build' in service_dict:
  1071. build = service_dict.get('build', {})
  1072. if isinstance(build, six.string_types):
  1073. build_path = build
  1074. elif isinstance(build, dict) and 'context' in build:
  1075. build_path = build['context']
  1076. else:
  1077. # We have a build section but no context, so nothing to validate
  1078. return
  1079. if (
  1080. not is_url(build_path) and
  1081. (not os.path.exists(build_path) or not os.access(build_path, os.R_OK))
  1082. ):
  1083. raise ConfigurationError(
  1084. "build path %s either does not exist, is not accessible, "
  1085. "or is not a valid URL." % build_path)
  1086. def merge_path_mappings(base, override):
  1087. d = dict_from_path_mappings(base)
  1088. d.update(dict_from_path_mappings(override))
  1089. return path_mappings_from_dict(d)
  1090. def dict_from_path_mappings(path_mappings):
  1091. if path_mappings:
  1092. return dict(split_path_mapping(v) for v in path_mappings)
  1093. else:
  1094. return {}
  1095. def path_mappings_from_dict(d):
  1096. return [join_path_mapping(v) for v in sorted(d.items())]
  1097. def split_path_mapping(volume_path):
  1098. """
  1099. Ascertain if the volume_path contains a host path as well as a container
  1100. path. Using splitdrive so windows absolute paths won't cause issues with
  1101. splitting on ':'.
  1102. """
  1103. if isinstance(volume_path, dict):
  1104. return (volume_path.get('target'), volume_path)
  1105. drive, volume_config = splitdrive(volume_path)
  1106. if ':' in volume_config:
  1107. (host, container) = volume_config.split(':', 1)
  1108. container_drive, container_path = splitdrive(container)
  1109. mode = None
  1110. if ':' in container_path:
  1111. container_path, mode = container_path.rsplit(':', 1)
  1112. return (container_drive + container_path, (drive + host, mode))
  1113. else:
  1114. return (volume_path, None)
  1115. def process_security_opt(service_dict):
  1116. security_opts = service_dict.get('security_opt', [])
  1117. result = []
  1118. for value in security_opts:
  1119. result.append(SecurityOpt.parse(value))
  1120. if result:
  1121. service_dict['security_opt'] = result
  1122. return service_dict
  1123. def join_path_mapping(pair):
  1124. (container, host) = pair
  1125. if isinstance(host, dict):
  1126. return host
  1127. elif host is None:
  1128. return container
  1129. else:
  1130. host, mode = host
  1131. result = ":".join((host, container))
  1132. if mode:
  1133. result += ":" + mode
  1134. return result
  1135. def expand_path(working_dir, path):
  1136. return os.path.abspath(os.path.join(working_dir, os.path.expanduser(path)))
  1137. def merge_list_or_string(base, override):
  1138. return to_list(base) + to_list(override)
  1139. def to_list(value):
  1140. if value is None:
  1141. return []
  1142. elif isinstance(value, six.string_types):
  1143. return [value]
  1144. else:
  1145. return value
  1146. def to_mapping(sequence, key_field):
  1147. return {getattr(item, key_field): item for item in sequence}
  1148. def has_uppercase(name):
  1149. return any(char in string.ascii_uppercase for char in name)
  1150. def load_yaml(filename, encoding=None, binary=True):
  1151. try:
  1152. with io.open(filename, 'rb' if binary else 'r', encoding=encoding) as fh:
  1153. return yaml.safe_load(fh)
  1154. except (IOError, yaml.YAMLError, UnicodeDecodeError) as e:
  1155. if encoding is None:
  1156. # Sometimes the user's locale sets an encoding that doesn't match
  1157. # the YAML files. Im such cases, retry once with the "default"
  1158. # UTF-8 encoding
  1159. return load_yaml(filename, encoding='utf-8-sig', binary=False)
  1160. error_name = getattr(e, '__module__', '') + '.' + e.__class__.__name__
  1161. raise ConfigurationError(u"{}: {}".format(error_name, e))