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()