config.py 38 KB

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