Final draft of improved state printing

1 files changed, 11 insertions, 0 deletions

diff --git a/astroid/context.py b/astroid/context.py
index 5a2ea52..3abea87 100644
--- a/astroid/context.py
+++ b/astroid/context.py
@@ -19,6 +19,7 @@
 """Various context related utilities, including inference and call contexts."""
 import contextlib
 import itertools
+import pprint
 
 from astroid import exceptions
 from astroid import util
@@ -64,6 +65,16 @@ class InferenceContext(object):
         yield
         self.path = path
 
+    def __str__(self):
+        return '%s(%s)' % (type(self).__name__, ',\n    '.join(
+            ('%s=%s' % (a, pprint.pformat(getattr(self, a), width=80-len(a)))
+             for a in self.__slots__)))
+    
+    def __repr__(self):
+        return '%s(%s)' % (type(self).__name__, ', '.join(
+            ('%s=%s' % (a, repr(getattr(self, a)))
+             for a in self.__slots__)))
+
 
 class CallContext(object):