Rewrite the docker_prune module (#410)

* Rewrite the docker_prune module.

* Improve error messages.
This commit is contained in:
Felix Fontein 2022-07-06 21:47:43 +02:00 committed by GitHub
parent 18fdd04782
commit e60ce69102
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 25 additions and 37 deletions

View File

@ -0,0 +1,4 @@
major_changes:
- "docker_prune - no longer uses the Docker SDK for Python. It requires ``requests`` to be installed,
and depending on the features used has some more requirements. If the Docker SDK for Python is installed,
these requirements are likely met (https://github.com/ansible-collections/community.docker/pull/410)."

View File

@ -68,20 +68,17 @@ options:
builder_cache: builder_cache:
description: description:
- Whether to prune the builder cache. - Whether to prune the builder cache.
- Requires version 3.3.0 of the Docker SDK for Python or newer.
type: bool type: bool
default: no default: no
extends_documentation_fragment: extends_documentation_fragment:
- community.docker.docker - community.docker.docker.api_documentation
- community.docker.docker.docker_py_2_documentation
author: author:
- "Felix Fontein (@felixfontein)" - "Felix Fontein (@felixfontein)"
requirements: requirements:
- "L(Docker SDK for Python,https://docker-py.readthedocs.io/en/stable/) >= 2.1.0"
- "Docker API >= 1.25" - "Docker API >= 1.25"
''' '''
@ -180,24 +177,15 @@ import traceback
from ansible.module_utils.common.text.converters import to_native from ansible.module_utils.common.text.converters import to_native
try: from ansible_collections.community.docker.plugins.module_utils.common_api import (
from docker.errors import DockerException
except ImportError:
# missing Docker SDK for Python handled in ansible.module_utils.docker.common
pass
from ansible_collections.community.docker.plugins.module_utils.version import LooseVersion
from ansible_collections.community.docker.plugins.module_utils.common import (
AnsibleDockerClient, AnsibleDockerClient,
RequestException, RequestException,
) )
try: from ansible_collections.community.docker.plugins.module_utils.util import clean_dict_booleans_for_docker_api
from ansible_collections.community.docker.plugins.module_utils.common import docker_version, clean_dict_booleans_for_docker_api
except Exception as dummy: from ansible_collections.community.docker.plugins.module_utils._api.errors import DockerException
# missing Docker SDK for Python handled in ansible.module_utils.docker.common from ansible_collections.community.docker.plugins.module_utils._api.utils.utils import convert_filters
pass
def main(): def main():
@ -216,51 +204,48 @@ def main():
client = AnsibleDockerClient( client = AnsibleDockerClient(
argument_spec=argument_spec, argument_spec=argument_spec,
# supports_check_mode=True, # supports_check_mode=True,
min_docker_version='2.1.0',
) )
# Version checks
cache_min_version = '3.3.0'
if client.module.params['builder_cache'] and client.docker_py_version < LooseVersion(cache_min_version):
msg = "Error: Docker SDK for Python's version is %s. Minimum version required for builds option is %s. Use `pip install --upgrade docker` to upgrade."
client.fail(msg % (docker_version, cache_min_version))
try: try:
result = dict() result = dict()
if client.module.params['containers']: if client.module.params['containers']:
filters = clean_dict_booleans_for_docker_api(client.module.params.get('containers_filters')) filters = clean_dict_booleans_for_docker_api(client.module.params.get('containers_filters'))
res = client.prune_containers(filters=filters) params = {'filters': convert_filters(filters)}
res = client.post_to_json('/containers/prune', params=params)
result['containers'] = res.get('ContainersDeleted') or [] result['containers'] = res.get('ContainersDeleted') or []
result['containers_space_reclaimed'] = res['SpaceReclaimed'] result['containers_space_reclaimed'] = res['SpaceReclaimed']
if client.module.params['images']: if client.module.params['images']:
filters = clean_dict_booleans_for_docker_api(client.module.params.get('images_filters')) filters = clean_dict_booleans_for_docker_api(client.module.params.get('images_filters'))
res = client.prune_images(filters=filters) params = {'filters': convert_filters(filters)}
res = client.post_to_json('/images/prune', params=params)
result['images'] = res.get('ImagesDeleted') or [] result['images'] = res.get('ImagesDeleted') or []
result['images_space_reclaimed'] = res['SpaceReclaimed'] result['images_space_reclaimed'] = res['SpaceReclaimed']
if client.module.params['networks']: if client.module.params['networks']:
filters = clean_dict_booleans_for_docker_api(client.module.params.get('networks_filters')) filters = clean_dict_booleans_for_docker_api(client.module.params.get('networks_filters'))
res = client.prune_networks(filters=filters) params = {'filters': convert_filters(filters)}
res = client.post_to_json('/networks/prune', params=params)
result['networks'] = res.get('NetworksDeleted') or [] result['networks'] = res.get('NetworksDeleted') or []
if client.module.params['volumes']: if client.module.params['volumes']:
filters = clean_dict_booleans_for_docker_api(client.module.params.get('volumes_filters')) filters = clean_dict_booleans_for_docker_api(client.module.params.get('volumes_filters'))
res = client.prune_volumes(filters=filters) params = {'filters': convert_filters(filters)}
res = client.post_to_json('/volumes/prune', params=params)
result['volumes'] = res.get('VolumesDeleted') or [] result['volumes'] = res.get('VolumesDeleted') or []
result['volumes_space_reclaimed'] = res['SpaceReclaimed'] result['volumes_space_reclaimed'] = res['SpaceReclaimed']
if client.module.params['builder_cache']: if client.module.params['builder_cache']:
res = client.prune_builds() res = client.post_to_json('/build/prune')
result['builder_cache_space_reclaimed'] = res['SpaceReclaimed'] result['builder_cache_space_reclaimed'] = res['SpaceReclaimed']
client.module.exit_json(**result) client.module.exit_json(**result)
except DockerException as e: except DockerException as e:
client.fail('An unexpected docker error occurred: {0}'.format(to_native(e)), exception=traceback.format_exc()) client.fail('An unexpected Docker error occurred: {0}'.format(to_native(e)), exception=traceback.format_exc())
except RequestException as e: except RequestException as e:
client.fail( client.fail(
'An unexpected requests error occurred when Docker SDK for Python tried to talk to the docker daemon: {0}'.format(to_native(e)), 'An unexpected requests error occurred when trying to talk to the Docker daemon: {0}'.format(to_native(e)),
exception=traceback.format_exc()) exception=traceback.format_exc())

View File

@ -32,7 +32,7 @@
images: yes images: yes
networks: yes networks: yes
volumes: yes volumes: yes
builder_cache: "{{ docker_py_version is version('3.3.0', '>=') }}" builder_cache: yes
register: result register: result
# Analyze result # Analyze result
@ -50,8 +50,7 @@
- volume.volume.Name in result.volumes - volume.volume.Name in result.volumes
- "'volumes_space_reclaimed' in result" - "'volumes_space_reclaimed' in result"
# builder_cache # builder_cache
- "'builder_cache_space_reclaimed' in result or docker_py_version is version('3.3.0', '<')" - "'builder_cache_space_reclaimed' in result"
- "'builder_cache_space_reclaimed' not in result or docker_py_version is version('3.3.0', '>=')"
# Test with filters # Test with filters
- docker_prune: - docker_prune:
@ -62,7 +61,7 @@
- debug: var=result - debug: var=result
when: docker_py_version is version('2.1.0', '>=') and docker_api_version is version('1.25', '>=') when: docker_api_version is version('1.25', '>=')
- fail: msg="Too old docker / docker-py version to run docker_prune tests!" - fail: msg="Too old docker / docker-py version to run docker_prune tests!"
when: not(docker_py_version is version('2.1.0', '>=') and docker_api_version is version('1.25', '>=')) and (ansible_distribution != 'CentOS' or ansible_distribution_major_version|int > 6) when: not(docker_api_version is version('1.25', '>=')) and (ansible_distribution != 'CentOS' or ansible_distribution_major_version|int > 6)