summaryrefslogtreecommitdiff
path: root/contrib/python/cmd/lib/actions/rmi_action.py
blob: db59fe03030b9fe1bcfb77f77ebf814cf7485c2a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
"""Remote client command for deleting images."""
import sys

import podman

from .. import AbstractActionBase


class Rmi(AbstractActionBase):
    """Clas for removing images from storage."""

    @classmethod
    def subparser(cls, parent):
        """Add Rmi command to parent parser."""
        parser = parent.add_parser('rmi', help='delete image(s)')
        parser.add_argument(
            '-f',
            '--force',
            action='store_true',
            help=('force delete of image(s) and associated containers.'
                  ' (default: %(default)s)'))
        parser.add_argument('targets', nargs='*', help='image id(s) to delete')
        parser.set_defaults(klass=cls, method='remove')

    def __init__(self, args):
        """Construct Rmi class."""
        super().__init__(args)
        if len(args.targets) < 1:
            raise ValueError('You must supply at least one image id'
                             ' or name to be deleted.')

    def remove(self):
        """Remove image(s)."""
        for id in self._args.targets:
            try:
                img = self.client.images.get(id)
                img.remove(self._args.force)
                print(id)
            except podman.ImageNotFound as e:
                sys.stdout.flush()
                print(
                    'Image {} not found.'.format(e.name),
                    file=sys.stderr,
                    flush=True)
            except podman.ErrorOccurred as e:
                sys.stdout.flush()
                print(
                    '{}'.format(e.reason).capitalize(),
                    file=sys.stderr,
                    flush=True)