use common resources folder location

This commit is contained in:
Alan Hamlett 2016-05-12 12:07:24 +02:00
parent 77c2041ad3
commit 3b2520afa9

View File

@ -140,6 +140,13 @@ def log(lvl, message, *args, **kwargs):
set_timeout(lambda: log(lvl, message, *args, **kwargs), 0) set_timeout(lambda: log(lvl, message, *args, **kwargs), 0)
def resources_folder():
if platform.system() == 'Windows':
return os.path.join(os.getenv('APPDATA'), 'WakaTime')
else:
return os.path.join(os.path.expanduser('~'), '.wakatime')
def update_status_bar(status): def update_status_bar(status):
"""Updates the status bar.""" """Updates the status bar."""
@ -157,7 +164,7 @@ def update_status_bar(status):
set_timeout(lambda: update_status_bar(status), 0) set_timeout(lambda: update_status_bar(status), 0)
def createConfigFile(): def create_config_file():
"""Creates the .wakatime.cfg INI file in $HOME directory, if it does """Creates the .wakatime.cfg INI file in $HOME directory, if it does
not already exist. not already exist.
""" """
@ -178,7 +185,7 @@ def createConfigFile():
def prompt_api_key(): def prompt_api_key():
global SETTINGS global SETTINGS
createConfigFile() create_config_file()
default_key = '' default_key = ''
try: try:
@ -211,7 +218,7 @@ def python_binary():
# look for python in PATH and common install locations # look for python in PATH and common install locations
paths = [ paths = [
os.path.join(os.path.expanduser('~'), '.wakatime', 'python'), os.path.join(resources_folder(), 'python'),
None, None,
'/', '/',
'/usr/local/bin/', '/usr/local/bin/',
@ -572,10 +579,10 @@ class DownloadPython(threading.Thread):
arch=arch, arch=arch,
) )
if not os.path.exists(os.path.join(os.path.expanduser('~'), '.wakatime')): if not os.path.exists(resources_folder()):
os.makedirs(os.path.join(os.path.expanduser('~'), '.wakatime')) os.makedirs(resources_folder())
zip_file = os.path.join(os.path.expanduser('~'), '.wakatime', 'python.zip') zip_file = os.path.join(resources_folder(), 'python.zip')
try: try:
urllib.urlretrieve(url, zip_file) urllib.urlretrieve(url, zip_file)
except AttributeError: except AttributeError:
@ -583,7 +590,7 @@ class DownloadPython(threading.Thread):
log(INFO, 'Extracting Python...') log(INFO, 'Extracting Python...')
with ZipFile(zip_file) as zf: with ZipFile(zip_file) as zf:
path = os.path.join(os.path.expanduser('~'), '.wakatime', 'python') path = os.path.join(resources_folder(), 'python')
zf.extractall(path) zf.extractall(path)
try: try: