config.py 37 KB

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