docker_client.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435
  1. from docker import Client
  2. from docker import tls
  3. import ssl
  4. import os
  5. def docker_client():
  6. """
  7. Returns a docker-py client configured using environment variables
  8. according to the same logic as the official Docker client.
  9. """
  10. cert_path = os.environ.get('DOCKER_CERT_PATH', '')
  11. if cert_path == '':
  12. cert_path = os.path.join(os.environ.get('HOME', ''), '.docker')
  13. base_url = os.environ.get('DOCKER_HOST')
  14. tls_config = None
  15. if os.environ.get('DOCKER_TLS_VERIFY', '') != '':
  16. parts = base_url.split('://', 1)
  17. base_url = '%s://%s' % ('https', parts[1])
  18. client_cert = (os.path.join(cert_path, 'cert.pem'), os.path.join(cert_path, 'key.pem'))
  19. ca_cert = os.path.join(cert_path, 'ca.pem')
  20. tls_config = tls.TLSConfig(
  21. ssl_version=ssl.PROTOCOL_TLSv1,
  22. verify=True,
  23. assert_hostname=False,
  24. client_cert=client_cert,
  25. ca_cert=ca_cert,
  26. )
  27. timeout = int(os.environ.get('DOCKER_CLIENT_TIMEOUT', 60))
  28. return Client(base_url=base_url, tls=tls_config, version='1.15', timeout=timeout)