ugly->bad

cdist/__init__.py

@@ -68,13 +68,13 @@ class CdistBetaRequired(cdist.Error):
             err_msg = ("\'{}\' command is beta, but beta is "
                        "not enabled. If you want to use it please enable beta "
                        "functionalities by using the -b/--enable-beta command "
-                       "line flag.")
+                       "line flag or setting CDIST_BETA env var.")
             fmt_args = [self.command, ]
         else:
             err_msg = ("\'{}\' argument of \'{}\' command is beta, but beta "
                        "is not enabled. If you want to use it please enable "
                        "beta functionalities by using the -b/--enable-beta "
-                       "command line flag.")
+                       "command line flag or setting CDIST_BETA env var.")
             fmt_args = [self.arg, self.command, ]
         return err_msg.format(*fmt_args)
 

cdist/core/explorer.py

@@ -149,12 +149,7 @@ class Explorer(object):
     def transfer_global_explorers(self):
         """Transfer the global explorers to the remote side."""
         self.remote.mkdir(self.remote.global_explorer_path)
-        if self.jobs is None:
         self.remote.transfer(self.local.global_explorer_path,
-                                 self.remote.global_explorer_path)
-        else:
-            self.remote.transfer_dir_parallel(
-                    self.local.global_explorer_path,
                              self.remote.global_explorer_path,
                              self.jobs)
         self.remote.run(["chmod", "0700",

cdist/exec/remote.py

@@ -118,30 +118,34 @@ class Remote(object):
         self.log.debug("Remote mkdir: %s", path)
         self.run(["mkdir", "-p", path])
 
-    def transfer(self, source, destination):
+    def transfer(self, source, destination, jobs=None):
         """Transfer a file or directory to the remote side."""
         self.log.debug("Remote transfer: %s -> %s", source, destination)
         self.rmdir(destination)
         if os.path.isdir(source):
             self.mkdir(destination)
-            for f in glob.glob1(source, '*'):
+            if jobs:
-                command = self._copy.split()
+                self._transfer_dir_parallel(source, destination, jobs)
-                path = os.path.join(source, f)
+            else:
-                command.extend([path, '{0}:{1}'.format(
+                self._transfer_dir_sequential(source, destination)
-                    _wrap_addr(self.target_host[0]), destination)])
+        elif jobs:
-                self._run_command(command)
+            raise cdist.Error("Source {} is not a directory".format(source))
         else:
             command = self._copy.split()
             command.extend([source, '{0}:{1}'.format(
                 _wrap_addr(self.target_host[0]), destination)])
             self._run_command(command)
 
-    def transfer_dir_parallel(self, source, destination, jobs):
+    def _transfer_dir_sequential(self, source, destination):
+        for f in glob.glob1(source, '*'):
+            command = self._copy.split()
+            path = os.path.join(source, f)
+            command.extend([path, '{0}:{1}'.format(
+                _wrap_addr(self.target_host[0]), destination)])
+            self._run_command(command)
+
+    def _transfer_dir_parallel(self, source, destination, jobs):
         """Transfer a directory to the remote side in parallel mode."""
-        self.log.debug("Remote transfer: %s -> %s", source, destination)
-        self.rmdir(destination)
-        if os.path.isdir(source):
-            self.mkdir(destination)
         self.log.info("Remote transfer in {} parallel jobs".format(
             jobs))
         self.log.debug("Multiprocessing start method is {}".format(
@@ -167,8 +171,6 @@ class Remote(object):
                 r.get()  # self._run_command returns None
             self.log.debug(("Multiprocessing for parallel transfer "
                             "finished"))
-        else:
-            raise cdist.Error("Source {} is not a directory".format(source))
 
     def run_script(self, script, env=None, return_output=False):
         """Run the given script with the given environment on the remote side.

cdist/test/exec/remote.py

@@ -136,7 +136,7 @@ class RemoteTestCase(test.CdistTestCase):
             source_file_name = os.path.split(source_file)[-1]
             filenames.append(source_file_name)
         target = self.mkdtemp(dir=self.temp_dir)
-        self.remote.transfer_dir_parallel(source, target,
+        self.remote.transfer(source, target,
                              multiprocessing.cpu_count())
         # test if the payload files are in the target directory
         for filename in filenames:

docs/src/cdist-reference.rst.sh

@@ -273,4 +273,7 @@ CDIST_REMOTE_EXEC
 
 CDIST_REMOTE_COPY
     Use this command for remote copy (should behave like scp).
+
+CDIST_BETA
+    Enable beta functionalities.
 eof

docs/src/man1/cdist.rst

@@ -236,6 +236,9 @@ CDIST_REMOTE_EXEC
 CDIST_REMOTE_COPY
     Use this command for remote copy (should behave like scp).
 
+CDIST_BETA
+    Enable beta functionalities.
+
 EXIT STATUS
 -----------
 The following exit values shall be returned: