From 7c320bbff3c72241ea7e8aac6b7219d92ae7687d Mon Sep 17 00:00:00 2001 From: Ruslan Kuprieiev Date: Sun, 26 Nov 2017 15:16:48 +0200 Subject: [PATCH] pull/push: generalize CmdDataBase class Signed-off-by: Ruslan Kuprieiev --- dvc/command/data_sync.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/dvc/command/data_sync.py b/dvc/command/data_sync.py index f8b188fc9..434c90675 100644 --- a/dvc/command/data_sync.py +++ b/dvc/command/data_sync.py @@ -3,21 +3,23 @@ from dvc.logger import Logger import dvc.data_cloud as cloud - -class CmdDataPull(CmdBase): - def __init__(self, settings): - super(CmdDataPull, self).__init__(settings) +class CmdDataBase(CmdBase): + def __init__(self, settings, cmd): + super(CmdDataBase, self).__init__(settings) + self.cmd = cmd def run(self): - self.cloud.pull(self.parsed_args.targets, self.parsed_args.jobs) + getattr(self.cloud, self.cmd)(self.parsed_args.targets, self.parsed_args.jobs) -class CmdDataPush(CmdBase): +class CmdDataPull(CmdDataBase): def __init__(self, settings): - super(CmdDataPush, self).__init__(settings) + super(CmdDataPull, self).__init__(settings, 'pull') - def run(self): - self.cloud.push(self.parsed_args.targets, self.parsed_args.jobs) + +class CmdDataPush(CmdDataBase): + def __init__(self, settings): + super(CmdDataPush, self).__init__(settings, 'push') class CmdDataStatus(CmdBase):