Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-07 15:28:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

pull/push: generalize CmdDataBase class

Browse Source 
Signed-off-by: Ruslan Kuprieiev <[email protected]>
This commit is contained in:
Ruslan Kuprieiev
2017-11-26 16:10:37 +02:00
parent d9e06c07b0
commit 7c320bbff3
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dvc/command/data_sync.py
+11 -9
View File
@@ -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):