from __future__ import absolute_import from __future__ import unicode_literals import logging import os import re import six from requests.exceptions import ConnectionError from requests.exceptions import SSLError from . import errors from . import verbose_proxy from .. import __version__ from .. import config from ..project import Project from ..service import ConfigError from .docker_client import docker_client from .docopt_command import DocoptCommand from .utils import call_silently from .utils import is_mac from .utils import is_ubuntu log = logging.getLogger(__name__) class Command(DocoptCommand): base_dir = '.' def dispatch(self, *args, **kwargs): try: super(Command, self).dispatch(*args, **kwargs) except SSLError as e: raise errors.UserError('SSL error: %s' % e) except ConnectionError: if call_silently(['which', 'docker']) != 0: if is_mac(): raise errors.DockerNotFoundMac() elif is_ubuntu(): raise errors.DockerNotFoundUbuntu() else: raise errors.DockerNotFoundGeneric() elif call_silently(['which', 'boot2docker']) == 0: raise errors.ConnectionErrorBoot2Docker() else: raise errors.ConnectionErrorGeneric(self.get_client().base_url) def perform_command(self, options, handler, command_options): if options['COMMAND'] in ('help', 'version'): # Skip looking up the compose file. handler(None, command_options) return if 'FIG_FILE' in os.environ: log.warn('The FIG_FILE environment variable is deprecated.') log.warn('Please use COMPOSE_FILE instead.') explicit_config_path = options.get('--file') or os.environ.get('COMPOSE_FILE') or os.environ.get('FIG_FILE') project = self.get_project( explicit_config_path, project_name=options.get('--project-name'), verbose=options.get('--verbose')) handler(project, command_options) def get_client(self, verbose=False): client = docker_client() if verbose: version_info = six.iteritems(client.version()) log.info("Compose version %s", __version__) 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(self, config_path=None, project_name=None, verbose=False): config_details = config.find(self.base_dir, config_path) try: return Project.from_dicts( self.get_project_name(config_details.working_dir, project_name), config.load(config_details), self.get_client(verbose=verbose)) except ConfigError as e: raise errors.UserError(six.text_type(e)) def get_project_name(self, working_dir, project_name=None): def normalize_name(name): return re.sub(r'[^a-z0-9]', '', name.lower()) if 'FIG_PROJECT_NAME' in os.environ: log.warn('The FIG_PROJECT_NAME environment variable is deprecated.') log.warn('Please use COMPOSE_PROJECT_NAME instead.') project_name = ( project_name or os.environ.get('COMPOSE_PROJECT_NAME') or os.environ.get('FIG_PROJECT_NAME')) if project_name is not None: return normalize_name(project_name) project = os.path.basename(os.path.abspath(working_dir)) if project: return normalize_name(project) return 'default'