config.py 37 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162
  1. from __future__ import absolute_import
  2. from __future__ import unicode_literals
  3. import functools
  4. import logging
  5. import os
  6. import string
  7. import sys
  8. from collections import namedtuple
  9. import six
  10. import yaml
  11. from cached_property import cached_property
  12. from . import types
  13. from ..const import COMPOSEFILE_V1 as V1
  14. from ..const import COMPOSEFILE_V2_0 as V2_0
  15. from ..const import COMPOSEFILE_V2_1 as V2_1
  16. from ..const import COMPOSEFILE_V3_0 as V3_0
  17. from ..const import COMPOSEFILE_V3_1 as V3_1
  18. from ..utils import build_string_dict
  19. from ..utils import parse_nanoseconds_int
  20. from ..utils import splitdrive
  21. from .environment import env_vars_from_file
  22. from .environment import Environment
  23. from .environment import split_env
  24. from .errors import CircularReference
  25. from .errors import ComposeFileNotFound
  26. from .errors import ConfigurationError
  27. from .errors import VERSION_EXPLANATION
  28. from .interpolation import interpolate_environment_variables
  29. from .sort_services import get_container_name_from_network_mode
  30. from .sort_services import get_service_name_from_network_mode
  31. from .sort_services import sort_service_dicts
  32. from .types import parse_extra_hosts
  33. from .types import parse_restart_spec
  34. from .types import ServiceLink
  35. from .types import ServicePort
  36. from .types import VolumeFromSpec
  37. from .types import VolumeSpec
  38. from .validation import match_named_volumes
  39. from .validation import validate_against_config_schema
  40. from .validation import validate_config_section
  41. from .validation import validate_depends_on
  42. from .validation import validate_extends_file_path
  43. from .validation import validate_links
  44. from .validation import validate_network_mode
  45. from .validation import validate_service_constraints
  46. from .validation import validate_top_level_object
  47. from .validation import validate_ulimits
  48. DOCKER_CONFIG_KEYS = [
  49. 'cap_add',
  50. 'cap_drop',
  51. 'cgroup_parent',
  52. 'command',
  53. 'cpu_quota',
  54. 'cpu_shares',
  55. 'cpuset',
  56. 'detach',
  57. 'devices',
  58. 'dns',
  59. 'dns_search',
  60. 'dns_opt',
  61. 'domainname',
  62. 'entrypoint',
  63. 'env_file',
  64. 'environment',
  65. 'extra_hosts',
  66. 'group_add',
  67. 'hostname',
  68. 'healthcheck',
  69. 'image',
  70. 'ipc',
  71. 'labels',
  72. 'links',
  73. 'mac_address',
  74. 'mem_limit'
  75. 'mem_reservation',
  76. 'memswap_limit',
  77. 'mem_swappiness',
  78. 'net',
  79. 'oom_score_adj',
  80. 'pid',
  81. 'ports',
  82. 'privileged',
  83. 'read_only',
  84. 'restart',
  85. 'secrets',
  86. 'security_opt',
  87. 'shm_size',
  88. 'pids_limit',
  89. 'stdin_open',
  90. 'stop_signal',
  91. 'sysctls',
  92. 'tty',
  93. 'user',
  94. 'userns_mode',
  95. 'volume_driver',
  96. 'volumes',
  97. 'volumes_from',
  98. 'working_dir',
  99. ]
  100. ALLOWED_KEYS = DOCKER_CONFIG_KEYS + [
  101. 'build',
  102. 'container_name',
  103. 'dockerfile',
  104. 'log_driver',
  105. 'log_opt',
  106. 'logging',
  107. 'network_mode',
  108. ]
  109. DOCKER_VALID_URL_PREFIXES = (
  110. 'http://',
  111. 'https://',
  112. 'git://',
  113. 'github.com/',
  114. 'git@',
  115. )
  116. SUPPORTED_FILENAMES = [
  117. 'docker-compose.yml',
  118. 'docker-compose.yaml',
  119. ]
  120. DEFAULT_OVERRIDE_FILENAME = 'docker-compose.override.yml'
  121. log = logging.getLogger(__name__)
  122. class ConfigDetails(namedtuple('_ConfigDetails', 'working_dir config_files environment')):
  123. """
  124. :param working_dir: the directory to use for relative paths in the config
  125. :type working_dir: string
  126. :param config_files: list of configuration files to load
  127. :type config_files: list of :class:`ConfigFile`
  128. :param environment: computed environment values for this project
  129. :type environment: :class:`environment.Environment`
  130. """
  131. def __new__(cls, working_dir, config_files, environment=None):
  132. if environment is None:
  133. environment = Environment.from_env_file(working_dir)
  134. return super(ConfigDetails, cls).__new__(
  135. cls, working_dir, config_files, environment
  136. )
  137. class ConfigFile(namedtuple('_ConfigFile', 'filename config')):
  138. """
  139. :param filename: filename of the config file
  140. :type filename: string
  141. :param config: contents of the config file
  142. :type config: :class:`dict`
  143. """
  144. @classmethod
  145. def from_filename(cls, filename):
  146. return cls(filename, load_yaml(filename))
  147. @cached_property
  148. def version(self):
  149. if 'version' not in self.config:
  150. return V1
  151. version = self.config['version']
  152. if isinstance(version, dict):
  153. log.warn('Unexpected type for "version" key in "{}". Assuming '
  154. '"version" is the name of a service, and defaulting to '
  155. 'Compose file version 1.'.format(self.filename))
  156. return V1
  157. if not isinstance(version, six.string_types):
  158. raise ConfigurationError(
  159. 'Version in "{}" is invalid - it should be a string.'
  160. .format(self.filename))
  161. if version == '1':
  162. raise ConfigurationError(
  163. 'Version in "{}" is invalid. {}'
  164. .format(self.filename, VERSION_EXPLANATION))
  165. if version == '2':
  166. version = V2_0
  167. if version == '3':
  168. version = V3_0
  169. return version
  170. def get_service(self, name):
  171. return self.get_service_dicts()[name]
  172. def get_service_dicts(self):
  173. return self.config if self.version == V1 else self.config.get('services', {})
  174. def get_volumes(self):
  175. return {} if self.version == V1 else self.config.get('volumes', {})
  176. def get_networks(self):
  177. return {} if self.version == V1 else self.config.get('networks', {})
  178. def get_secrets(self):
  179. return {} if self.version < V3_1 else self.config.get('secrets', {})
  180. class Config(namedtuple('_Config', 'version services volumes networks secrets')):
  181. """
  182. :param version: configuration version
  183. :type version: int
  184. :param services: List of service description dictionaries
  185. :type services: :class:`list`
  186. :param volumes: Dictionary mapping volume names to description dictionaries
  187. :type volumes: :class:`dict`
  188. :param networks: Dictionary mapping network names to description dictionaries
  189. :type networks: :class:`dict`
  190. """
  191. class ServiceConfig(namedtuple('_ServiceConfig', 'working_dir filename name config')):
  192. @classmethod
  193. def with_abs_paths(cls, working_dir, filename, name, config):
  194. if not working_dir:
  195. raise ValueError("No working_dir for ServiceConfig.")
  196. return cls(
  197. os.path.abspath(working_dir),
  198. os.path.abspath(filename) if filename else filename,
  199. name,
  200. config)
  201. def find(base_dir, filenames, environment):
  202. if filenames == ['-']:
  203. return ConfigDetails(
  204. os.getcwd(),
  205. [ConfigFile(None, yaml.safe_load(sys.stdin))],
  206. environment
  207. )
  208. if filenames:
  209. filenames = [os.path.join(base_dir, f) for f in filenames]
  210. else:
  211. filenames = get_default_config_files(base_dir)
  212. log.debug("Using configuration files: {}".format(",".join(filenames)))
  213. return ConfigDetails(
  214. os.path.dirname(filenames[0]),
  215. [ConfigFile.from_filename(f) for f in filenames],
  216. environment
  217. )
  218. def validate_config_version(config_files):
  219. main_file = config_files[0]
  220. validate_top_level_object(main_file)
  221. for next_file in config_files[1:]:
  222. validate_top_level_object(next_file)
  223. if main_file.version != next_file.version:
  224. raise ConfigurationError(
  225. "Version mismatch: file {0} specifies version {1} but "
  226. "extension file {2} uses version {3}".format(
  227. main_file.filename,
  228. main_file.version,
  229. next_file.filename,
  230. next_file.version))
  231. def get_default_config_files(base_dir):
  232. (candidates, path) = find_candidates_in_parent_dirs(SUPPORTED_FILENAMES, base_dir)
  233. if not candidates:
  234. raise ComposeFileNotFound(SUPPORTED_FILENAMES)
  235. winner = candidates[0]
  236. if len(candidates) > 1:
  237. log.warn("Found multiple config files with supported names: %s", ", ".join(candidates))
  238. log.warn("Using %s\n", winner)
  239. return [os.path.join(path, winner)] + get_default_override_file(path)
  240. def get_default_override_file(path):
  241. override_filename = os.path.join(path, DEFAULT_OVERRIDE_FILENAME)
  242. return [override_filename] if os.path.exists(override_filename) else []
  243. def find_candidates_in_parent_dirs(filenames, path):
  244. """
  245. Given a directory path to start, looks for filenames in the
  246. directory, and then each parent directory successively,
  247. until found.
  248. Returns tuple (candidates, path).
  249. """
  250. candidates = [filename for filename in filenames
  251. if os.path.exists(os.path.join(path, filename))]
  252. if not candidates:
  253. parent_dir = os.path.join(path, '..')
  254. if os.path.abspath(parent_dir) != os.path.abspath(path):
  255. return find_candidates_in_parent_dirs(filenames, parent_dir)
  256. return (candidates, path)
  257. def load(config_details):
  258. """Load the configuration from a working directory and a list of
  259. configuration files. Files are loaded in order, and merged on top
  260. of each other to create the final configuration.
  261. Return a fully interpolated, extended and validated configuration.
  262. """
  263. validate_config_version(config_details.config_files)
  264. processed_files = [
  265. process_config_file(config_file, config_details.environment)
  266. for config_file in config_details.config_files
  267. ]
  268. config_details = config_details._replace(config_files=processed_files)
  269. main_file = config_details.config_files[0]
  270. volumes = load_mapping(
  271. config_details.config_files, 'get_volumes', 'Volume'
  272. )
  273. networks = load_mapping(
  274. config_details.config_files, 'get_networks', 'Network'
  275. )
  276. secrets = load_secrets(config_details.config_files, config_details.working_dir)
  277. service_dicts = load_services(config_details, main_file)
  278. if main_file.version != V1:
  279. for service_dict in service_dicts:
  280. match_named_volumes(service_dict, volumes)
  281. services_using_deploy = [s for s in service_dicts if s.get('deploy')]
  282. if services_using_deploy:
  283. log.warn(
  284. "Some services ({}) use the 'deploy' key, which will be ignored. "
  285. "Compose does not support deploy configuration - use "
  286. "`docker stack deploy` to deploy to a swarm."
  287. .format(", ".join(sorted(s['name'] for s in services_using_deploy))))
  288. return Config(main_file.version, service_dicts, volumes, networks, secrets)
  289. def load_mapping(config_files, get_func, entity_type):
  290. mapping = {}
  291. for config_file in config_files:
  292. for name, config in getattr(config_file, get_func)().items():
  293. mapping[name] = config or {}
  294. if not config:
  295. continue
  296. external = config.get('external')
  297. if external:
  298. validate_external(entity_type, name, config)
  299. if isinstance(external, dict):
  300. config['external_name'] = external.get('name')
  301. else:
  302. config['external_name'] = name
  303. if 'driver_opts' in config:
  304. config['driver_opts'] = build_string_dict(
  305. config['driver_opts']
  306. )
  307. if 'labels' in config:
  308. config['labels'] = parse_labels(config['labels'])
  309. return mapping
  310. def validate_external(entity_type, name, config):
  311. if len(config.keys()) <= 1:
  312. return
  313. raise ConfigurationError(
  314. "{} {} declared as external but specifies additional attributes "
  315. "({}).".format(
  316. entity_type, name, ', '.join(k for k in config if k != 'external')))
  317. def load_secrets(config_files, working_dir):
  318. mapping = {}
  319. for config_file in config_files:
  320. for name, config in config_file.get_secrets().items():
  321. mapping[name] = config or {}
  322. if not config:
  323. continue
  324. external = config.get('external')
  325. if external:
  326. validate_external('Secret', name, config)
  327. if isinstance(external, dict):
  328. config['external_name'] = external.get('name')
  329. else:
  330. config['external_name'] = name
  331. if 'file' in config:
  332. config['file'] = expand_path(working_dir, config['file'])
  333. return mapping
  334. def load_services(config_details, config_file):
  335. def build_service(service_name, service_dict, service_names):
  336. service_config = ServiceConfig.with_abs_paths(
  337. config_details.working_dir,
  338. config_file.filename,
  339. service_name,
  340. service_dict)
  341. resolver = ServiceExtendsResolver(
  342. service_config, config_file, environment=config_details.environment
  343. )
  344. service_dict = process_service(resolver.run())
  345. service_config = service_config._replace(config=service_dict)
  346. validate_service(service_config, service_names, config_file.version)
  347. service_dict = finalize_service(
  348. service_config,
  349. service_names,
  350. config_file.version,
  351. config_details.environment)
  352. return service_dict
  353. def build_services(service_config):
  354. service_names = service_config.keys()
  355. return sort_service_dicts([
  356. build_service(name, service_dict, service_names)
  357. for name, service_dict in service_config.items()
  358. ])
  359. def merge_services(base, override):
  360. all_service_names = set(base) | set(override)
  361. return {
  362. name: merge_service_dicts_from_files(
  363. base.get(name, {}),
  364. override.get(name, {}),
  365. config_file.version)
  366. for name in all_service_names
  367. }
  368. service_configs = [
  369. file.get_service_dicts() for file in config_details.config_files
  370. ]
  371. service_config = service_configs[0]
  372. for next_config in service_configs[1:]:
  373. service_config = merge_services(service_config, next_config)
  374. return build_services(service_config)
  375. def interpolate_config_section(config_file, config, section, environment):
  376. validate_config_section(config_file.filename, config, section)
  377. return interpolate_environment_variables(
  378. config_file.version,
  379. config,
  380. section,
  381. environment
  382. )
  383. def process_config_file(config_file, environment, service_name=None):
  384. services = interpolate_config_section(
  385. config_file,
  386. config_file.get_service_dicts(),
  387. 'service',
  388. environment)
  389. if config_file.version in (V2_0, V2_1, V3_0, V3_1):
  390. processed_config = dict(config_file.config)
  391. processed_config['services'] = services
  392. processed_config['volumes'] = interpolate_config_section(
  393. config_file,
  394. config_file.get_volumes(),
  395. 'volume',
  396. environment)
  397. processed_config['networks'] = interpolate_config_section(
  398. config_file,
  399. config_file.get_networks(),
  400. 'network',
  401. environment)
  402. elif config_file.version == V1:
  403. processed_config = services
  404. else:
  405. raise ConfigurationError(
  406. 'Version in "{}" is unsupported. {}'
  407. .format(config_file.filename, VERSION_EXPLANATION))
  408. config_file = config_file._replace(config=processed_config)
  409. validate_against_config_schema(config_file)
  410. if service_name and service_name not in services:
  411. raise ConfigurationError(
  412. "Cannot extend service '{}' in {}: Service not found".format(
  413. service_name, config_file.filename))
  414. return config_file
  415. class ServiceExtendsResolver(object):
  416. def __init__(self, service_config, config_file, environment, already_seen=None):
  417. self.service_config = service_config
  418. self.working_dir = service_config.working_dir
  419. self.already_seen = already_seen or []
  420. self.config_file = config_file
  421. self.environment = environment
  422. @property
  423. def signature(self):
  424. return self.service_config.filename, self.service_config.name
  425. def detect_cycle(self):
  426. if self.signature in self.already_seen:
  427. raise CircularReference(self.already_seen + [self.signature])
  428. def run(self):
  429. self.detect_cycle()
  430. if 'extends' in self.service_config.config:
  431. service_dict = self.resolve_extends(*self.validate_and_construct_extends())
  432. return self.service_config._replace(config=service_dict)
  433. return self.service_config
  434. def validate_and_construct_extends(self):
  435. extends = self.service_config.config['extends']
  436. if not isinstance(extends, dict):
  437. extends = {'service': extends}
  438. config_path = self.get_extended_config_path(extends)
  439. service_name = extends['service']
  440. extends_file = ConfigFile.from_filename(config_path)
  441. validate_config_version([self.config_file, extends_file])
  442. extended_file = process_config_file(
  443. extends_file, self.environment, service_name=service_name
  444. )
  445. service_config = extended_file.get_service(service_name)
  446. return config_path, service_config, service_name
  447. def resolve_extends(self, extended_config_path, service_dict, service_name):
  448. resolver = ServiceExtendsResolver(
  449. ServiceConfig.with_abs_paths(
  450. os.path.dirname(extended_config_path),
  451. extended_config_path,
  452. service_name,
  453. service_dict),
  454. self.config_file,
  455. already_seen=self.already_seen + [self.signature],
  456. environment=self.environment
  457. )
  458. service_config = resolver.run()
  459. other_service_dict = process_service(service_config)
  460. validate_extended_service_dict(
  461. other_service_dict,
  462. extended_config_path,
  463. service_name)
  464. return merge_service_dicts(
  465. other_service_dict,
  466. self.service_config.config,
  467. self.config_file.version)
  468. def get_extended_config_path(self, extends_options):
  469. """Service we are extending either has a value for 'file' set, which we
  470. need to obtain a full path too or we are extending from a service
  471. defined in our own file.
  472. """
  473. filename = self.service_config.filename
  474. validate_extends_file_path(
  475. self.service_config.name,
  476. extends_options,
  477. filename)
  478. if 'file' in extends_options:
  479. return expand_path(self.working_dir, extends_options['file'])
  480. return filename
  481. def resolve_environment(service_dict, environment=None):
  482. """Unpack any environment variables from an env_file, if set.
  483. Interpolate environment values if set.
  484. """
  485. env = {}
  486. for env_file in service_dict.get('env_file', []):
  487. env.update(env_vars_from_file(env_file))
  488. env.update(parse_environment(service_dict.get('environment')))
  489. return dict(resolve_env_var(k, v, environment) for k, v in six.iteritems(env))
  490. def resolve_build_args(build, environment):
  491. args = parse_build_arguments(build.get('args'))
  492. return dict(resolve_env_var(k, v, environment) for k, v in six.iteritems(args))
  493. def validate_extended_service_dict(service_dict, filename, service):
  494. error_prefix = "Cannot extend service '%s' in %s:" % (service, filename)
  495. if 'links' in service_dict:
  496. raise ConfigurationError(
  497. "%s services with 'links' cannot be extended" % error_prefix)
  498. if 'volumes_from' in service_dict:
  499. raise ConfigurationError(
  500. "%s services with 'volumes_from' cannot be extended" % error_prefix)
  501. if 'net' in service_dict:
  502. if get_container_name_from_network_mode(service_dict['net']):
  503. raise ConfigurationError(
  504. "%s services with 'net: container' cannot be extended" % error_prefix)
  505. if 'network_mode' in service_dict:
  506. if get_service_name_from_network_mode(service_dict['network_mode']):
  507. raise ConfigurationError(
  508. "%s services with 'network_mode: service' cannot be extended" % error_prefix)
  509. if 'depends_on' in service_dict:
  510. raise ConfigurationError(
  511. "%s services with 'depends_on' cannot be extended" % error_prefix)
  512. def validate_service(service_config, service_names, version):
  513. service_dict, service_name = service_config.config, service_config.name
  514. validate_service_constraints(service_dict, service_name, version)
  515. validate_paths(service_dict)
  516. validate_ulimits(service_config)
  517. validate_network_mode(service_config, service_names)
  518. validate_depends_on(service_config, service_names)
  519. validate_links(service_config, service_names)
  520. if not service_dict.get('image') and has_uppercase(service_name):
  521. raise ConfigurationError(
  522. "Service '{name}' contains uppercase characters which are not valid "
  523. "as part of an image name. Either use a lowercase service name or "
  524. "use the `image` field to set a custom name for the service image."
  525. .format(name=service_name))
  526. def process_service(service_config):
  527. working_dir = service_config.working_dir
  528. service_dict = dict(service_config.config)
  529. if 'env_file' in service_dict:
  530. service_dict['env_file'] = [
  531. expand_path(working_dir, path)
  532. for path in to_list(service_dict['env_file'])
  533. ]
  534. if 'build' in service_dict:
  535. if isinstance(service_dict['build'], six.string_types):
  536. service_dict['build'] = resolve_build_path(working_dir, service_dict['build'])
  537. elif isinstance(service_dict['build'], dict) and 'context' in service_dict['build']:
  538. path = service_dict['build']['context']
  539. service_dict['build']['context'] = resolve_build_path(working_dir, path)
  540. if 'volumes' in service_dict and service_dict.get('volume_driver') is None:
  541. service_dict['volumes'] = resolve_volume_paths(working_dir, service_dict)
  542. if 'labels' in service_dict:
  543. service_dict['labels'] = parse_labels(service_dict['labels'])
  544. if 'extra_hosts' in service_dict:
  545. service_dict['extra_hosts'] = parse_extra_hosts(service_dict['extra_hosts'])
  546. if 'sysctls' in service_dict:
  547. service_dict['sysctls'] = build_string_dict(parse_sysctls(service_dict['sysctls']))
  548. service_dict = process_depends_on(service_dict)
  549. for field in ['dns', 'dns_search', 'tmpfs']:
  550. if field in service_dict:
  551. service_dict[field] = to_list(service_dict[field])
  552. service_dict = process_healthcheck(service_dict, service_config.name)
  553. service_dict = process_ports(service_dict)
  554. return service_dict
  555. def process_ports(service_dict):
  556. if 'ports' not in service_dict:
  557. return service_dict
  558. ports = []
  559. for port_definition in service_dict['ports']:
  560. if isinstance(port_definition, ServicePort):
  561. ports.append(port_definition)
  562. else:
  563. ports.extend(ServicePort.parse(port_definition))
  564. service_dict['ports'] = ports
  565. return service_dict
  566. def process_depends_on(service_dict):
  567. if 'depends_on' in service_dict and not isinstance(service_dict['depends_on'], dict):
  568. service_dict['depends_on'] = dict([
  569. (svc, {'condition': 'service_started'}) for svc in service_dict['depends_on']
  570. ])
  571. return service_dict
  572. def process_healthcheck(service_dict, service_name):
  573. if 'healthcheck' not in service_dict:
  574. return service_dict
  575. hc = {}
  576. raw = service_dict['healthcheck']
  577. if raw.get('disable'):
  578. if len(raw) > 1:
  579. raise ConfigurationError(
  580. 'Service "{}" defines an invalid healthcheck: '
  581. '"disable: true" cannot be combined with other options'
  582. .format(service_name))
  583. hc['test'] = ['NONE']
  584. elif 'test' in raw:
  585. hc['test'] = raw['test']
  586. if 'interval' in raw:
  587. if not isinstance(raw['interval'], six.integer_types):
  588. hc['interval'] = parse_nanoseconds_int(raw['interval'])
  589. else: # Conversion has been done previously
  590. hc['interval'] = raw['interval']
  591. if 'timeout' in raw:
  592. if not isinstance(raw['timeout'], six.integer_types):
  593. hc['timeout'] = parse_nanoseconds_int(raw['timeout'])
  594. else: # Conversion has been done previously
  595. hc['timeout'] = raw['timeout']
  596. if 'retries' in raw:
  597. hc['retries'] = raw['retries']
  598. service_dict['healthcheck'] = hc
  599. return service_dict
  600. def finalize_service(service_config, service_names, version, environment):
  601. service_dict = dict(service_config.config)
  602. if 'environment' in service_dict or 'env_file' in service_dict:
  603. service_dict['environment'] = resolve_environment(service_dict, environment)
  604. service_dict.pop('env_file', None)
  605. if 'volumes_from' in service_dict:
  606. service_dict['volumes_from'] = [
  607. VolumeFromSpec.parse(vf, service_names, version)
  608. for vf in service_dict['volumes_from']
  609. ]
  610. if 'volumes' in service_dict:
  611. service_dict['volumes'] = [
  612. VolumeSpec.parse(
  613. v, environment.get_boolean('COMPOSE_CONVERT_WINDOWS_PATHS')
  614. ) for v in service_dict['volumes']
  615. ]
  616. if 'net' in service_dict:
  617. network_mode = service_dict.pop('net')
  618. container_name = get_container_name_from_network_mode(network_mode)
  619. if container_name and container_name in service_names:
  620. service_dict['network_mode'] = 'service:{}'.format(container_name)
  621. else:
  622. service_dict['network_mode'] = network_mode
  623. if 'networks' in service_dict:
  624. service_dict['networks'] = parse_networks(service_dict['networks'])
  625. if 'restart' in service_dict:
  626. service_dict['restart'] = parse_restart_spec(service_dict['restart'])
  627. if 'secrets' in service_dict:
  628. service_dict['secrets'] = [
  629. types.ServiceSecret.parse(s) for s in service_dict['secrets']
  630. ]
  631. normalize_build(service_dict, service_config.working_dir, environment)
  632. service_dict['name'] = service_config.name
  633. return normalize_v1_service_format(service_dict)
  634. def normalize_v1_service_format(service_dict):
  635. if 'log_driver' in service_dict or 'log_opt' in service_dict:
  636. if 'logging' not in service_dict:
  637. service_dict['logging'] = {}
  638. if 'log_driver' in service_dict:
  639. service_dict['logging']['driver'] = service_dict['log_driver']
  640. del service_dict['log_driver']
  641. if 'log_opt' in service_dict:
  642. service_dict['logging']['options'] = service_dict['log_opt']
  643. del service_dict['log_opt']
  644. if 'dockerfile' in service_dict:
  645. service_dict['build'] = service_dict.get('build', {})
  646. service_dict['build'].update({
  647. 'dockerfile': service_dict.pop('dockerfile')
  648. })
  649. return service_dict
  650. def merge_service_dicts_from_files(base, override, version):
  651. """When merging services from multiple files we need to merge the `extends`
  652. field. This is not handled by `merge_service_dicts()` which is used to
  653. perform the `extends`.
  654. """
  655. new_service = merge_service_dicts(base, override, version)
  656. if 'extends' in override:
  657. new_service['extends'] = override['extends']
  658. elif 'extends' in base:
  659. new_service['extends'] = base['extends']
  660. return new_service
  661. class MergeDict(dict):
  662. """A dict-like object responsible for merging two dicts into one."""
  663. def __init__(self, base, override):
  664. self.base = base
  665. self.override = override
  666. def needs_merge(self, field):
  667. return field in self.base or field in self.override
  668. def merge_field(self, field, merge_func, default=None):
  669. if not self.needs_merge(field):
  670. return
  671. self[field] = merge_func(
  672. self.base.get(field, default),
  673. self.override.get(field, default))
  674. def merge_mapping(self, field, parse_func):
  675. if not self.needs_merge(field):
  676. return
  677. self[field] = parse_func(self.base.get(field))
  678. self[field].update(parse_func(self.override.get(field)))
  679. def merge_sequence(self, field, parse_func):
  680. def parse_sequence_func(seq):
  681. return to_mapping((parse_func(item) for item in seq), 'merge_field')
  682. if not self.needs_merge(field):
  683. return
  684. merged = parse_sequence_func(self.base.get(field, []))
  685. merged.update(parse_sequence_func(self.override.get(field, [])))
  686. self[field] = [item.repr() for item in sorted(merged.values())]
  687. def merge_scalar(self, field):
  688. if self.needs_merge(field):
  689. self[field] = self.override.get(field, self.base.get(field))
  690. def merge_service_dicts(base, override, version):
  691. md = MergeDict(base, override)
  692. md.merge_mapping('environment', parse_environment)
  693. md.merge_mapping('labels', parse_labels)
  694. md.merge_mapping('ulimits', parse_ulimits)
  695. md.merge_mapping('networks', parse_networks)
  696. md.merge_mapping('sysctls', parse_sysctls)
  697. md.merge_mapping('depends_on', parse_depends_on)
  698. md.merge_sequence('links', ServiceLink.parse)
  699. md.merge_sequence('secrets', types.ServiceSecret.parse)
  700. for field in ['volumes', 'devices']:
  701. md.merge_field(field, merge_path_mappings)
  702. for field in [
  703. 'cap_add', 'cap_drop', 'expose', 'external_links',
  704. 'security_opt', 'volumes_from',
  705. ]:
  706. md.merge_field(field, merge_unique_items_lists, default=[])
  707. for field in ['dns', 'dns_search', 'env_file', 'tmpfs']:
  708. md.merge_field(field, merge_list_or_string)
  709. md.merge_field('logging', merge_logging, default={})
  710. merge_ports(md, base, override)
  711. for field in set(ALLOWED_KEYS) - set(md):
  712. md.merge_scalar(field)
  713. if version == V1:
  714. legacy_v1_merge_image_or_build(md, base, override)
  715. elif md.needs_merge('build'):
  716. md['build'] = merge_build(md, base, override)
  717. return dict(md)
  718. def merge_unique_items_lists(base, override):
  719. override = [str(o) for o in override]
  720. base = [str(b) for b in base]
  721. return sorted(set().union(base, override))
  722. def merge_ports(md, base, override):
  723. def parse_sequence_func(seq):
  724. acc = []
  725. for item in seq:
  726. acc.extend(ServicePort.parse(item))
  727. return to_mapping(acc, 'merge_field')
  728. field = 'ports'
  729. if not md.needs_merge(field):
  730. return
  731. merged = parse_sequence_func(md.base.get(field, []))
  732. merged.update(parse_sequence_func(md.override.get(field, [])))
  733. md[field] = [item for item in sorted(merged.values())]
  734. def merge_build(output, base, override):
  735. def to_dict(service):
  736. build_config = service.get('build', {})
  737. if isinstance(build_config, six.string_types):
  738. return {'context': build_config}
  739. return build_config
  740. md = MergeDict(to_dict(base), to_dict(override))
  741. md.merge_scalar('context')
  742. md.merge_scalar('dockerfile')
  743. md.merge_mapping('args', parse_build_arguments)
  744. return dict(md)
  745. def merge_logging(base, override):
  746. md = MergeDict(base, override)
  747. md.merge_scalar('driver')
  748. if md.get('driver') == base.get('driver') or base.get('driver') is None:
  749. md.merge_mapping('options', lambda m: m or {})
  750. else:
  751. md['options'] = override.get('options')
  752. return dict(md)
  753. def legacy_v1_merge_image_or_build(output, base, override):
  754. output.pop('image', None)
  755. output.pop('build', None)
  756. if 'image' in override:
  757. output['image'] = override['image']
  758. elif 'build' in override:
  759. output['build'] = override['build']
  760. elif 'image' in base:
  761. output['image'] = base['image']
  762. elif 'build' in base:
  763. output['build'] = base['build']
  764. def merge_environment(base, override):
  765. env = parse_environment(base)
  766. env.update(parse_environment(override))
  767. return env
  768. def split_kv(kvpair):
  769. if '=' in kvpair:
  770. return kvpair.split('=', 1)
  771. else:
  772. return kvpair, ''
  773. def parse_dict_or_list(split_func, type_name, arguments):
  774. if not arguments:
  775. return {}
  776. if isinstance(arguments, list):
  777. return dict(split_func(e) for e in arguments)
  778. if isinstance(arguments, dict):
  779. return dict(arguments)
  780. raise ConfigurationError(
  781. "%s \"%s\" must be a list or mapping," %
  782. (type_name, arguments)
  783. )
  784. parse_build_arguments = functools.partial(parse_dict_or_list, split_env, 'build arguments')
  785. parse_environment = functools.partial(parse_dict_or_list, split_env, 'environment')
  786. parse_labels = functools.partial(parse_dict_or_list, split_kv, 'labels')
  787. parse_networks = functools.partial(parse_dict_or_list, lambda k: (k, None), 'networks')
  788. parse_sysctls = functools.partial(parse_dict_or_list, split_kv, 'sysctls')
  789. parse_depends_on = functools.partial(
  790. parse_dict_or_list, lambda k: (k, {'condition': 'service_started'}), 'depends_on'
  791. )
  792. def parse_ulimits(ulimits):
  793. if not ulimits:
  794. return {}
  795. if isinstance(ulimits, dict):
  796. return dict(ulimits)
  797. def resolve_env_var(key, val, environment):
  798. if val is not None:
  799. return key, val
  800. elif environment and key in environment:
  801. return key, environment[key]
  802. else:
  803. return key, None
  804. def resolve_volume_paths(working_dir, service_dict):
  805. return [
  806. resolve_volume_path(working_dir, volume)
  807. for volume in service_dict['volumes']
  808. ]
  809. def resolve_volume_path(working_dir, volume):
  810. container_path, host_path = split_path_mapping(volume)
  811. if host_path is not None:
  812. if host_path.startswith('.'):
  813. host_path = expand_path(working_dir, host_path)
  814. host_path = os.path.expanduser(host_path)
  815. return u"{}:{}".format(host_path, container_path)
  816. else:
  817. return container_path
  818. def normalize_build(service_dict, working_dir, environment):
  819. if 'build' in service_dict:
  820. build = {}
  821. # Shortcut where specifying a string is treated as the build context
  822. if isinstance(service_dict['build'], six.string_types):
  823. build['context'] = service_dict.pop('build')
  824. else:
  825. build.update(service_dict['build'])
  826. if 'args' in build:
  827. build['args'] = build_string_dict(
  828. resolve_build_args(build, environment)
  829. )
  830. service_dict['build'] = build
  831. def resolve_build_path(working_dir, build_path):
  832. if is_url(build_path):
  833. return build_path
  834. return expand_path(working_dir, build_path)
  835. def is_url(build_path):
  836. return build_path.startswith(DOCKER_VALID_URL_PREFIXES)
  837. def validate_paths(service_dict):
  838. if 'build' in service_dict:
  839. build = service_dict.get('build', {})
  840. if isinstance(build, six.string_types):
  841. build_path = build
  842. elif isinstance(build, dict) and 'context' in build:
  843. build_path = build['context']
  844. else:
  845. # We have a build section but no context, so nothing to validate
  846. return
  847. if (
  848. not is_url(build_path) and
  849. (not os.path.exists(build_path) or not os.access(build_path, os.R_OK))
  850. ):
  851. raise ConfigurationError(
  852. "build path %s either does not exist, is not accessible, "
  853. "or is not a valid URL." % build_path)
  854. def merge_path_mappings(base, override):
  855. d = dict_from_path_mappings(base)
  856. d.update(dict_from_path_mappings(override))
  857. return path_mappings_from_dict(d)
  858. def dict_from_path_mappings(path_mappings):
  859. if path_mappings:
  860. return dict(split_path_mapping(v) for v in path_mappings)
  861. else:
  862. return {}
  863. def path_mappings_from_dict(d):
  864. return [join_path_mapping(v) for v in sorted(d.items())]
  865. def split_path_mapping(volume_path):
  866. """
  867. Ascertain if the volume_path contains a host path as well as a container
  868. path. Using splitdrive so windows absolute paths won't cause issues with
  869. splitting on ':'.
  870. """
  871. drive, volume_config = splitdrive(volume_path)
  872. if ':' in volume_config:
  873. (host, container) = volume_config.split(':', 1)
  874. return (container, drive + host)
  875. else:
  876. return (volume_path, None)
  877. def join_path_mapping(pair):
  878. (container, host) = pair
  879. if host is None:
  880. return container
  881. else:
  882. return ":".join((host, container))
  883. def expand_path(working_dir, path):
  884. return os.path.abspath(os.path.join(working_dir, os.path.expanduser(path)))
  885. def merge_list_or_string(base, override):
  886. return to_list(base) + to_list(override)
  887. def to_list(value):
  888. if value is None:
  889. return []
  890. elif isinstance(value, six.string_types):
  891. return [value]
  892. else:
  893. return value
  894. def to_mapping(sequence, key_field):
  895. return {getattr(item, key_field): item for item in sequence}
  896. def has_uppercase(name):
  897. return any(char in string.ascii_uppercase for char in name)
  898. def load_yaml(filename):
  899. try:
  900. with open(filename, 'r') as fh:
  901. return yaml.safe_load(fh)
  902. except (IOError, yaml.YAMLError) as e:
  903. error_name = getattr(e, '__module__', '') + '.' + e.__class__.__name__
  904. raise ConfigurationError(u"{}: {}".format(error_name, e))