# Buggy code
def main():
    argument_spec = dict(
        containers=dict(type="bool", default=False),
        containers_filters=dict(type="dict"),
        images=dict(type="bool", default=False),
        images_filters=dict(type="dict"),
        networks=dict(type="bool", default=False),
        networks_filters=dict(type="dict"),
        volumes=dict(type="bool", default=False),
        volumes_filters=dict(type="dict"),
        builder_cache=dict(type="bool", default=False),
    )

    client = AnsibleDockerClient(
        argument_spec=argument_spec,
        # supports_check_mode=True,
        min_docker_api_version="1.25",
        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:
        result = dict()

        if client.module.params["containers"]:
            filters = clean_dict_booleans_for_docker_api(
                client.module.params.get("containers_filters")
            )
            res = client.prune_containers(filters=filters)
            result["containers"] = res.get("ContainersDeleted") or []
            result["containers_space_reclaimed"] = res["SpaceReclaimed"]

        if client.module.params["images"]:
            filters = clean_dict_booleans_for_docker_api(
                client.module.params.get("images_filters")
            )
            res = client.prune_images(filters=filters)
            result["images"] = res.get("ImagesDeleted") or []
            result["images_space_reclaimed"] = res["SpaceReclaimed"]

        if client.module.params["networks"]:
            filters = clean_dict_booleans_for_docker_api(
                client.module.params.get("networks_filters")
            )
            res = client.prune_networks(filters=filters)
            result["networks"] = res.get("NetworksDeleted") or []

        if client.module.params["volumes"]:
            filters = clean_dict_booleans_for_docker_api(
                client.module.params.get("volumes_filters")
            )
            res = client.prune_volumes(filters=filters)
            result["volumes"] = res.get("VolumesDeleted") or []
            result["volumes_space_reclaimed"] = res["SpaceReclaimed"]

        if client.module.params["builder_cache"]:
            res = client.prune_builds()
            result["builder_cache_space_reclaimed"] = res["SpaceReclaimed"]

        client.module.exit_json(**result)
    except DockerException as e:
        client.fail(
            "An unexpected docker error occurred: {0}".format(e),
            exception=traceback.format_exc(),
        )


# Traceback (requests.exceptions.ConnectionError)
An exception occurred during task execution. To see the full traceback, use -vvv. The error was: requests.exceptions.ConnectionError: ('Connection aborted.', error(104, 'Connection reset by peer'))
fatal: [buster_local_docker]: FAILED! => changed=false
module_stderr: |-
Traceback (most recent call last):
File "/tmp/.ansible/tmp/ansible-tmp-1562335690.251731-79429560017149/AnsiballZ_docker_image.py", line 114, in <module>
_ansiballz_main()
File "/tmp/.ansible/tmp/ansible-tmp-1562335690.251731-79429560017149/AnsiballZ_docker_image.py", line 106, in _ansiballz_main
invoke_module(zipped_mod, temp_path, ANSIBALLZ_PARAMS)
File "/tmp/.ansible/tmp/ansible-tmp-1562335690.251731-79429560017149/AnsiballZ_docker_image.py", line 49, in invoke_module
imp.load_module('__main__', mod, module, MOD_DESC)
File "/tmp/ansible_docker_image_payload_jIHMqS/__main__.py", line 926, in <module>
File "/tmp/ansible_docker_image_payload_jIHMqS/__main__.py", line 921, in main
File "/tmp/ansible_docker_image_payload_jIHMqS/__main__.py", line 476, in __init__
File "/tmp/ansible_docker_image_payload_jIHMqS/__main__.py", line 504, in present
File "/tmp/ansible_docker_image_payload_jIHMqS/__main__.py", line 731, in build_image
File "/usr/local/lib/python2.7/dist-packages/docker/api/build.py", line 269, in build
timeout=timeout,
File "/usr/local/lib/python2.7/dist-packages/docker/utils/decorators.py", line 46, in inner
return f(self, *args, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/docker/api/client.py", line 226, in _post
return self.post(url, **self._set_request_timeout(kwargs))
File "/usr/local/lib/python2.7/dist-packages/requests/sessions.py", line 581, in post
return self.request('POST', url, data=data, json=json, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/requests/sessions.py", line 533, in request
resp = self.send(prep, **send_kwargs)
File "/usr/local/lib/python2.7/dist-packages/requests/sessions.py", line 646, in send
r = adapter.send(request, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/requests/adapters.py", line 498, in send
raise ConnectionError(err, request=request)
requests.exceptions.ConnectionError: ('Connection aborted.', error(104, 'Connection reset by peer'))
module_stdout: ''
msg: |-
MODULE FAILURE
See stdout/stderr for the exact error
rc: 1

# Fixed code
def main():
    argument_spec = dict(
        containers=dict(type="bool", default=False),
        containers_filters=dict(type="dict"),
        images=dict(type="bool", default=False),
        images_filters=dict(type="dict"),
        networks=dict(type="bool", default=False),
        networks_filters=dict(type="dict"),
        volumes=dict(type="bool", default=False),
        volumes_filters=dict(type="dict"),
        builder_cache=dict(type="bool", default=False),
    )

    client = AnsibleDockerClient(
        argument_spec=argument_spec,
        # supports_check_mode=True,
        min_docker_api_version="1.25",
        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:
        result = dict()

        if client.module.params["containers"]:
            filters = clean_dict_booleans_for_docker_api(
                client.module.params.get("containers_filters")
            )
            res = client.prune_containers(filters=filters)
            result["containers"] = res.get("ContainersDeleted") or []
            result["containers_space_reclaimed"] = res["SpaceReclaimed"]

        if client.module.params["images"]:
            filters = clean_dict_booleans_for_docker_api(
                client.module.params.get("images_filters")
            )
            res = client.prune_images(filters=filters)
            result["images"] = res.get("ImagesDeleted") or []
            result["images_space_reclaimed"] = res["SpaceReclaimed"]

        if client.module.params["networks"]:
            filters = clean_dict_booleans_for_docker_api(
                client.module.params.get("networks_filters")
            )
            res = client.prune_networks(filters=filters)
            result["networks"] = res.get("NetworksDeleted") or []

        if client.module.params["volumes"]:
            filters = clean_dict_booleans_for_docker_api(
                client.module.params.get("volumes_filters")
            )
            res = client.prune_volumes(filters=filters)
            result["volumes"] = res.get("VolumesDeleted") or []
            result["volumes_space_reclaimed"] = res["SpaceReclaimed"]

        if client.module.params["builder_cache"]:
            res = client.prune_builds()
            result["builder_cache_space_reclaimed"] = res["SpaceReclaimed"]

        client.module.exit_json(**result)
    except DockerException as e:
        client.fail(
            "An unexpected docker error occurred: {0}".format(e),
            exception=traceback.format_exc(),
        )
    except RequestException as e:
        client.fail(
            "An unexpected requests error occurred when docker-py tried to talk to the docker daemon: {0}".format(
                e
            ),
            exception=traceback.format_exc(),
        )