Use io for py3

unnonouno · Oct 23, 2014 · c2325b2 · c2325b2
1 parent 8a59952
commit c2325b2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/mrep/builder.py b/mrep/builder.py
@@ -1,6 +1,14 @@
 import mrep.pattern as pattern
 import re
-from cStringIO import StringIO
+import sys
+
+if sys.version_info >= (3, 0):
+    from io import StringIO
+else:
+    try:
+        from cStringIO import StringIO
+    except ImportError:
+        from StringIO import StringIO
 
 
 class ParseError(Exception):