StringIO compat

octodns/compat.py

@@ -0,0 +1,10 @@
+#
+# Python 2/3 compat bits
+#
+
+try:  # pragma: no cover
+    from StringIO import StringIO
+except ImportError:  # pragma: no cover
+    from io import StringIO
+
+StringIO

octodns/provider/plan.py

@@ -5,11 +5,11 @@
 from __future__ import absolute_import, division, print_function, \
     unicode_literals
 
-from StringIO import StringIO
 from logging import DEBUG, ERROR, INFO, WARN, getLogger
 from sys import stdout
 
 from six import text_type
+from ..compat import StringIO
 
 
 class UnsafePlan(Exception):

tests/test_octodns_plan.py

@@ -5,11 +5,11 @@
 from __future__ import absolute_import, division, print_function, \
     unicode_literals
 
-from StringIO import StringIO
 from logging import getLogger
 from six import text_type
 from unittest import TestCase
 
+from octodns.compat import StringIO
 from octodns.provider.plan import Plan, PlanHtml, PlanLogger, PlanMarkdown
 from octodns.record import Create, Delete, Record, Update
 from octodns.zone import Zone

tests/test_octodns_yaml.py

@@ -5,10 +5,10 @@
 from __future__ import absolute_import, division, print_function, \
     unicode_literals
 
-from StringIO import StringIO
 from unittest import TestCase
 from yaml.constructor import ConstructorError
 
+from octodns.compat import StringIO
 from octodns.yaml import safe_dump, safe_load