From def2ab01381b8fcbc9f33b5270b64df149b10e40 Mon Sep 17 00:00:00 2001 From: Dmitry Petrov Date: Thu, 27 Apr 2017 21:34:14 -0700 Subject: [PATCH] data-sync multiple imports. Close #50 --- dvc/command/data_sync.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/dvc/command/data_sync.py b/dvc/command/data_sync.py index c461feb87..38da1fab0 100644 --- a/dvc/command/data_sync.py +++ b/dvc/command/data_sync.py @@ -44,7 +44,10 @@ class CmdDataSync(CmdBase): super(CmdDataSync, self).__init__(settings) def define_args(self, parser): - self.add_string_arg(parser, 'target', 'Target to sync - file or directory') + parser.add_argument('targets', + metavar='', + help='File or directory to sync', + nargs='*') def run(self): if self.is_locker: @@ -55,12 +58,13 @@ class CmdDataSync(CmdBase): return 1 try: - if System.islink(self.parsed_args.target): - data_item = self.settings.path_factory.data_item(self.parsed_args.target) - return self.sync_symlink(data_item) + for target in self.parsed_args.targets: + if System.islink(target): + data_item = self.settings.path_factory.data_item(target) + return self.sync_symlink(data_item) - if os.path.isdir(self.parsed_args.target): - return self.sync_dir(self.parsed_args.target) + if os.path.isdir(target): + return self.sync_dir(target) raise DataSyncError('File "{}" does not exit'.format(self.parsed_args.target)) finally: