cola/compat.py

   1 # pylint: disable=unused-import,redefined-builtin,undefined-variable
   2 from __future__ import absolute_import, division, print_function, unicode_literals
   3 import os
   4 import sys
   5
   6 try:
   7     import urllib2 as parse  # noqa
   8 except ImportError:
   9     # Python 3
  10     from urllib import parse  # noqa
  11
  12
  13 PY_VERSION = sys.version_info[:2]  # (2, 7)
  14 PY_VERSION_MAJOR = PY_VERSION[0]
  15 PY2 = PY_VERSION_MAJOR == 2
  16 PY3 = PY_VERSION_MAJOR >= 3
  17 PY26_PLUS = PY2 and sys.version_info[1] >= 6
  18 WIN32 = sys.platform in {'win32', 'cygwin'}
  19 ENCODING = 'utf-8'
  20
  21
  22 if PY3:
  23
  24     def bstr(x, encoding=ENCODING):
  25         return bytes(x, encoding=encoding)
  26
  27 elif PY26_PLUS:
  28     bstr = bytes
  29 else:
  30     # Python <= 2.5
  31     bstr = str
  32
  33
  34 if PY3:
  35
  36     def bchr(i):
  37         return bytes([i])
  38
  39     int_types = (int,)
  40     maxsize = sys.maxsize
  41     ustr = str
  42     uchr = chr
  43 else:
  44     bchr = chr
  45     maxsize = 2**31
  46     # pylint: disable=unicode-builtin
  47     ustr = unicode  # noqa
  48     # pylint: disable=unichr-builtin
  49     uchr = unichr  # noqa
  50     # pylint: disable=long-builtin
  51     int_types = (int, long)  # noqa
  52
  53 # Qt's max 32-bit signed integer range (-2147483648 to 2147483647)
  54 maxint = (2**31) - 1
  55
  56
  57 def setenv(key, value):
  58     """Compatibility wrapper for setting environment variables
  59
  60     Why?  win32 requires putenv().  UNIX only requires os.environ.
  61     """
  62     if not PY3 and isinstance(value, ustr):
  63         value = value.encode(ENCODING, 'replace')
  64     os.environ[key] = value
  65     os.putenv(key, value)
  66
  67
  68 def unsetenv(key):
  69     """Compatibility wrapper for unsetting environment variables"""
  70     os.environ.pop(key, None)
  71     if hasattr(os, 'unsetenv'):
  72         os.unsetenv(key)
  73
  74
  75 def no_op(value):
  76     """Return the value as-is"""
  77     return value
  78
  79
  80 def byte_offset_to_int_converter():
  81     """Return a function to convert byte string offsets into ints
  82
  83     Indexing into python3 bytes returns ints, Python2 returns str.
  84     Thus, on Python2 we need to use `ord()` to convert the byte into
  85     an integer.  It's already an int on Python3, so we use no_op there.
  86     """
  87     if PY2:
  88         result = ord
  89     else:
  90         result = no_op
  91     return result