config.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995
  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(filename, config, section, environment):
  339. validate_config_section(filename, config, section)
  340. return interpolate_environment_variables(config, section, environment)
  341. def process_config_file(config_file, environment, service_name=None):
  342. services = interpolate_config_section(
  343. config_file.filename,
  344. config_file.get_service_dicts(),
  345. 'service',
  346. environment,)
  347. if config_file.version in (V2_0, V2_1):
  348. processed_config = dict(config_file.config)
  349. processed_config['services'] = services
  350. processed_config['volumes'] = interpolate_config_section(
  351. config_file.filename,
  352. config_file.get_volumes(),
  353. 'volume',
  354. environment,)
  355. processed_config['networks'] = interpolate_config_section(
  356. config_file.filename,
  357. config_file.get_networks(),
  358. 'network',
  359. environment,)
  360. if config_file.version == V1:
  361. processed_config = services
  362. config_file = config_file._replace(config=processed_config)
  363. validate_against_config_schema(config_file)
  364. if service_name and service_name not in services:
  365. raise ConfigurationError(
  366. "Cannot extend service '{}' in {}: Service not found".format(
  367. service_name, config_file.filename))
  368. return config_file
  369. class ServiceExtendsResolver(object):
  370. def __init__(self, service_config, config_file, environment, already_seen=None):
  371. self.service_config = service_config
  372. self.working_dir = service_config.working_dir
  373. self.already_seen = already_seen or []
  374. self.config_file = config_file
  375. self.environment = environment
  376. @property
  377. def signature(self):
  378. return self.service_config.filename, self.service_config.name
  379. def detect_cycle(self):
  380. if self.signature in self.already_seen:
  381. raise CircularReference(self.already_seen + [self.signature])
  382. def run(self):
  383. self.detect_cycle()
  384. if 'extends' in self.service_config.config:
  385. service_dict = self.resolve_extends(*self.validate_and_construct_extends())
  386. return self.service_config._replace(config=service_dict)
  387. return self.service_config
  388. def validate_and_construct_extends(self):
  389. extends = self.service_config.config['extends']
  390. if not isinstance(extends, dict):
  391. extends = {'service': extends}
  392. config_path = self.get_extended_config_path(extends)
  393. service_name = extends['service']
  394. extends_file = ConfigFile.from_filename(config_path)
  395. validate_config_version([self.config_file, extends_file])
  396. extended_file = process_config_file(
  397. extends_file, self.environment, service_name=service_name
  398. )
  399. service_config = extended_file.get_service(service_name)
  400. return config_path, service_config, service_name
  401. def resolve_extends(self, extended_config_path, service_dict, service_name):
  402. resolver = ServiceExtendsResolver(
  403. ServiceConfig.with_abs_paths(
  404. os.path.dirname(extended_config_path),
  405. extended_config_path,
  406. service_name,
  407. service_dict),
  408. self.config_file,
  409. already_seen=self.already_seen + [self.signature],
  410. environment=self.environment
  411. )
  412. service_config = resolver.run()
  413. other_service_dict = process_service(service_config)
  414. validate_extended_service_dict(
  415. other_service_dict,
  416. extended_config_path,
  417. service_name)
  418. return merge_service_dicts(
  419. other_service_dict,
  420. self.service_config.config,
  421. self.config_file.version)
  422. def get_extended_config_path(self, extends_options):
  423. """Service we are extending either has a value for 'file' set, which we
  424. need to obtain a full path too or we are extending from a service
  425. defined in our own file.
  426. """
  427. filename = self.service_config.filename
  428. validate_extends_file_path(
  429. self.service_config.name,
  430. extends_options,
  431. filename)
  432. if 'file' in extends_options:
  433. return expand_path(self.working_dir, extends_options['file'])
  434. return filename
  435. def resolve_environment(service_dict, environment=None):
  436. """Unpack any environment variables from an env_file, if set.
  437. Interpolate environment values if set.
  438. """
  439. env = {}
  440. for env_file in service_dict.get('env_file', []):
  441. env.update(env_vars_from_file(env_file))
  442. env.update(parse_environment(service_dict.get('environment')))
  443. return dict(resolve_env_var(k, v, environment) for k, v in six.iteritems(env))
  444. def resolve_build_args(build, environment):
  445. args = parse_build_arguments(build.get('args'))
  446. return dict(resolve_env_var(k, v, environment) for k, v in six.iteritems(args))
  447. def validate_extended_service_dict(service_dict, filename, service):
  448. error_prefix = "Cannot extend service '%s' in %s:" % (service, filename)
  449. if 'links' in service_dict:
  450. raise ConfigurationError(
  451. "%s services with 'links' cannot be extended" % error_prefix)
  452. if 'volumes_from' in service_dict:
  453. raise ConfigurationError(
  454. "%s services with 'volumes_from' cannot be extended" % error_prefix)
  455. if 'net' in service_dict:
  456. if get_container_name_from_network_mode(service_dict['net']):
  457. raise ConfigurationError(
  458. "%s services with 'net: container' cannot be extended" % error_prefix)
  459. if 'network_mode' in service_dict:
  460. if get_service_name_from_network_mode(service_dict['network_mode']):
  461. raise ConfigurationError(
  462. "%s services with 'network_mode: service' cannot be extended" % error_prefix)
  463. if 'depends_on' in service_dict:
  464. raise ConfigurationError(
  465. "%s services with 'depends_on' cannot be extended" % error_prefix)
  466. def validate_service(service_config, service_names, version):
  467. service_dict, service_name = service_config.config, service_config.name
  468. validate_service_constraints(service_dict, service_name, version)
  469. validate_paths(service_dict)
  470. validate_ulimits(service_config)
  471. validate_network_mode(service_config, service_names)
  472. validate_depends_on(service_config, service_names)
  473. validate_links(service_config, service_names)
  474. if not service_dict.get('image') and has_uppercase(service_name):
  475. raise ConfigurationError(
  476. "Service '{name}' contains uppercase characters which are not valid "
  477. "as part of an image name. Either use a lowercase service name or "
  478. "use the `image` field to set a custom name for the service image."
  479. .format(name=service_name))
  480. def process_service(service_config):
  481. working_dir = service_config.working_dir
  482. service_dict = dict(service_config.config)
  483. if 'env_file' in service_dict:
  484. service_dict['env_file'] = [
  485. expand_path(working_dir, path)
  486. for path in to_list(service_dict['env_file'])
  487. ]
  488. if 'build' in service_dict:
  489. if isinstance(service_dict['build'], six.string_types):
  490. service_dict['build'] = resolve_build_path(working_dir, service_dict['build'])
  491. elif isinstance(service_dict['build'], dict) and 'context' in service_dict['build']:
  492. path = service_dict['build']['context']
  493. service_dict['build']['context'] = resolve_build_path(working_dir, path)
  494. if 'volumes' in service_dict and service_dict.get('volume_driver') is None:
  495. service_dict['volumes'] = resolve_volume_paths(working_dir, service_dict)
  496. if 'labels' in service_dict:
  497. service_dict['labels'] = parse_labels(service_dict['labels'])
  498. if 'extra_hosts' in service_dict:
  499. service_dict['extra_hosts'] = parse_extra_hosts(service_dict['extra_hosts'])
  500. for field in ['dns', 'dns_search', 'tmpfs']:
  501. if field in service_dict:
  502. service_dict[field] = to_list(service_dict[field])
  503. return service_dict
  504. def finalize_service(service_config, service_names, version, environment):
  505. service_dict = dict(service_config.config)
  506. if 'environment' in service_dict or 'env_file' in service_dict:
  507. service_dict['environment'] = resolve_environment(service_dict, environment)
  508. service_dict.pop('env_file', None)
  509. if 'volumes_from' in service_dict:
  510. service_dict['volumes_from'] = [
  511. VolumeFromSpec.parse(vf, service_names, version)
  512. for vf in service_dict['volumes_from']
  513. ]
  514. if 'volumes' in service_dict:
  515. service_dict['volumes'] = [
  516. VolumeSpec.parse(v) for v in service_dict['volumes']]
  517. if 'net' in service_dict:
  518. network_mode = service_dict.pop('net')
  519. container_name = get_container_name_from_network_mode(network_mode)
  520. if container_name and container_name in service_names:
  521. service_dict['network_mode'] = 'service:{}'.format(container_name)
  522. else:
  523. service_dict['network_mode'] = network_mode
  524. if 'networks' in service_dict:
  525. service_dict['networks'] = parse_networks(service_dict['networks'])
  526. if 'restart' in service_dict:
  527. service_dict['restart'] = parse_restart_spec(service_dict['restart'])
  528. normalize_build(service_dict, service_config.working_dir, environment)
  529. service_dict['name'] = service_config.name
  530. return normalize_v1_service_format(service_dict)
  531. def normalize_v1_service_format(service_dict):
  532. if 'log_driver' in service_dict or 'log_opt' in service_dict:
  533. if 'logging' not in service_dict:
  534. service_dict['logging'] = {}
  535. if 'log_driver' in service_dict:
  536. service_dict['logging']['driver'] = service_dict['log_driver']
  537. del service_dict['log_driver']
  538. if 'log_opt' in service_dict:
  539. service_dict['logging']['options'] = service_dict['log_opt']
  540. del service_dict['log_opt']
  541. if 'dockerfile' in service_dict:
  542. service_dict['build'] = service_dict.get('build', {})
  543. service_dict['build'].update({
  544. 'dockerfile': service_dict.pop('dockerfile')
  545. })
  546. return service_dict
  547. def merge_service_dicts_from_files(base, override, version):
  548. """When merging services from multiple files we need to merge the `extends`
  549. field. This is not handled by `merge_service_dicts()` which is used to
  550. perform the `extends`.
  551. """
  552. new_service = merge_service_dicts(base, override, version)
  553. if 'extends' in override:
  554. new_service['extends'] = override['extends']
  555. elif 'extends' in base:
  556. new_service['extends'] = base['extends']
  557. return new_service
  558. class MergeDict(dict):
  559. """A dict-like object responsible for merging two dicts into one."""
  560. def __init__(self, base, override):
  561. self.base = base
  562. self.override = override
  563. def needs_merge(self, field):
  564. return field in self.base or field in self.override
  565. def merge_field(self, field, merge_func, default=None):
  566. if not self.needs_merge(field):
  567. return
  568. self[field] = merge_func(
  569. self.base.get(field, default),
  570. self.override.get(field, default))
  571. def merge_mapping(self, field, parse_func):
  572. if not self.needs_merge(field):
  573. return
  574. self[field] = parse_func(self.base.get(field))
  575. self[field].update(parse_func(self.override.get(field)))
  576. def merge_sequence(self, field, parse_func):
  577. def parse_sequence_func(seq):
  578. return to_mapping((parse_func(item) for item in seq), 'merge_field')
  579. if not self.needs_merge(field):
  580. return
  581. merged = parse_sequence_func(self.base.get(field, []))
  582. merged.update(parse_sequence_func(self.override.get(field, [])))
  583. self[field] = [item.repr() for item in sorted(merged.values())]
  584. def merge_scalar(self, field):
  585. if self.needs_merge(field):
  586. self[field] = self.override.get(field, self.base.get(field))
  587. def merge_service_dicts(base, override, version):
  588. md = MergeDict(base, override)
  589. md.merge_mapping('environment', parse_environment)
  590. md.merge_mapping('labels', parse_labels)
  591. md.merge_mapping('ulimits', parse_ulimits)
  592. md.merge_mapping('networks', parse_networks)
  593. md.merge_sequence('links', ServiceLink.parse)
  594. for field in ['volumes', 'devices']:
  595. md.merge_field(field, merge_path_mappings)
  596. for field in [
  597. 'ports', 'cap_add', 'cap_drop', 'expose', 'external_links',
  598. 'security_opt', 'volumes_from', 'depends_on',
  599. ]:
  600. md.merge_field(field, merge_unique_items_lists, default=[])
  601. for field in ['dns', 'dns_search', 'env_file', 'tmpfs']:
  602. md.merge_field(field, merge_list_or_string)
  603. for field in set(ALLOWED_KEYS) - set(md):
  604. md.merge_scalar(field)
  605. if version == V1:
  606. legacy_v1_merge_image_or_build(md, base, override)
  607. elif md.needs_merge('build'):
  608. md['build'] = merge_build(md, base, override)
  609. return dict(md)
  610. def merge_unique_items_lists(base, override):
  611. return sorted(set().union(base, override))
  612. def merge_build(output, base, override):
  613. def to_dict(service):
  614. build_config = service.get('build', {})
  615. if isinstance(build_config, six.string_types):
  616. return {'context': build_config}
  617. return build_config
  618. md = MergeDict(to_dict(base), to_dict(override))
  619. md.merge_scalar('context')
  620. md.merge_scalar('dockerfile')
  621. md.merge_mapping('args', parse_build_arguments)
  622. return dict(md)
  623. def legacy_v1_merge_image_or_build(output, base, override):
  624. output.pop('image', None)
  625. output.pop('build', None)
  626. if 'image' in override:
  627. output['image'] = override['image']
  628. elif 'build' in override:
  629. output['build'] = override['build']
  630. elif 'image' in base:
  631. output['image'] = base['image']
  632. elif 'build' in base:
  633. output['build'] = base['build']
  634. def merge_environment(base, override):
  635. env = parse_environment(base)
  636. env.update(parse_environment(override))
  637. return env
  638. def split_label(label):
  639. if '=' in label:
  640. return label.split('=', 1)
  641. else:
  642. return label, ''
  643. def parse_dict_or_list(split_func, type_name, arguments):
  644. if not arguments:
  645. return {}
  646. if isinstance(arguments, list):
  647. return dict(split_func(e) for e in arguments)
  648. if isinstance(arguments, dict):
  649. return dict(arguments)
  650. raise ConfigurationError(
  651. "%s \"%s\" must be a list or mapping," %
  652. (type_name, arguments)
  653. )
  654. parse_build_arguments = functools.partial(parse_dict_or_list, split_env, 'build arguments')
  655. parse_environment = functools.partial(parse_dict_or_list, split_env, 'environment')
  656. parse_labels = functools.partial(parse_dict_or_list, split_label, 'labels')
  657. parse_networks = functools.partial(parse_dict_or_list, lambda k: (k, None), 'networks')
  658. def parse_ulimits(ulimits):
  659. if not ulimits:
  660. return {}
  661. if isinstance(ulimits, dict):
  662. return dict(ulimits)
  663. def resolve_env_var(key, val, environment):
  664. if val is not None:
  665. return key, val
  666. elif environment and key in environment:
  667. return key, environment[key]
  668. else:
  669. return key, None
  670. def resolve_volume_paths(working_dir, service_dict):
  671. return [
  672. resolve_volume_path(working_dir, volume)
  673. for volume in service_dict['volumes']
  674. ]
  675. def resolve_volume_path(working_dir, volume):
  676. container_path, host_path = split_path_mapping(volume)
  677. if host_path is not None:
  678. if host_path.startswith('.'):
  679. host_path = expand_path(working_dir, host_path)
  680. host_path = os.path.expanduser(host_path)
  681. return u"{}:{}".format(host_path, container_path)
  682. else:
  683. return container_path
  684. def normalize_build(service_dict, working_dir, environment):
  685. if 'build' in service_dict:
  686. build = {}
  687. # Shortcut where specifying a string is treated as the build context
  688. if isinstance(service_dict['build'], six.string_types):
  689. build['context'] = service_dict.pop('build')
  690. else:
  691. build.update(service_dict['build'])
  692. if 'args' in build:
  693. build['args'] = build_string_dict(
  694. resolve_build_args(build, environment)
  695. )
  696. service_dict['build'] = build
  697. def resolve_build_path(working_dir, build_path):
  698. if is_url(build_path):
  699. return build_path
  700. return expand_path(working_dir, build_path)
  701. def is_url(build_path):
  702. return build_path.startswith(DOCKER_VALID_URL_PREFIXES)
  703. def validate_paths(service_dict):
  704. if 'build' in service_dict:
  705. build = service_dict.get('build', {})
  706. if isinstance(build, six.string_types):
  707. build_path = build
  708. elif isinstance(build, dict) and 'context' in build:
  709. build_path = build['context']
  710. else:
  711. # We have a build section but no context, so nothing to validate
  712. return
  713. if (
  714. not is_url(build_path) and
  715. (not os.path.exists(build_path) or not os.access(build_path, os.R_OK))
  716. ):
  717. raise ConfigurationError(
  718. "build path %s either does not exist, is not accessible, "
  719. "or is not a valid URL." % build_path)
  720. def merge_path_mappings(base, override):
  721. d = dict_from_path_mappings(base)
  722. d.update(dict_from_path_mappings(override))
  723. return path_mappings_from_dict(d)
  724. def dict_from_path_mappings(path_mappings):
  725. if path_mappings:
  726. return dict(split_path_mapping(v) for v in path_mappings)
  727. else:
  728. return {}
  729. def path_mappings_from_dict(d):
  730. return [join_path_mapping(v) for v in sorted(d.items())]
  731. def split_path_mapping(volume_path):
  732. """
  733. Ascertain if the volume_path contains a host path as well as a container
  734. path. Using splitdrive so windows absolute paths won't cause issues with
  735. splitting on ':'.
  736. """
  737. drive, volume_config = splitdrive(volume_path)
  738. if ':' in volume_config:
  739. (host, container) = volume_config.split(':', 1)
  740. return (container, drive + host)
  741. else:
  742. return (volume_path, None)
  743. def join_path_mapping(pair):
  744. (container, host) = pair
  745. if host is None:
  746. return container
  747. else:
  748. return ":".join((host, container))
  749. def expand_path(working_dir, path):
  750. return os.path.abspath(os.path.join(working_dir, os.path.expanduser(path)))
  751. def merge_list_or_string(base, override):
  752. return to_list(base) + to_list(override)
  753. def to_list(value):
  754. if value is None:
  755. return []
  756. elif isinstance(value, six.string_types):
  757. return [value]
  758. else:
  759. return value
  760. def to_mapping(sequence, key_field):
  761. return {getattr(item, key_field): item for item in sequence}
  762. def has_uppercase(name):
  763. return any(char in string.ascii_uppercase for char in name)
  764. def load_yaml(filename):
  765. try:
  766. with open(filename, 'r') as fh:
  767. return yaml.safe_load(fh)
  768. except (IOError, yaml.YAMLError) as e:
  769. error_name = getattr(e, '__module__', '') + '.' + e.__class__.__name__
  770. raise ConfigurationError(u"{}: {}".format(error_name, e))