dvc/dvc/command/remove.py

import dvc.prompt as prompt
import dvc.logger as logger
from dvc.exceptions import DvcException
from dvc.command.base import CmdBase


class CmdRemove(CmdBase):
    def _is_outs_only(self, target):
        if not self.args.purge:
            return True

        if self.args.force:
            return False

        msg = (
            u'Are you sure you want to remove {} with its outputs?'
            .format(target)
        )

        if prompt.confirm(msg):
            return False

        raise DvcException(
            u'Cannot purge without a confirmation from the user.'
            u" Use '-f' to force."
        )

    def run(self):
        for target in self.args.targets:
            try:
                outs_only = self._is_outs_only(target)
                self.project.remove(target, outs_only=outs_only)
            except DvcException:
                logger.error('failed to remove {}'.format(target))
                return 1
        return 0


def add_parser(subparsers, parent_parser):
    REMOVE_HELP = 'Remove outputs of DVC file.'
    remove_parser = subparsers.add_parser(
        'remove',
        parents=[parent_parser],
        description=REMOVE_HELP,
        help=REMOVE_HELP)
    remove_parser_group = remove_parser.add_mutually_exclusive_group()
    remove_parser_group.add_argument(
        '-o',
        '--outs',
        action='store_true',
        default=True,
        help='Only remove DVC file outputs.(default)')
    remove_parser_group.add_argument(
        '-p',
        '--purge',
        action='store_true',
        default=False,
        help='Remove DVC file and all its outputs.')
    remove_parser.add_argument(
        '-f',
        '--force',
        action='store_true',
        default=False,
        help='Force purge.')
    remove_parser.add_argument(
        'targets',
        nargs='+',
        help='DVC files.')
    remove_parser.set_defaults(func=CmdRemove)