make sure python --version outputs a version number
This commit is contained in:
parent
4223f3575f
commit
a97792c23c
1 changed files with 66 additions and 15 deletions
79
WakaTime.py
79
WakaTime.py
|
@ -23,7 +23,7 @@ import urllib
|
||||||
import webbrowser
|
import webbrowser
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from zipfile import ZipFile
|
from zipfile import ZipFile
|
||||||
from subprocess import Popen
|
from subprocess import Popen, STDOUT, PIPE
|
||||||
try:
|
try:
|
||||||
import _winreg as winreg # py2
|
import _winreg as winreg # py2
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
@ -33,6 +33,49 @@ except ImportError:
|
||||||
winreg = None
|
winreg = None
|
||||||
|
|
||||||
|
|
||||||
|
is_py2 = (sys.version_info[0] == 2)
|
||||||
|
is_py3 = (sys.version_info[0] == 3)
|
||||||
|
|
||||||
|
if is_py2:
|
||||||
|
def u(text):
|
||||||
|
if text is None:
|
||||||
|
return None
|
||||||
|
try:
|
||||||
|
return text.decode('utf-8')
|
||||||
|
except:
|
||||||
|
try:
|
||||||
|
return text.decode(sys.getdefaultencoding())
|
||||||
|
except:
|
||||||
|
try:
|
||||||
|
return unicode(text)
|
||||||
|
except:
|
||||||
|
return text
|
||||||
|
|
||||||
|
elif is_py3:
|
||||||
|
def u(text):
|
||||||
|
if text is None:
|
||||||
|
return None
|
||||||
|
if isinstance(text, bytes):
|
||||||
|
try:
|
||||||
|
return text.decode('utf-8')
|
||||||
|
except:
|
||||||
|
try:
|
||||||
|
return text.decode(sys.getdefaultencoding())
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
try:
|
||||||
|
return str(text)
|
||||||
|
except:
|
||||||
|
return text
|
||||||
|
|
||||||
|
else:
|
||||||
|
raise Exception('Unsupported Python version: {0}.{1}.{2}'.format(
|
||||||
|
sys.version_info[0],
|
||||||
|
sys.version_info[1],
|
||||||
|
sys.version_info[2],
|
||||||
|
))
|
||||||
|
|
||||||
|
|
||||||
# globals
|
# globals
|
||||||
HEARTBEAT_FREQUENCY = 2
|
HEARTBEAT_FREQUENCY = 2
|
||||||
ST_VERSION = int(sublime.version())
|
ST_VERSION = int(sublime.version())
|
||||||
|
@ -160,7 +203,7 @@ def python_binary():
|
||||||
def set_python_binary_location(path):
|
def set_python_binary_location(path):
|
||||||
global PYTHON_LOCATION
|
global PYTHON_LOCATION
|
||||||
PYTHON_LOCATION = path
|
PYTHON_LOCATION = path
|
||||||
log(DEBUG, 'Python Binary Found: {0}'.format(path))
|
log(DEBUG, 'Found Python at: {0}'.format(path))
|
||||||
|
|
||||||
|
|
||||||
def find_python_from_registry(location, reg=None):
|
def find_python_from_registry(location, reg=None):
|
||||||
|
@ -221,23 +264,31 @@ def find_python_from_registry(location, reg=None):
|
||||||
return val
|
return val
|
||||||
|
|
||||||
|
|
||||||
def find_python_in_folder(folder):
|
def find_python_in_folder(folder, headless=True):
|
||||||
path = 'pythonw'
|
pattern = re.compile(r'\d+\.\d+')
|
||||||
if folder is not None:
|
|
||||||
path = os.path.realpath(os.path.join(folder, 'pythonw'))
|
|
||||||
try:
|
|
||||||
Popen([path, '--version'])
|
|
||||||
return path
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
path = 'python'
|
path = 'python'
|
||||||
if folder is not None:
|
if folder is not None:
|
||||||
path = os.path.realpath(os.path.join(folder, 'python'))
|
path = os.path.realpath(os.path.join(folder, 'python'))
|
||||||
|
if headless:
|
||||||
|
path = path + 'w'
|
||||||
|
log(DEBUG, 'Looking for Python at: {0}'.format(path))
|
||||||
try:
|
try:
|
||||||
Popen([path, '--version'])
|
process = Popen([path, '--version'], stdout=PIPE, stderr=STDOUT)
|
||||||
|
output, err = process.communicate()
|
||||||
|
output = u(output).strip()
|
||||||
|
retcode = process.poll()
|
||||||
|
log(DEBUG, 'Python Version Output: {0}'.format(output))
|
||||||
|
if not retcode and pattern.search(output):
|
||||||
return path
|
return path
|
||||||
except:
|
except:
|
||||||
pass
|
log(DEBUG, 'Python Version Output: {0}'.format(u(sys.exc_info()[1])))
|
||||||
|
|
||||||
|
if headless:
|
||||||
|
path = find_python_in_folder(folder, headless=False)
|
||||||
|
if path is not None:
|
||||||
|
return path
|
||||||
|
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
@ -366,7 +417,7 @@ class SendHeartbeatThread(threading.Thread):
|
||||||
cmd.insert(0, python_binary())
|
cmd.insert(0, python_binary())
|
||||||
log(DEBUG, ' '.join(obfuscate_apikey(cmd)))
|
log(DEBUG, ' '.join(obfuscate_apikey(cmd)))
|
||||||
if platform.system() == 'Windows':
|
if platform.system() == 'Windows':
|
||||||
Popen(cmd, shell=False)
|
Popen(cmd)
|
||||||
else:
|
else:
|
||||||
with open(os.path.join(os.path.expanduser('~'), '.wakatime.log'), 'a') as stderr:
|
with open(os.path.join(os.path.expanduser('~'), '.wakatime.log'), 'a') as stderr:
|
||||||
Popen(cmd, stderr=stderr)
|
Popen(cmd, stderr=stderr)
|
||||||
|
|
Loading…
Reference in a new issue