config.py 32 KB

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