Fix ColorStream

2 files changed, 14 insertions, 10 deletions

diff --git a/lib/spack/llnl/util/tty/color.py b/lib/spack/llnl/util/tty/color.py
index 598e9d44f5..81688d7f14 100644
--- a/lib/spack/llnl/util/tty/color.py
+++ b/lib/spack/llnl/util/tty/color.py
@@ -177,17 +177,20 @@ def cescape(string):
 
 class ColorStream(object):
     def __init__(self, stream, color=None):
-        self.__class__ = type(stream.__class__.__name__,
-                              (self.__class__, stream.__class__), {})
-        self.__dict__ = stream.__dict__
-        self.color = color
-        self.stream = stream
+        self._stream = stream
+        self._color = color
 
     def write(self, string, **kwargs):
-        if kwargs.get('raw', False):
-            super(ColorStream, self).write(string)
-        else:
-            cwrite(string, self.stream, self.color)
+        raw = kwargs.get('raw', False)
+        raw_write = getattr(self._stream, 'write')
+
+        color = self._color
+        if self._color is None:
+            if raw:
+                color=True
+            else:
+                color = self._stream.isatty()
+        raw_write(colorize(string, color=color))
 
     def writelines(self, sequence, **kwargs):
         raw = kwargs.get('raw', False)
diff --git a/lib/spack/spack/spec.py b/lib/spack/spack/spec.py
index 9239bac08f..3386af8d7f 100644
--- a/lib/spack/spack/spec.py
+++ b/lib/spack/spack/spec.py
@@ -1328,7 +1328,8 @@ class Spec(object):
 
     def graph(self, **kwargs):
         N      = kwargs.get('node', 'o')        # Node character
-        out    = kwargs.get('out', sys.stdout)
+        color  = kwargs.get('color', True)
+        out    = kwargs.get('out', ColorStream(sys.stdout, color=color))
         indent = kwargs.get('indent', 0)
         indent *= ' '