save_labscripts()'s calls to git/hg for a given file now run in parallel.

zakv · zakv · commit 58ed6db264c2 · 2020-10-04T12:28:35.000-04:00
diff --git a/labscript/labscript.py b/labscript/labscript.py
@@ -2201,18 +2201,24 @@ def save_labscripts(hdf5_file):
                     hdf5_file.create_dataset(save_path, data=open(path).read())
                     if compiler.save_hg_info:
                         hg_commands = [['log', '--limit', '1'], ['status'], ['diff']]
+                        process_list = []
                         for command in hg_commands:
                             process = subprocess.Popen(['hg'] + command + [os.path.split(path)[1]], cwd=os.path.split(path)[0],
                                                        stdout=subprocess.PIPE, stderr=subprocess.PIPE, startupinfo=startupinfo)
+                            process_list.append(process)
+                        for process, command in zip(process_list, hg_commands):
                             info, err = process.communicate()
                             if info or err:
                                 hdf5_file[save_path].attrs['hg ' + str(command[0])] = info.decode('utf-8') + '\n' + err.decode('utf-8')
                     if compiler.save_git_info:
                         module_filename = os.path.split(path)[1]
                         git_commands = [['branch', '--show-current'], ['rev-parse', '--verify', 'HEAD'], ['diff', 'HEAD', module_filename]]
+                        process_list = []
                         for command in git_commands:
                             process = subprocess.Popen(['git'] + command, cwd=os.path.split(path)[0], stdout=subprocess.PIPE,
                                                        stderr=subprocess.PIPE, startupinfo=startupinfo)
+                            process_list.append(process)
+                        for process, command in zip(process_list, git_commands):
                             info, err = process.communicate()
                             hdf5_file[save_path].attrs['git ' + str(command[0])] = info.decode('utf-8') + '\n' + err.decode('utf-8')
     except ImportError:
