iterative/dvc

import os
from abc import ABC, abstractmethod
from typing import TYPE_CHECKING, Any

from dvc.log import logger

if TYPE_CHECKING:
    from dvc.config import Config

logger = logger.getChild(__name__)


class CmdBase(ABC):
    UNINITIALIZED = False

    def __init__(self, args: Any):
        from dvc.repo import Repo

        os.chdir(args.cd)

        self.repo: "Repo" = Repo(uninitialized=self.UNINITIALIZED)
        self.config: "Config" = self.repo.config
        self.args = args

    def do_run(self):
        with self.repo:
            return self.run()

    @abstractmethod
    def run(self):
        pass


class CmdBaseNoRepo(CmdBase):
    def __init__(self, args):
        self.args = args

        os.chdir(args.cd)

    def do_run(self):
        return self.run()