Implement FileDisposition in C for speed. Final part of fix for #387.

1 files changed, 12 insertions, 2 deletions

diff --git a/coverage/collector.py b/coverage/collector.py
index b8e0841..ae5f6c8 100644
--- a/coverage/collector.py
+++ b/coverage/collector.py
@@ -13,7 +13,7 @@ from coverage.pytracer import PyTracer
 
 try:
     # Use the C extension code when we can, for speed.
-    from coverage.tracer import CTracer         # pylint: disable=no-name-in-module
+    from coverage.tracer import CTracer, CFileDisposition   # pylint: disable=no-name-in-module
 except ImportError:
     # Couldn't import the C extension, maybe it isn't built.
     if os.getenv('COVERAGE_TEST_TRACER') == 'c':
@@ -30,6 +30,11 @@ except ImportError:
     CTracer = None
 
 
+class FileDisposition(object):
+    """A simple value type for recording what to do with a file."""
+    pass
+
+
 class Collector(object):
     """Collects trace data.
 
@@ -124,7 +129,12 @@ class Collector(object):
             # trace function.
             self._trace_class = CTracer or PyTracer
 
-        self.supports_plugins = self._trace_class is CTracer
+        if self._trace_class is CTracer:
+            self.file_disposition_class = CFileDisposition
+            self.supports_plugins = True
+        else:
+            self.file_disposition_class = FileDisposition
+            self.supports_plugins = False
 
     def __repr__(self):
         return "<Collector at 0x%x>" % id(self)