command.py 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
  1. from __future__ import absolute_import
  2. from __future__ import unicode_literals
  3. import logging
  4. import os
  5. import re
  6. import six
  7. from . import errors
  8. from . import verbose_proxy
  9. from .. import config
  10. from .. import parallel
  11. from ..config.environment import Environment
  12. from ..const import API_VERSIONS
  13. from ..project import Project
  14. from .docker_client import docker_client
  15. from .docker_client import get_tls_version
  16. from .docker_client import tls_config_from_options
  17. from .utils import get_version_info
  18. log = logging.getLogger(__name__)
  19. def project_from_options(project_dir, options):
  20. environment = Environment.from_env_file(project_dir)
  21. set_parallel_limit(environment)
  22. host = options.get('--host')
  23. if host is not None:
  24. host = host.lstrip('=')
  25. return get_project(
  26. project_dir,
  27. get_config_path_from_options(project_dir, options, environment),
  28. project_name=options.get('--project-name'),
  29. verbose=options.get('--verbose'),
  30. host=host,
  31. tls_config=tls_config_from_options(options, environment),
  32. environment=environment,
  33. override_dir=options.get('--project-directory'),
  34. compatibility=options.get('--compatibility'),
  35. )
  36. def set_parallel_limit(environment):
  37. parallel_limit = environment.get('COMPOSE_PARALLEL_LIMIT')
  38. if parallel_limit:
  39. try:
  40. parallel_limit = int(parallel_limit)
  41. except ValueError:
  42. raise errors.UserError(
  43. 'COMPOSE_PARALLEL_LIMIT must be an integer (found: "{}")'.format(
  44. environment.get('COMPOSE_PARALLEL_LIMIT')
  45. )
  46. )
  47. if parallel_limit <= 1:
  48. raise errors.UserError('COMPOSE_PARALLEL_LIMIT can not be less than 2')
  49. parallel.GlobalLimit.set_global_limit(parallel_limit)
  50. def get_config_from_options(base_dir, options):
  51. environment = Environment.from_env_file(base_dir)
  52. config_path = get_config_path_from_options(
  53. base_dir, options, environment
  54. )
  55. return config.load(
  56. config.find(base_dir, config_path, environment),
  57. options.get('--compatibility')
  58. )
  59. def get_config_path_from_options(base_dir, options, environment):
  60. def unicode_paths(paths):
  61. return [p.decode('utf-8') if isinstance(p, six.binary_type) else p for p in paths]
  62. file_option = options.get('--file')
  63. if file_option:
  64. return unicode_paths(file_option)
  65. config_files = environment.get('COMPOSE_FILE')
  66. if config_files:
  67. pathsep = environment.get('COMPOSE_PATH_SEPARATOR', os.pathsep)
  68. return unicode_paths(config_files.split(pathsep))
  69. return None
  70. def get_client(environment, verbose=False, version=None, tls_config=None, host=None,
  71. tls_version=None):
  72. client = docker_client(
  73. version=version, tls_config=tls_config, host=host,
  74. environment=environment, tls_version=get_tls_version(environment)
  75. )
  76. if verbose:
  77. version_info = six.iteritems(client.version())
  78. log.info(get_version_info('full'))
  79. log.info("Docker base_url: %s", client.base_url)
  80. log.info("Docker version: %s",
  81. ", ".join("%s=%s" % item for item in version_info))
  82. return verbose_proxy.VerboseProxy('docker', client)
  83. return client
  84. def get_project(project_dir, config_path=None, project_name=None, verbose=False,
  85. host=None, tls_config=None, environment=None, override_dir=None,
  86. compatibility=False):
  87. if not environment:
  88. environment = Environment.from_env_file(project_dir)
  89. config_details = config.find(project_dir, config_path, environment, override_dir)
  90. project_name = get_project_name(
  91. config_details.working_dir, project_name, environment
  92. )
  93. config_data = config.load(config_details, compatibility)
  94. api_version = environment.get(
  95. 'COMPOSE_API_VERSION',
  96. API_VERSIONS[config_data.version])
  97. client = get_client(
  98. verbose=verbose, version=api_version, tls_config=tls_config,
  99. host=host, environment=environment
  100. )
  101. with errors.handle_connection_errors(client):
  102. return Project.from_config(
  103. project_name, config_data, client, environment.get('DOCKER_DEFAULT_PLATFORM')
  104. )
  105. def get_project_name(working_dir, project_name=None, environment=None):
  106. def normalize_name(name):
  107. return re.sub(r'[^-_a-z0-9]', '', name.lower())
  108. if not environment:
  109. environment = Environment.from_env_file(working_dir)
  110. project_name = project_name or environment.get('COMPOSE_PROJECT_NAME')
  111. if project_name:
  112. return normalize_name(project_name)
  113. project = os.path.basename(os.path.abspath(working_dir))
  114. if project:
  115. return normalize_name(project)
  116. return 'default'