# -*- coding: utf-8 -*- """ ========================================================== File: WakaTime.py Description: Automatic time tracking for Sublime Text 2 and 3. Maintainer: WakaTime License: BSD, see LICENSE for more details. Website: https://wakatime.com/ ===========================================================""" __version__ = '11.0.8' import sublime import sublime_plugin import json import os import platform import re import shutil import ssl import subprocess import sys import time import threading import traceback import webbrowser from subprocess import STDOUT, PIPE from zipfile import ZipFile try: import Queue as queue # py2 except ImportError: import queue # py3 try: from ConfigParser import SafeConfigParser as ConfigParser from ConfigParser import Error as ConfigParserError except ImportError: from configparser import ConfigParser, Error as ConfigParserError try: from urllib2 import Request, urlopen, HTTPError except ImportError: from urllib.request import Request, urlopen from urllib.error import HTTPError is_py2 = (sys.version_info[0] == 2) is_py3 = (sys.version_info[0] == 3) is_win = platform.system() == 'Windows' if is_py2: import codecs open = codecs.open def u(text): if text is None: return None if isinstance(text, unicode): # noqa: F821 return text try: return text.decode('utf-8') except: try: return text.decode(sys.getdefaultencoding()) except: try: return unicode(text) # noqa: F821 except: try: return text.decode('utf-8', 'replace') except: try: return unicode(str(text)) # noqa: F821 except: return unicode('') # noqa: F821 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.decode('utf-8', 'replace') else: raise Exception('Unsupported Python version: {0}.{1}.{2}'.format( sys.version_info[0], sys.version_info[1], sys.version_info[2], )) class Popen(subprocess.Popen): """Patched Popen to prevent opening cmd window on Windows platform.""" def __init__(self, *args, **kwargs): if is_win: startupinfo = kwargs.get('startupinfo') try: startupinfo = startupinfo or subprocess.STARTUPINFO() startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW except AttributeError: pass kwargs['startupinfo'] = startupinfo super(Popen, self).__init__(*args, **kwargs) # globals ST_VERSION = int(sublime.version()) HOME_FOLDER = os.path.realpath(os.environ.get('WAKATIME_HOME') or os.path.expanduser('~')) RESOURCES_FOLDER = os.path.join(HOME_FOLDER, '.wakatime') CONFIG_FILE = os.path.join(HOME_FOLDER, '.wakatime.cfg') INTERNAL_CONFIG_FILE = os.path.join(HOME_FOLDER, '.wakatime-internal.cfg') GITHUB_RELEASES_STABLE_URL = 'https://api.github.com/repos/wakatime/wakatime-cli/releases/latest' GITHUB_DOWNLOAD_PREFIX = 'https://github.com/wakatime/wakatime-cli/releases/download' SETTINGS_FILE = 'WakaTime.sublime-settings' SETTINGS = {} LAST_HEARTBEAT = { 'time': 0, 'file': None, 'is_write': False, } LAST_HEARTBEAT_SENT_AT = 0 LAST_FETCH_TODAY_CODING_TIME = 0 FETCH_TODAY_DEBOUNCE_COUNTER = 0 FETCH_TODAY_DEBOUNCE_SECONDS = 60 LATEST_CLI_VERSION = None WAKATIME_CLI_LOCATION = None HEARTBEATS = queue.Queue() HEARTBEAT_FREQUENCY = 2 # minutes between logging heartbeat when editing same file SEND_BUFFER_SECONDS = 30 # seconds between sending buffered heartbeats to API # Log Levels DEBUG = 'DEBUG' INFO = 'INFO' WARNING = 'WARNING' ERROR = 'ERROR' def parseConfigFile(configFile): """Returns a configparser.SafeConfigParser instance with configs read from the config file. Default location of the config file is at ~/.wakatime.cfg. """ kwargs = {} if is_py2 else {'strict': False} configs = ConfigParser(**kwargs) try: with open(configFile, 'r', encoding='utf-8') as fh: try: if is_py2: configs.readfp(fh) else: configs.read_file(fh) return configs except ConfigParserError: log(ERROR, traceback.format_exc()) return None except IOError: log(DEBUG, "Error: Could not read from config file {0}\n".format(configFile)) return configs class ApiKey(object): _key = None def read(self): if self._key: return self._key key = SETTINGS.get('api_key') if key: self._key = key return self._key configs = None try: configs = parseConfigFile(CONFIG_FILE) if configs: if configs.has_option('settings', 'api_key'): key = configs.get('settings', 'api_key') if key: self._key = key return self._key except: pass key = self.api_key_from_vault_cmd(configs) if key: self._key = key return self._key return self._key def api_key_from_vault_cmd(self, configs): vault_cmd = SETTINGS.get('api_key_vault_cmd') if not vault_cmd and configs: try: if configs.has_option('settings', 'api_key_vault_cmd'): vault_cmd = configs.get('settings', 'api_key_vault_cmd') except: pass if not vault_cmd or not vault_cmd.strip(): return None try: process = Popen(vault_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True) stdout, stderr = process.communicate() retcode = process.poll() if retcode: log(ERROR, 'Vault command error ({retcode}): {stderr}'.format(retcode=retcode, stderr=u(stderr))) return None return stdout.strip() or None except: log(ERROR, traceback.format_exc()) return None def write(self, key): global SETTINGS self._key = key SETTINGS.set('api_key', str(key)) sublime.save_settings(SETTINGS_FILE) APIKEY = ApiKey() def set_timeout(callback, seconds): """Runs the callback after the given seconds delay. If this is Sublime Text 3, runs the callback on an alternate thread. If this is Sublime Text 2, runs the callback in the main thread. """ milliseconds = int(seconds * 1000) try: sublime.set_timeout_async(callback, milliseconds) except AttributeError: sublime.set_timeout(callback, milliseconds) def log(lvl, message, *args, **kwargs): try: if lvl == DEBUG and not SETTINGS.get('debug'): return msg = message if len(args) > 0: msg = message.format(*args) elif len(kwargs) > 0: msg = message.format(**kwargs) try: print('[WakaTime] [{lvl}] {msg}'.format(lvl=lvl, msg=msg)) except UnicodeDecodeError: print(u('[WakaTime] [{lvl}] {msg}').format(lvl=lvl, msg=u(msg))) except RuntimeError: set_timeout(lambda: log(lvl, message, *args, **kwargs), 0) def update_status_bar(status=None, debounced=False, msg=None): """Updates the status bar.""" global LAST_FETCH_TODAY_CODING_TIME, FETCH_TODAY_DEBOUNCE_COUNTER try: if not msg and SETTINGS.get('status_bar_message') is not False and SETTINGS.get('status_bar_enabled'): if SETTINGS.get('status_bar_coding_activity') and status == 'OK': if debounced: FETCH_TODAY_DEBOUNCE_COUNTER -= 1 if debounced or not LAST_FETCH_TODAY_CODING_TIME: now = int(time.time()) if LAST_FETCH_TODAY_CODING_TIME and (FETCH_TODAY_DEBOUNCE_COUNTER > 0 or LAST_FETCH_TODAY_CODING_TIME > now - FETCH_TODAY_DEBOUNCE_SECONDS): return LAST_FETCH_TODAY_CODING_TIME = now FetchStatusBarCodingTime().start() return else: FETCH_TODAY_DEBOUNCE_COUNTER += 1 set_timeout(lambda: update_status_bar(status, debounced=True), FETCH_TODAY_DEBOUNCE_SECONDS) return else: msg = 'WakaTime: {status}'.format(status=status) if msg: active_window = sublime.active_window() if active_window: for view in active_window.views(): view.set_status('wakatime', msg) except RuntimeError: set_timeout(lambda: update_status_bar(status=status, debounced=debounced, msg=msg), 0) class FetchStatusBarCodingTime(threading.Thread): def __init__(self): threading.Thread.__init__(self) self.debug = SETTINGS.get('debug') self.api_key = APIKEY.read() or '' self.proxy = SETTINGS.get('proxy') def run(self): if not self.api_key: log(DEBUG, 'Missing WakaTime API key.') return if not isCliInstalled(): return ua = 'sublime/%d sublime-wakatime/%s' % (ST_VERSION, __version__) cmd = [ getCliLocation(), '--today', '--key', str(bytes.decode(self.api_key.encode('utf8'))), '--plugin', ua, ] if self.debug: cmd.append('--verbose') if self.proxy: cmd.extend(['--proxy', self.proxy]) log(DEBUG, ' '.join(obfuscate_apikey(cmd))) try: process = Popen(cmd, stdout=PIPE, stderr=STDOUT) output, err = process.communicate() output = u(output) if output: output = output.strip() retcode = process.poll() if not retcode and output: msg = 'Today: {output}'.format(output=output) update_status_bar(msg=msg) else: log(DEBUG, 'wakatime-core today exited with status: {0}'.format(retcode)) if output: log(DEBUG, u('wakatime-core today output: {0}').format(output)) except: pass def prompt_api_key(): if APIKEY.read(): return True window = sublime.active_window() if window: def got_key(text): if text: APIKEY.write(text) window.show_input_panel('[WakaTime] Enter your wakatime.com api key:', '', got_key, None, None) return True else: log(ERROR, 'Could not prompt for api key because no window found.') return False def obfuscate_apikey(command_list): cmd = list(command_list) apikey_index = None for num in range(len(cmd)): if cmd[num] == '--key': apikey_index = num + 1 break if apikey_index is not None and apikey_index < len(cmd): cmd[apikey_index] = 'XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXX' + cmd[apikey_index][-4:] return cmd def enough_time_passed(now, is_write): if now - LAST_HEARTBEAT['time'] > HEARTBEAT_FREQUENCY * 60: return True if is_write and now - LAST_HEARTBEAT['time'] > 2: return True return False def find_folder_containing_file(folders, current_file): """Returns absolute path to folder containing the file. """ parent_folder = None current_folder = current_file while True: for folder in folders: if os.path.realpath(os.path.dirname(current_folder)) == os.path.realpath(folder): parent_folder = folder break if parent_folder is not None: break if not current_folder or os.path.dirname(current_folder) == current_folder: break current_folder = os.path.dirname(current_folder) return parent_folder def find_project_from_folders(folders, current_file): """Find project name from open folders. """ folder = find_folder_containing_file(folders, current_file) return os.path.basename(folder) if folder else None def is_view_active(view): if view: active_window = sublime.active_window() if active_window: active_view = active_window.active_view() if active_view: return active_view.buffer_id() == view.buffer_id() return False def handle_activity(view, is_write=False): window = view.window() if window is not None: entity = view.file_name() if entity: timestamp = time.time() last_file = LAST_HEARTBEAT['file'] if entity != last_file or enough_time_passed(timestamp, is_write): project = window.project_data() if hasattr(window, 'project_data') else None folders = window.folders() append_heartbeat(entity, timestamp, is_write, view, project, folders) def append_heartbeat(entity, timestamp, is_write, view, project, folders): global LAST_HEARTBEAT # add this heartbeat to queue heartbeat = { 'entity': entity, 'timestamp': timestamp, 'is_write': is_write, 'project': project, 'folders': folders, 'lines_in_file': view.rowcol(view.size())[0] + 1, } selections = view.sel() if selections and len(selections) > 0: rowcol = view.rowcol(selections[0].begin()) row, col = rowcol[0] + 1, rowcol[1] + 1 heartbeat['lineno'] = row heartbeat['cursorpos'] = col HEARTBEATS.put_nowait(heartbeat) # make this heartbeat the LAST_HEARTBEAT LAST_HEARTBEAT = { 'file': entity, 'time': timestamp, 'is_write': is_write, } # process the queue of heartbeats in the future set_timeout(lambda: process_queue(timestamp), SEND_BUFFER_SECONDS) def process_queue(timestamp): global LAST_HEARTBEAT_SENT_AT if not isCliInstalled(): return # Prevent sending heartbeats more often than SEND_BUFFER_SECONDS now = int(time.time()) if timestamp != LAST_HEARTBEAT['time'] and LAST_HEARTBEAT_SENT_AT > now - SEND_BUFFER_SECONDS: return LAST_HEARTBEAT_SENT_AT = now try: heartbeat = HEARTBEATS.get_nowait() except queue.Empty: return has_extra_heartbeats = False extra_heartbeats = [] try: while True: extra_heartbeats.append(HEARTBEATS.get_nowait()) has_extra_heartbeats = True except queue.Empty: pass thread = SendHeartbeatsThread(heartbeat) if has_extra_heartbeats: thread.add_extra_heartbeats(extra_heartbeats) thread.start() class SendHeartbeatsThread(threading.Thread): """Non-blocking thread for sending heartbeats to api. """ def __init__(self, heartbeat): threading.Thread.__init__(self) self.debug = SETTINGS.get('debug') self.api_key = APIKEY.read() or '' self.ignore = SETTINGS.get('ignore', []) self.include = SETTINGS.get('include', []) self.hidefilenames = SETTINGS.get('hidefilenames') self.proxy = SETTINGS.get('proxy') self.heartbeat = heartbeat self.has_extra_heartbeats = False def add_extra_heartbeats(self, extra_heartbeats): self.has_extra_heartbeats = True self.extra_heartbeats = extra_heartbeats def run(self): """Running in background thread.""" self.send_heartbeats() def build_heartbeat(self, entity=None, timestamp=None, is_write=None, lineno=None, cursorpos=None, lines_in_file=None, project=None, folders=None): """Returns a dict for passing to wakatime-cli as arguments.""" heartbeat = { 'entity': entity, 'timestamp': timestamp, 'is_write': is_write, } if project and project.get('name'): heartbeat['alternate_project'] = project.get('name') elif folders: project_name = find_project_from_folders(folders, entity) if project_name: heartbeat['alternate_project'] = project_name if lineno is not None: heartbeat['lineno'] = lineno if cursorpos is not None: heartbeat['cursorpos'] = cursorpos if lines_in_file is not None: heartbeat['lines-in-file'] = lines_in_file return heartbeat def send_heartbeats(self): heartbeat = self.build_heartbeat(**self.heartbeat) ua = 'sublime/%d sublime-wakatime/%s' % (ST_VERSION, __version__) cmd = [ getCliLocation(), '--entity', heartbeat['entity'], '--time', str('%f' % heartbeat['timestamp']), '--plugin', ua, ] if self.api_key: cmd.extend(['--key', str(bytes.decode(self.api_key.encode('utf8')))]) if heartbeat['is_write']: cmd.append('--write') if heartbeat.get('alternate_project'): cmd.extend(['--alternate-project', heartbeat['alternate_project']]) if heartbeat.get('lineno') is not None: cmd.extend(['--lineno', '{0}'.format(heartbeat['lineno'])]) if heartbeat.get('cursorpos') is not None: cmd.extend(['--cursorpos', '{0}'.format(heartbeat['cursorpos'])]) if heartbeat.get('lines_in_file') is not None: cmd.extend(['--lines-in-file', '{0}'.format(heartbeat['lines_in_file'])]) for pattern in self.ignore: cmd.extend(['--exclude', pattern]) for pattern in self.include: cmd.extend(['--include', pattern]) if self.debug: cmd.append('--verbose') if self.hidefilenames: cmd.append('--hidefilenames') if self.proxy: cmd.extend(['--proxy', self.proxy]) if self.has_extra_heartbeats: cmd.append('--extra-heartbeats') stdin = PIPE extra_heartbeats = json.dumps([self.build_heartbeat(**x) for x in self.extra_heartbeats]) inp = "{0}\n".format(extra_heartbeats).encode('utf-8') else: extra_heartbeats = None stdin = None inp = None log(DEBUG, ' '.join(obfuscate_apikey(cmd))) try: process = Popen(cmd, stdin=stdin, stdout=PIPE, stderr=STDOUT) output, _err = process.communicate(input=inp) retcode = process.poll() if (not retcode or retcode == 102 or retcode == 112) and not output: self.sent() else: update_status_bar('Error') if retcode: log(DEBUG if retcode == 102 or retcode == 112 else ERROR, 'wakatime-core exited with status: {0}'.format(retcode)) if output: log(ERROR, u('wakatime-core output: {0}').format(output)) except: log(ERROR, u(sys.exc_info()[1])) update_status_bar('Error') def sent(self): update_status_bar('OK') def plugin_loaded(): global SETTINGS SETTINGS = sublime.load_settings(SETTINGS_FILE) log(INFO, 'Initializing WakaTime plugin v%s' % __version__) update_status_bar('Initializing...') UpdateCLI().start() after_loaded() def after_loaded(): if not prompt_api_key(): set_timeout(after_loaded, 0.5) update_status_bar('OK') # need to call plugin_loaded because only ST3 will auto-call it if ST_VERSION < 3000: plugin_loaded() class WakatimeListener(sublime_plugin.EventListener): def on_post_save(self, view): handle_activity(view, is_write=True) def on_selection_modified(self, view): if is_view_active(view): handle_activity(view) def on_modified(self, view): if is_view_active(view): handle_activity(view) class WakatimeDashboardCommand(sublime_plugin.ApplicationCommand): def run(self): webbrowser.open_new_tab('https://wakatime.com/dashboard') class UpdateCLI(threading.Thread): """Non-blocking thread for downloading latest wakatime-cli from GitHub. """ def run(self): if isCliLatest(): return log(INFO, 'Downloading wakatime-cli...') if os.path.isdir(os.path.join(RESOURCES_FOLDER, 'wakatime-cli')): shutil.rmtree(os.path.join(RESOURCES_FOLDER, 'wakatime-cli')) if not os.path.exists(RESOURCES_FOLDER): os.makedirs(RESOURCES_FOLDER) try: url = cliDownloadUrl() log(DEBUG, 'Downloading wakatime-cli from {url}'.format(url=url)) zip_file = os.path.join(RESOURCES_FOLDER, 'wakatime-cli.zip') download(url, zip_file) if isCliInstalled(): try: os.remove(getCliLocation()) except: log(DEBUG, traceback.format_exc()) log(INFO, 'Extracting wakatime-cli...') with ZipFile(zip_file) as zf: zf.extractall(RESOURCES_FOLDER) if not is_win: os.chmod(getCliLocation(), 509) # 755 try: os.remove(os.path.join(RESOURCES_FOLDER, 'wakatime-cli.zip')) except: log(DEBUG, traceback.format_exc()) except: log(DEBUG, traceback.format_exc()) createSymlink() log(INFO, 'Finished extracting wakatime-cli.') def getCliLocation(): global WAKATIME_CLI_LOCATION if not WAKATIME_CLI_LOCATION: binary = 'wakatime-cli-{osname}-{arch}{ext}'.format( osname=platform.system().lower(), arch=architecture(), ext='.exe' if is_win else '', ) WAKATIME_CLI_LOCATION = os.path.join(RESOURCES_FOLDER, binary) return WAKATIME_CLI_LOCATION def architecture(): arch = platform.machine() or platform.processor() if arch == 'armv7l': return 'arm' if arch == 'aarch64': return 'arm64' if 'arm' in arch: return 'arm64' if sys.maxsize > 2**32 else 'arm' return 'amd64' if sys.maxsize > 2**32 else '386' def isCliInstalled(): return os.path.exists(getCliLocation()) def isCliLatest(): if not isCliInstalled(): return False args = [getCliLocation(), '--version'] try: stdout, stderr = Popen(args, stdout=PIPE, stderr=PIPE).communicate() except: return False stdout = (stdout or b'') + (stderr or b'') localVer = extractVersion(stdout.decode('utf-8')) if not localVer: log(DEBUG, 'Local wakatime-cli version not found.') return False log(INFO, 'Current wakatime-cli version is %s' % localVer) log(INFO, 'Checking for updates to wakatime-cli...') remoteVer = getLatestCliVersion() if not remoteVer: return True if remoteVer == localVer: log(INFO, 'wakatime-cli is up to date.') return True log(INFO, 'Found an updated wakatime-cli %s' % remoteVer) return False def getLatestCliVersion(): global LATEST_CLI_VERSION if LATEST_CLI_VERSION: return LATEST_CLI_VERSION configs, last_modified, last_version = None, None, None try: configs = parseConfigFile(INTERNAL_CONFIG_FILE) if configs: last_modified, last_version = lastModifiedAndVersion(configs) except: log(DEBUG, traceback.format_exc()) try: headers, contents, code = request(GITHUB_RELEASES_STABLE_URL, last_modified=last_modified) log(DEBUG, 'GitHub API Response {0}'.format(code)) if code == 304: LATEST_CLI_VERSION = last_version return last_version data = json.loads(contents.decode('utf-8')) ver = data['tag_name'] log(DEBUG, 'Latest wakatime-cli version from GitHub: {0}'.format(ver)) if configs: last_modified = headers.get('Last-Modified') if not configs.has_section('internal'): configs.add_section('internal') configs.set('internal', 'cli_version', ver) configs.set('internal', 'cli_version_last_modified', last_modified) with open(INTERNAL_CONFIG_FILE, 'w', encoding='utf-8') as fh: configs.write(fh) LATEST_CLI_VERSION = ver return ver except: log(DEBUG, traceback.format_exc()) return None def lastModifiedAndVersion(configs): last_modified, last_version = None, None if configs.has_option('internal', 'cli_version'): last_version = configs.get('internal', 'cli_version') if last_version and configs.has_option('internal', 'cli_version_last_modified'): last_modified = configs.get('internal', 'cli_version_last_modified') if last_modified and last_version and extractVersion(last_version): return last_modified, last_version return None, None def extractVersion(text): pattern = re.compile(r"([0-9]+\.[0-9]+\.[0-9]+)") match = pattern.search(text) if match: return 'v{ver}'.format(ver=match.group(1)) return None def cliDownloadUrl(): osname = platform.system().lower() arch = architecture() validCombinations = [ 'darwin-amd64', 'darwin-arm64', 'freebsd-386', 'freebsd-amd64', 'freebsd-arm', 'linux-386', 'linux-amd64', 'linux-arm', 'linux-arm64', 'netbsd-386', 'netbsd-amd64', 'netbsd-arm', 'openbsd-386', 'openbsd-amd64', 'openbsd-arm', 'openbsd-arm64', 'windows-386', 'windows-amd64', 'windows-arm64', ] check = '{osname}-{arch}'.format(osname=osname, arch=arch) if check not in validCombinations: reportMissingPlatformSupport(osname, arch) version = getLatestCliVersion() return '{prefix}/{version}/wakatime-cli-{osname}-{arch}.zip'.format( prefix=GITHUB_DOWNLOAD_PREFIX, version=version, osname=osname, arch=arch, ) def reportMissingPlatformSupport(osname, arch): url = 'https://api.wakatime.com/api/v1/cli-missing?osname={osname}&architecture={arch}&plugin=sublime'.format( osname=osname, arch=arch, ) request(url) def request(url, last_modified=None): req = Request(url) req.add_header('User-Agent', 'github.com/wakatime/sublime-wakatime') proxy = SETTINGS.get('proxy') if proxy: req.set_proxy(proxy, 'https') if last_modified: req.add_header('If-Modified-Since', last_modified) try: resp = urlopen(req) headers = dict(resp.getheaders()) if is_py2 else resp.headers return headers, resp.read(), resp.getcode() except HTTPError as err: if err.code == 304: return None, None, 304 if is_py2: with SSLCertVerificationDisabled(): try: resp = urlopen(req) headers = dict(resp.getheaders()) if is_py2 else resp.headers return headers, resp.read(), resp.getcode() except HTTPError as err2: if err2.code == 304: return None, None, 304 log(DEBUG, err.read().decode()) log(DEBUG, err2.read().decode()) raise except IOError: raise log(DEBUG, err.read().decode()) raise except IOError: if is_py2: with SSLCertVerificationDisabled(): try: resp = urlopen(url) headers = dict(resp.getheaders()) if is_py2 else resp.headers return headers, resp.read(), resp.getcode() except HTTPError as err: if err.code == 304: return None, None, 304 log(DEBUG, err.read().decode()) raise except IOError: raise raise def download(url, filePath): req = Request(url) req.add_header('User-Agent', 'github.com/wakatime/sublime-wakatime') proxy = SETTINGS.get('proxy') if proxy: req.set_proxy(proxy, 'https') with open(filePath, 'wb') as fh: try: resp = urlopen(req) fh.write(resp.read()) except HTTPError as err: if err.code == 304: return None, None, 304 if is_py2: with SSLCertVerificationDisabled(): try: resp = urlopen(req) fh.write(resp.read()) return except HTTPError as err2: log(DEBUG, err.read().decode()) log(DEBUG, err2.read().decode()) raise except IOError: raise log(DEBUG, err.read().decode()) raise except IOError: if is_py2: with SSLCertVerificationDisabled(): try: resp = urlopen(url) fh.write(resp.read()) return except HTTPError as err: log(DEBUG, err.read().decode()) raise except IOError: raise raise def is_symlink(path): try: return os.is_symlink(path) except: return False def createSymlink(): link = os.path.join(RESOURCES_FOLDER, 'wakatime-cli') if is_win: link = link + '.exe' elif os.path.exists(link) and is_symlink(link): return # don't re-create symlink on Unix-like platforms try: os.symlink(getCliLocation(), link) except: try: shutil.copy2(getCliLocation(), link) if not is_win: os.chmod(link, 509) # 755 except: log(WARNING, traceback.format_exc()) class SSLCertVerificationDisabled(object): def __enter__(self): self.original_context = ssl._create_default_https_context ssl._create_default_https_context = ssl._create_unverified_context def __exit__(self, *args, **kwargs): ssl._create_default_https_context = self.original_context