sublime-rana/packages/wakatime/compat.py

94 lines
2.5 KiB
Python
Raw Normal View History

2014-09-30 16:27:35 +00:00
# -*- coding: utf-8 -*-
"""
wakatime.compat
~~~~~~~~~~~~~~~
For working with Python2 and Python3.
:copyright: (c) 2014 Alan Hamlett.
:license: BSD, see LICENSE for more details.
"""
import codecs
import sys
is_py2 = (sys.version_info[0] == 2)
is_py3 = (sys.version_info[0] == 3)
2015-08-25 07:42:37 +00:00
if is_py2: # pragma: nocover
2014-09-30 16:27:35 +00:00
def u(text):
2015-08-25 07:42:37 +00:00
if text is None:
return None
try:
2014-09-30 16:27:35 +00:00
return text.decode('utf-8')
except:
try:
2015-09-29 10:11:25 +00:00
return text.decode(sys.getdefaultencoding())
except:
2015-09-29 10:11:25 +00:00
try:
return unicode(text)
except:
return text
2014-09-30 16:27:35 +00:00
open = codecs.open
basestring = basestring
2015-08-25 07:42:37 +00:00
elif is_py3: # pragma: nocover
2014-09-30 16:27:35 +00:00
def u(text):
2015-08-25 07:42:37 +00:00
if text is None:
return None
2014-09-30 16:27:35 +00:00
if isinstance(text, bytes):
2015-09-29 10:11:25 +00:00
try:
return text.decode('utf-8')
except:
try:
return text.decode(sys.getdefaultencoding())
except:
pass
try:
return str(text)
except:
return text
2014-09-30 16:27:35 +00:00
open = open
basestring = (str, bytes)
2014-12-23 11:39:07 +00:00
2015-08-25 07:42:37 +00:00
2014-12-23 11:39:07 +00:00
try:
from importlib import import_module
2015-08-25 07:42:37 +00:00
except ImportError: # pragma: nocover
2014-12-23 11:39:07 +00:00
def _resolve_name(name, package, level):
"""Return the absolute name of the module to be imported."""
if not hasattr(package, 'rindex'):
raise ValueError("'package' not set to a string")
dot = len(package)
for x in xrange(level, 1, -1):
try:
dot = package.rindex('.', 0, dot)
except ValueError:
raise ValueError("attempted relative import beyond top-level "
"package")
return "%s.%s" % (package[:dot], name)
def import_module(name, package=None):
"""Import a module.
The 'package' argument is required when performing a relative import.
It specifies the package to use as the anchor point from which to
resolve the relative import to an absolute import.
"""
if name.startswith('.'):
if not package:
raise TypeError("relative imports require the 'package' "
+ "argument")
level = 0
for character in name:
if character != '.':
break
level += 1
name = _resolve_name(name[level:], package, level)
__import__(name)
return sys.modules[name]