Python 2.6 compatibilty for isclass

thomasballinger · thomasballinger · commit 92dccc9dee11 · 2015-12-25T12:35:57.000-08:00
diff --git a/bpython/simpleeval.py b/bpython/simpleeval.py
@@ -29,9 +29,11 @@
 """
 
 import ast
-from six import string_types
 import inspect
+from six import string_types
 from six.moves import builtins
+import sys
+import types
 
 from bpython import line as line_properties
 from bpython._py3compat import py3
@@ -47,6 +49,14 @@
     _name_type_nodes = (ast.Name,)
 
 
+# inspect.isclass is broken in Python 2.6
+if sys.version_info[:2] == (2, 6):
+    def isclass(obj):
+        return isinstance(obj, (type, types.ClassType))
+else:
+    isclass = inspect.isclass
+
+
 class EvaluationError(Exception):
     """Raised if an exception occurred in safe_eval."""
 
@@ -267,7 +277,7 @@ def safe_get_attribute_new_style(obj, attr):
     if not is_new_style(obj):
         raise ValueError("%r is not a new-style class or object" % obj)
     to_look_through = (obj.__mro__
-                       if inspect.isclass(obj)
+                       if isclass(obj)
                        else (obj,) + type(obj).__mro__)
 
     for cls in to_look_through:
