Docker Tasks could be more generic to support all API methods.

[bukowa]

Docker tasks are limited, and we cannot pass custom kwargs into new APIClient. Do you think that Docker based Tasks could be more generic / abstract to handle all API endpoints?

[joshmeek]

@bukowa Yeah these could definitely be expanded upon! For example, the CreateContainer task has extra_docker_kwargs for setting any custom kwargs that you want to forward to the Docker client:

prefect/src/prefect/tasks/docker/containers.py

Line 27 in ce234bf

- extra_docker_kwargs (dict, optional): Extra kwargs to pass through to `create_container`

The task library is generally community-driven and we highly encourage contributions! 🙂

[bukowa]

What about something like that?
@joshmeek

import operator

import docker
from prefect import Task


class DockerTask(Task):

    def __init__(self, client, method, **kwargs):
        self.docker_func = operator.attrgetter(method)(client)
        super().__init__(**kwargs)

    def run(self, **kwargs) -> None:
        return self.docker_func(**kwargs)


if __name__ == '__main__':
    client = docker.client.APIClient()
    task = DockerTask(
        client=client,
        method='containers',
    )
    task.run(all=True)

[joshmeek]

Yeah that would be an appropriate docker task!