diff --git a/colout/colout.py b/colout/colout.py index 69cd78a..40335fd 100755 --- a/colout/colout.py +++ b/colout/colout.py @@ -31,6 +31,8 @@ signal.signal( signal.SIGPIPE, signal.SIG_DFL ) # Global variable(s) ############################################################################### +PY2 = sys.version_info.major == 2 + context = {} debug = False @@ -753,7 +755,7 @@ def write(colored, stream = sys.stdout): """ Write "colored" on sys.stdout, then flush. """ - if six.PY2: # If Python 2.x: force unicode + if PY2: # If Python 2.x: force unicode if isinstance(colored, unicode): colored = colored.encode('utf-8') try: diff --git a/setup.py b/setup.py index daa8d35..a79d0cf 100644 --- a/setup.py +++ b/setup.py @@ -14,7 +14,7 @@ if sys.argv[-1] == 'publish': packages = ['colout'] -requires = ['argparse', 'pygments', 'babel'] +requires = ['argparse; python_version < "2.7"', 'pygments', 'babel'] setup( name='colout',