io.StringIO fallback for py3k

2013-07-19 21:27:06 -05:00 · 2013-07-19 21:27:06 -05:00 · 71aaf23285
parent 528eae7e46
commit 71aaf23285
1 changed files with 4 additions and 1 deletions
--- a/pdfparanoia/parser.py
+++ b/pdfparanoia/parser.py
@ -7,7 +7,10 @@ Deals with the existential nature of parsing pdfs.

 """

-from StringIO import StringIO
+try:
+    from StringIO import StringIO
+except ImportError: # py3k
+    from io import StringIO, BytesIO

 # Maybe one day pdfquery will be able to save pdf.
 # from pdfquery import PDFQuery