[docker] Add compatibility with docker-py v2.0

lib/ansible/module_utils/docker_common.py

@@ -27,17 +27,24 @@ from urlparse import urlparse
 from ansible.module_utils.basic import *
 
 HAS_DOCKER_PY = True
+HAS_DOCKER_PY_2 = False
 HAS_DOCKER_ERROR = None
 
 try:
     from requests.exceptions import SSLError
-    from docker import Client
     from docker import __version__ as docker_version
     from docker.errors import APIError, TLSParameterError, NotFound
     from docker.tls import TLSConfig
     from docker.constants import DEFAULT_TIMEOUT_SECONDS, DEFAULT_DOCKER_API_VERSION
-    from docker.utils.types import Ulimit, LogConfig
     from docker import auth
+    if LooseVersion(docker_version) >= LooseVersion('2.0.0'):
+        HAS_DOCKER_PY_2 = True
+        from docker import APIClient as Client
+        from docker.types import Ulimit, LogConfig
+    else:
+        from docker import Client
+        from docker.utils.types import Ulimit, LogConfig
+
 except ImportError as exc:
     HAS_DOCKER_ERROR = str(exc)
     HAS_DOCKER_PY = False

lib/ansible/modules/cloud/docker/docker_container.py

@@ -656,7 +656,10 @@ from ansible.module_utils.docker_common import *
 
 try:
     from docker import utils
-    from docker.utils.types import Ulimit
+    if HAS_DOCKER_PY_2:
+        from docker.types import Ulimit
+    else:
+        from docker.utils.types import Ulimit
 except:
     # missing docker-py handled in ansible.module_utils.docker
     pass

lib/ansible/modules/cloud/docker/docker_image.py

@@ -250,7 +250,10 @@ image:
 from ansible.module_utils.docker_common import *
 
 try:
-    from docker.auth.auth import resolve_repository_name
+    if HAS_DOCKER_PY_2:
+        from docker.auth import resolve_repository_name
+    else:
+        from docker.auth.auth import resolve_repository_name
     from docker.utils.utils import parse_repository_tag
 except ImportError:
     # missing docker-py handled in docker_common

lib/ansible/modules/cloud/docker/docker_network.py

@@ -164,7 +164,10 @@ from ansible.module_utils.docker_common import *
 
 try:
     from docker import utils
-    from docker.utils.types import Ulimit
+    if HAS_DOCKER_PY_2:
+        from docker.types import Ulimit, IPAMPool, IPAMConfig
+    else:
+        from docker.utils.types import Ulimit
 except:
     # missing docker-py handled in ansible.module_utils.docker
     pass
@@ -275,9 +278,16 @@ class DockerNetworkManager(object):
         if not self.existing_network:
             ipam_pools = []
             if self.parameters.ipam_options:
-                ipam_pools.append(utils.create_ipam_pool(**self.parameters.ipam_options))
+                if HAS_DOCKER_PY_2:
+                    ipam_pools.append(IPAMPool(**self.parameters.ipam_options))
+                else:
+                    ipam_pools.append(utils.create_ipam_pool(**self.parameters.ipam_options))
 
-            ipam_config = utils.create_ipam_config(driver=self.parameters.ipam_driver,
+            if HAS_DOCKER_PY_2:
+                ipam_config = IPAMConfig(driver=self.parameters.ipam_driver,
+                                                   pool_configs=ipam_pools)
+            else:
+                ipam_config = utils.create_ipam_config(driver=self.parameters.ipam_driver,
                                                    pool_configs=ipam_pools)
 
             if not self.check_mode: