vim-rana/packages/wakatime/compat.py

123 lines
3.5 KiB
Python
Raw Normal View History

2014-09-30 16:23:17 +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.
"""
2017-11-09 07:11:11 +00:00
2014-09-30 16:23:17 +00:00
import codecs
2018-10-03 07:46:21 +00:00
import os
2018-10-01 04:49:26 +00:00
import platform
import subprocess
2014-09-30 16:23:17 +00:00
import sys
is_py2 = (sys.version_info[0] == 2)
is_py3 = (sys.version_info[0] == 3)
2018-10-01 04:49:26 +00:00
is_win = platform.system() == 'Windows'
2014-09-30 16:23:17 +00:00
if is_py2: # pragma: nocover
2014-09-30 16:23:17 +00:00
def u(text):
2015-08-25 07:51:01 +00:00
if text is None:
return None
try:
2014-09-30 16:23:17 +00:00
return text.decode('utf-8')
except:
try:
2015-09-08 04:29:53 +00:00
return text.decode(sys.getdefaultencoding())
except:
2015-09-08 04:29:53 +00:00
try:
return unicode(text)
except:
2016-09-02 08:47:21 +00:00
return text.decode('utf-8', 'replace')
2014-09-30 16:23:17 +00:00
open = codecs.open
basestring = basestring
elif is_py3: # pragma: nocover
2014-09-30 16:23:17 +00:00
def u(text):
2015-08-25 07:51:01 +00:00
if text is None:
return None
2014-09-30 16:23:17 +00:00
if isinstance(text, bytes):
2015-09-08 04:29:53 +00:00
try:
return text.decode('utf-8')
except:
try:
return text.decode(sys.getdefaultencoding())
except:
pass
try:
return str(text)
except:
2016-09-02 08:47:21 +00:00
return text.decode('utf-8', 'replace')
2014-09-30 16:23:17 +00:00
open = open
basestring = (str, bytes)
2014-12-23 11:22:49 +00:00
2014-12-23 11:22:49 +00:00
try:
from importlib import import_module
except ImportError: # pragma: nocover
2014-12-23 11:22:49 +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' "
2017-10-29 19:49:14 +00:00
"argument")
2014-12-23 11:22:49 +00:00
level = 0
for character in name:
if character != '.':
break
level += 1
name = _resolve_name(name[level:], package, level)
__import__(name)
return sys.modules[name]
2017-11-09 07:11:11 +00:00
try:
from .packages import simplejson as json
2017-11-23 20:50:41 +00:00
except (ImportError, SyntaxError): # pragma: nocover
2017-11-09 07:11:11 +00:00
import json
2018-10-01 04:49:26 +00:00
class Popen(subprocess.Popen):
"""Patched Popen to prevent opening cmd window on Windows platform."""
def __init__(self, *args, **kwargs):
startupinfo = kwargs.get('startupinfo')
if is_win or True:
try:
startupinfo = startupinfo or subprocess.STARTUPINFO()
startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
except AttributeError:
pass
kwargs['startupinfo'] = startupinfo
2018-10-03 07:46:21 +00:00
if 'env' not in kwargs:
kwargs['env'] = os.environ.copy()
kwargs['env']['LANG'] = 'en-US' if is_win else 'en_US.UTF-8'
subprocess.Popen.__init__(self, *args, **kwargs)