diff --git a/docker_custodian/docker_autostop.py b/docker_custodian/docker_autostop.py index 021f60e..98d0df8 100644 --- a/docker_custodian/docker_autostop.py +++ b/docker_custodian/docker_autostop.py @@ -13,6 +13,7 @@ import docker.errors import requests.exceptions from docker_custodian.args import timedelta_type +from docker.utils import kwargs_from_env log = logging.getLogger(__name__) @@ -66,7 +67,9 @@ def main(): stream=sys.stdout) opts = get_opts() - client = docker.Client(version='auto', timeout=opts.timeout) + client = docker.APIClient(version='auto', + timeout=opts.timeout, + **kwargs_from_env()) matcher = build_container_matcher(opts.prefix) stop_containers(client, opts.max_run_time, matcher, opts.dry_run) diff --git a/docker_custodian/docker_gc.py b/docker_custodian/docker_gc.py index db7aba9..51d0881 100644 --- a/docker_custodian/docker_gc.py +++ b/docker_custodian/docker_gc.py @@ -14,6 +14,7 @@ import docker.errors import requests.exceptions from docker_custodian.args import timedelta_type +from docker.utils import kwargs_from_env log = logging.getLogger(__name__) @@ -220,7 +221,9 @@ def main(): stream=sys.stdout) args = get_args() - client = docker.Client(version='auto', timeout=args.timeout) + client = docker.APIClient(version='auto', + timeout=args.timeout, + **kwargs_from_env()) if args.max_container_age: cleanup_containers(client, args.max_container_age, args.dry_run) diff --git a/requirements.txt b/requirements.txt index 35faae8..884a9e4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,14 @@ -argparse==1.3.0 -backports.ssl-match-hostname==3.4.0.2 -docker-py==1.7.2 -future==0.14.3 -python-dateutil==2.4.0 -pytimeparse==1.1.2 -requests==2.7.0 -six==1.9.0 -websocket-client==0.32.0 +backports.ssl-match-hostname==3.5.0.1 +certifi==2018.1.18 +chardet==3.0.4 +docker==3.1.0 +docker-pycreds==0.2.2 +future==0.16.0 +idna==2.6 +ipaddress==1.0.19 +python-dateutil==2.6.1 +pytimeparse==1.1.7 +requests==2.18.4 +six==1.11.0 +urllib3==1.22 +websocket-client==0.47.0 diff --git a/setup.py b/setup.py index 556e8aa..7873542 100644 --- a/setup.py +++ b/setup.py @@ -14,7 +14,7 @@ setup( include_package_data=True, install_requires=[ 'python-dateutil', - 'docker-py >= 0.5', + 'docker', 'pytimeparse', ], license="Apache License 2.0", diff --git a/tests/conftest.py b/tests/conftest.py index 2fbc83c..b891cc2 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -48,6 +48,6 @@ def later_time(): @pytest.fixture def mock_client(): - client = mock.create_autospec(docker.Client) - client._version = '1.17' + client = mock.create_autospec(docker.APIClient) + client._version = '1.21' return client diff --git a/tests/docker_gc_test.py b/tests/docker_gc_test.py index b980181..54cbe6f 100644 --- a/tests/docker_gc_test.py +++ b/tests/docker_gc_test.py @@ -466,7 +466,7 @@ def test_build_exclude_set_empty(): def test_main(mock_client): with mock.patch( - 'docker_custodian.docker_gc.docker.Client', + 'docker_custodian.docker_gc.docker.APIClient', return_value=mock_client): with mock.patch(