From 9e0842fe32bb0fee205c9132d55e384b29446bc9 Mon Sep 17 00:00:00 2001
From: Christine Lytwynec <chris.lytwynec@gmail.com>
Date: Tue, 21 Apr 2015 11:28:13 -0400
Subject: Added ability to combine coverage data files from multiple
 directories into one file via command line args.

--HG--
branch : combine-from-multiple-dirs
---
 coverage/data.py | 40 +++++++++++++++++++++++++---------------
 1 file changed, 25 insertions(+), 15 deletions(-)

(limited to 'coverage/data.py')

diff --git a/coverage/data.py b/coverage/data.py
index 2c5d3516..ed79f794 100644
--- a/coverage/data.py
+++ b/coverage/data.py
@@ -1,5 +1,6 @@
 """Coverage data for Coverage."""
 
+import glob
 import os
 
 from coverage.backward import iitems, pickle
@@ -190,7 +191,7 @@ class CoverageData(object):
             pass
         return lines, arcs, plugins
 
-    def combine_parallel_data(self, aliases=None):
+    def combine_parallel_data(self, aliases=None, data_dirs=None):
         """Combine a number of data files together.
 
         Treat `self.filename` as a file prefix, and combine the data from all
@@ -199,23 +200,32 @@ class CoverageData(object):
         If `aliases` is provided, it's a `PathAliases` object that is used to
         re-map paths to match the local machine's.
 
+        If `data_dirs` is provided, then it combines the data files from each
+        directory into a single file.
+
         """
         aliases = aliases or PathAliases()
         data_dir, local = os.path.split(self.filename)
-        localdot = local + '.'
-        for f in os.listdir(data_dir or '.'):
-            if f.startswith(localdot):
-                full_path = os.path.join(data_dir, f)
-                new_lines, new_arcs, new_plugins = self._read_file(full_path)
-                for filename, file_data in iitems(new_lines):
-                    filename = aliases.map(filename)
-                    self.lines.setdefault(filename, {}).update(file_data)
-                for filename, file_data in iitems(new_arcs):
-                    filename = aliases.map(filename)
-                    self.arcs.setdefault(filename, {}).update(file_data)
-                self.plugins.update(new_plugins)
-                if f != local:
-                    os.remove(full_path)
+        localdot = local + '.*'
+
+        data_dirs = data_dirs or [data_dir] or ['.']
+        files_to_combine = []
+        for d in data_dirs:
+            pattern = os.path.join(os.path.abspath(d), localdot)
+            files_to_combine.extend(glob.glob(pattern))
+
+        for f in files_to_combine:
+            new_lines, new_arcs, new_plugins = self._read_file(f)
+            for filename, file_data in iitems(new_lines):
+                filename = aliases.map(filename)
+                self.lines.setdefault(filename, {}).update(file_data)
+            for filename, file_data in iitems(new_arcs):
+                filename = aliases.map(filename)
+                self.arcs.setdefault(filename, {}).update(file_data)
+            self.plugins.update(new_plugins)
+
+            if os.path.basename(f) != local:
+                os.remove(f)
 
     def add_line_data(self, line_data):
         """Add executed line data.
-- 
cgit v1.2.1