123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144 |
- from __future__ import absolute_import
- from __future__ import unicode_literals
- import logging
- import os
- import re
- import six
- from . import errors
- from . import verbose_proxy
- from .. import config
- from .. import parallel
- from ..config.environment import Environment
- from ..const import API_VERSIONS
- from ..project import Project
- from .docker_client import docker_client
- from .docker_client import get_tls_version
- from .docker_client import tls_config_from_options
- from .utils import get_version_info
- log = logging.getLogger(__name__)
- def project_from_options(project_dir, options):
- environment = Environment.from_env_file(project_dir)
- set_parallel_limit(environment)
- host = options.get('--host')
- if host is not None:
- host = host.lstrip('=')
- return get_project(
- project_dir,
- get_config_path_from_options(project_dir, options, environment),
- project_name=options.get('--project-name'),
- verbose=options.get('--verbose'),
- host=host,
- tls_config=tls_config_from_options(options, environment),
- environment=environment,
- override_dir=options.get('--project-directory'),
- compatibility=options.get('--compatibility'),
- )
- def set_parallel_limit(environment):
- parallel_limit = environment.get('COMPOSE_PARALLEL_LIMIT')
- if parallel_limit:
- try:
- parallel_limit = int(parallel_limit)
- except ValueError:
- raise errors.UserError(
- 'COMPOSE_PARALLEL_LIMIT must be an integer (found: "{}")'.format(
- environment.get('COMPOSE_PARALLEL_LIMIT')
- )
- )
- if parallel_limit <= 1:
- raise errors.UserError('COMPOSE_PARALLEL_LIMIT can not be less than 2')
- parallel.GlobalLimit.set_global_limit(parallel_limit)
- def get_config_from_options(base_dir, options):
- environment = Environment.from_env_file(base_dir)
- config_path = get_config_path_from_options(
- base_dir, options, environment
- )
- return config.load(
- config.find(base_dir, config_path, environment),
- options.get('--compatibility')
- )
- def get_config_path_from_options(base_dir, options, environment):
- def unicode_paths(paths):
- return [p.decode('utf-8') if isinstance(p, six.binary_type) else p for p in paths]
- file_option = options.get('--file')
- if file_option:
- return unicode_paths(file_option)
- config_files = environment.get('COMPOSE_FILE')
- if config_files:
- pathsep = environment.get('COMPOSE_PATH_SEPARATOR', os.pathsep)
- return unicode_paths(config_files.split(pathsep))
- return None
- def get_client(environment, verbose=False, version=None, tls_config=None, host=None,
- tls_version=None):
- client = docker_client(
- version=version, tls_config=tls_config, host=host,
- environment=environment, tls_version=get_tls_version(environment)
- )
- if verbose:
- version_info = six.iteritems(client.version())
- log.info(get_version_info('full'))
- log.info("Docker base_url: %s", client.base_url)
- log.info("Docker version: %s",
- ", ".join("%s=%s" % item for item in version_info))
- return verbose_proxy.VerboseProxy('docker', client)
- return client
- def get_project(project_dir, config_path=None, project_name=None, verbose=False,
- host=None, tls_config=None, environment=None, override_dir=None,
- compatibility=False):
- if not environment:
- environment = Environment.from_env_file(project_dir)
- config_details = config.find(project_dir, config_path, environment, override_dir)
- project_name = get_project_name(
- config_details.working_dir, project_name, environment
- )
- config_data = config.load(config_details, compatibility)
- api_version = environment.get(
- 'COMPOSE_API_VERSION',
- API_VERSIONS[config_data.version])
- client = get_client(
- verbose=verbose, version=api_version, tls_config=tls_config,
- host=host, environment=environment
- )
- with errors.handle_connection_errors(client):
- return Project.from_config(
- project_name, config_data, client, environment.get('DOCKER_DEFAULT_PLATFORM')
- )
- def get_project_name(working_dir, project_name=None, environment=None):
- def normalize_name(name):
- return re.sub(r'[^-_a-z0-9]', '', name.lower())
- if not environment:
- environment = Environment.from_env_file(working_dir)
- project_name = project_name or environment.get('COMPOSE_PROJECT_NAME')
- if project_name:
- return normalize_name(project_name)
- project = os.path.basename(os.path.abspath(working_dir))
- if project:
- return normalize_name(project)
- return 'default'
|