config.py 31 KB

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