Bytes: are just constants like Str and Num

2 files changed, 11 insertions, 2 deletions

diff --git a/_nodes_ast.py b/_nodes_ast.py
index a49a2fd1..b1facea3 100644
--- a/_nodes_ast.py
+++ b/_nodes_ast.py
@@ -529,14 +529,20 @@ class TreeRebuilder(RebuildVisitor):
         newnode.set_line_info(newnode.last_child())
         return newnode
 
+    def visit_bytes(self, node, parent):
+        """visit a Bytes node by returning a fresh instance of Const"""
+        newnode = new.Const(node.s)
+        self._set_infos(node, newnode, parent)
+        return newnode
+
     def visit_num(self, node, parent):
-        """visit a a Num node by returning a fresh instance of Const"""
+        """visit a Num node by returning a fresh instance of Const"""
         newnode = new.Const(node.n)
         self._set_infos(node, newnode, parent)
         return newnode
 
     def visit_str(self, node, parent):
-        """visit a a Str node by returning a fresh instance of Const"""
+        """visit a Str node by returning a fresh instance of Const"""
         newnode = new.Const(node.s)
         self._set_infos(node, newnode, parent)
         return newnode
diff --git a/inference.py b/inference.py
index 257177c0..25c5fed5 100644
--- a/inference.py
+++ b/inference.py
@@ -25,6 +25,7 @@ from __future__ import generators
 __doctype__ = "restructuredtext en"
 
 from itertools import chain
+import sys
 
 try:
     GeneratorExit # introduced in py2.5
@@ -52,6 +53,8 @@ _CONST_PROXY = {
     unicode: MANAGER.astng_from_class(unicode),
     }
 _CONST_PROXY[type(None)].parent = _CONST_PROXY[bool].parent
+if sys.version_info >= (2, 6):
+    _CONST_PROXY[bytes] = MANAGER.astng_from_class(bytes)
 
 # TODO : find a nicer way to handle this situation; we should at least
 # be able to avoid calling MANAGER.astng_from_class(const.value.__class__)