Use six to support Python 2 and 3 without using 2to3

4 files changed, 19 insertions, 20 deletions

diff --git a/blessings/__init__.py b/blessings/__init__.py
index 9e10ebc..0212ae9 100644
--- a/blessings/__init__.py
+++ b/blessings/__init__.py
@@ -1,10 +1,12 @@
 """A thin, practical wrapper around terminal coloring, styling, and
 positioning"""
 
+from __future__ import absolute_import
 from contextlib import contextmanager
 import curses
 from curses import setupterm, tigetnum, tigetstr, tparm
 from fcntl import ioctl
+import six
 
 try:
     from io import UnsupportedOperation as IOUnsupportedOperation
@@ -420,7 +422,7 @@ COMPOUNDABLES = (COLORS |
                       'shadow', 'standout', 'subscript', 'superscript']))
 
 
-class ParametrizingString(unicode):
+class ParametrizingString(six.text_type):
     """A Unicode string which can be called to parametrize it as a terminal
     capability"""
 
@@ -432,7 +434,7 @@ class ParametrizingString(unicode):
             "normal" capability.
 
         """
-        new = unicode.__new__(cls, formatting)
+        new = six.text_type.__new__(cls, formatting)
         new._normal = normal
         return new
 
@@ -461,7 +463,7 @@ class ParametrizingString(unicode):
         except TypeError:
             # If the first non-int (i.e. incorrect) arg was a string, suggest
             # something intelligent:
-            if len(args) == 1 and isinstance(args[0], basestring):
+            if len(args) == 1 and isinstance(args[0], six.string_types):
                 raise TypeError(
                     'A native or nonexistent capability template received '
                     '%r when it was expecting ints. You probably misspelled a '
@@ -472,12 +474,12 @@ class ParametrizingString(unicode):
                 raise
 
 
-class FormattingString(unicode):
+class FormattingString(six.text_type):
     """A Unicode string which can be called upon a piece of text to wrap it in
     formatting"""
 
     def __new__(cls, formatting, normal):
-        new = unicode.__new__(cls, formatting)
+        new = six.text_type.__new__(cls, formatting)
         new._normal = normal
         return new
 
@@ -492,7 +494,7 @@ class FormattingString(unicode):
         return self + text + self._normal
 
 
-class NullCallableString(unicode):
+class NullCallableString(six.text_type):
     """A dummy callable Unicode to stand in for ``FormattingString`` and
     ``ParametrizingString``
 
@@ -500,7 +502,7 @@ class NullCallableString(unicode):
 
     """
     def __new__(cls):
-        new = unicode.__new__(cls, u'')
+        new = six.text_type.__new__(cls, u'')
         return new
 
     def __call__(self, *args):
diff --git a/blessings/tests.py b/blessings/tests.py
index aff3a2f..c51b36c 100644
--- a/blessings/tests.py
+++ b/blessings/tests.py
@@ -10,13 +10,14 @@ xterm-256color exists.
 
 """
 from __future__ import with_statement  # Make 2.5-compatible
+from __future__ import absolute_import
 from curses import tigetstr, tparm
 from functools import partial
-from StringIO import StringIO
 import sys
 
 from nose import SkipTest
 from nose.tools import eq_
+from six import StringIO
 
 # This tests that __all__ is correct, since we use below everything that should
 # be imported:
@@ -229,22 +230,22 @@ def test_nice_formatting_errors():
     t = TestTerminal()
     try:
         t.bold_misspelled('hey')
-    except TypeError, e:
+    except TypeError as e:
         assert 'probably misspelled' in e.args[0]
 
     try:
         t.bold_misspelled(u'hey')  # unicode
-    except TypeError, e:
+    except TypeError as e:
         assert 'probably misspelled' in e.args[0]
 
     try:
         t.bold_misspelled(None)  # an arbitrary non-string
-    except TypeError, e:
+    except TypeError as e:
         assert 'probably misspelled' not in e.args[0]
 
     try:
         t.bold_misspelled('a', 'b')  # >1 string arg
-    except TypeError, e:
+    except TypeError as e:
         assert 'probably misspelled' not in e.args[0]
 
 
diff --git a/setup.py b/setup.py
index 8584935..adf5060 100644
--- a/setup.py
+++ b/setup.py
@@ -10,10 +10,6 @@ except ImportError:
 from setuptools import setup, find_packages
 
 
-extra_setup = {}
-if sys.version_info >= (3,):
-    extra_setup['use_2to3'] = True
-
 setup(
     name='blessings',
     version='1.6.1',
@@ -23,6 +19,7 @@ setup(
     author_email='erikrose@grinchcentral.com',
     license='MIT',
     packages=find_packages(exclude=['ez_setup']),
+    install_requires=['six'],
     tests_require=['nose'],
     test_suite='nose.collector',
     url='https://github.com/erikrose/blessings',
@@ -49,5 +46,4 @@ setup(
         'Topic :: Terminals'
         ],
     keywords=['terminal', 'tty', 'curses', 'ncurses', 'formatting', 'style', 'color', 'console'],
-    **extra_setup
 )
diff --git a/tox.ini b/tox.ini
index cf09e35..b7dcfb6 100644
--- a/tox.ini
+++ b/tox.ini
@@ -3,6 +3,6 @@ envlist = py27, py34, py36
 
 [testenv]
 commands = nosetests blessings
-deps = nose
-# So Python 3 runs don't pick up incompatible, un-2to3'd source from the cwd:
-changedir = .tox
\ No newline at end of file
+deps =
+    nose
+    six