diff --git a/easy_pdf/compat.py b/easy_pdf/compat.py new file mode 100644 index 0000000..3a3e176 --- /dev/null +++ b/easy_pdf/compat.py @@ -0,0 +1,14 @@ +# coding=utf-8 + +import sys + +__all__ = ['BytesIO'] + +PY2 = sys.version_info[0] == 2 + +if PY2: + import StringIO + BytesIO = StringIO.StringIO +else: + import io + BytesIO = io.BytesIO diff --git a/easy_pdf/rendering.py b/easy_pdf/rendering.py index 53aa3f0..0d4340c 100644 --- a/easy_pdf/rendering.py +++ b/easy_pdf/rendering.py @@ -9,11 +9,11 @@ from django.template import loader from django.http import HttpResponse from django.utils.http import urlquote -from django.utils.six import BytesIO import xhtml2pdf.default from xhtml2pdf import pisa +from .compat import BytesIO from .exceptions import UnsupportedMediaPathException, PDFRenderingError logger = logging.getLogger("app.pdf")