General StringIO for python 2 and 3

tahajahangir · tahajahangir · commit 0eacde62e0f9 · 2011-07-07T16:24:34.000+04:30
diff --git a/html5lib/inputstream.py b/html5lib/inputstream.py
@@ -195,8 +195,12 @@ def openStream(self, source):
             if isinstance(source, unicode):
                 source = source.encode('utf-8')
                 self.charEncoding = ("utf-8", "certain")
-            import cStringIO
-            stream = cStringIO.StringIO(str(source))
+            try:
+                from io import BytesIO
+            except:
+                # 2to3 converts this line to: from io import StringIO  
+                from cStringIO import StringIO as BytesIO
+            stream = BytesIO(source)
 
         if (not(hasattr(stream, "tell") and hasattr(stream, "seek")) or
             stream is sys.stdin):
