config.py 31 KB

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