from .base import Docker


class Images(Docker):
    def list_images(self):
        images = self.client.images.list()
        return images

    def pull_image(self, repository):
        image = self.client.images.pull(repository)
        return image