Compare commits

...

14 Commits
4.0.0 ... 4.0.3

7 changed files with 85 additions and 28 deletions

View File

@ -3,6 +3,26 @@ History
------- -------
4.0.3 (2015-05-06)
++++++++++++++++++
- send cursorpos to wakatime cli
- upgrade wakatime cli to v4.0.10
4.0.2 (2015-05-06)
++++++++++++++++++
- only send heartbeats for the currently active buffer
4.0.1 (2015-05-06)
++++++++++++++++++
- ignore git temporary files
- don't send two write heartbeats within 2 seconds of eachother
4.0.0 (2015-04-12) 4.0.0 (2015-04-12)
++++++++++++++++++ ++++++++++++++++++

View File

@ -7,7 +7,7 @@ Website: https://wakatime.com/
===========================================================""" ==========================================================="""
__version__ = '4.0.0' __version__ = '4.0.3'
import sublime import sublime
@ -25,13 +25,13 @@ from subprocess import Popen
# globals # globals
ACTION_FREQUENCY = 2 HEARTBEAT_FREQUENCY = 2
ST_VERSION = int(sublime.version()) ST_VERSION = int(sublime.version())
PLUGIN_DIR = os.path.dirname(os.path.realpath(__file__)) PLUGIN_DIR = os.path.dirname(os.path.realpath(__file__))
API_CLIENT = os.path.join(PLUGIN_DIR, 'packages', 'wakatime', 'cli.py') API_CLIENT = os.path.join(PLUGIN_DIR, 'packages', 'wakatime', 'cli.py')
SETTINGS_FILE = 'WakaTime.sublime-settings' SETTINGS_FILE = 'WakaTime.sublime-settings'
SETTINGS = {} SETTINGS = {}
LAST_ACTION = { LAST_HEARTBEAT = {
'time': 0, 'time': 0,
'file': None, 'file': None,
'is_write': False, 'is_write': False,
@ -136,8 +136,10 @@ def obfuscate_apikey(command_list):
return cmd return cmd
def enough_time_passed(now, last_time): def enough_time_passed(now, last_heartbeat, is_write):
if now - last_time > ACTION_FREQUENCY * 60: if now - last_heartbeat['time'] > HEARTBEAT_FREQUENCY * 60:
return True
if is_write and now - last_heartbeat['time'] > 2:
return True return True
return False return False
@ -171,17 +173,27 @@ def find_project_from_folders(folders, current_file):
return os.path.basename(folder) if folder else None return os.path.basename(folder) if folder else None
def handle_action(view, is_write=False): 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_heartbeat(view, is_write=False):
window = view.window() window = view.window()
if window is not None: if window is not None:
target_file = view.file_name() target_file = view.file_name()
project = window.project_data() if hasattr(window, 'project_data') else None project = window.project_data() if hasattr(window, 'project_data') else None
folders = window.folders() folders = window.folders()
thread = SendActionThread(target_file, view, is_write=is_write, project=project, folders=folders) thread = SendHeartbeatThread(target_file, view, is_write=is_write, project=project, folders=folders)
thread.start() thread.start()
class SendActionThread(threading.Thread): class SendHeartbeatThread(threading.Thread):
def __init__(self, target_file, view, is_write=False, project=None, folders=None, force=False): def __init__(self, target_file, view, is_write=False, project=None, folders=None, force=False):
threading.Thread.__init__(self) threading.Thread.__init__(self)
@ -194,14 +206,15 @@ class SendActionThread(threading.Thread):
self.debug = SETTINGS.get('debug') self.debug = SETTINGS.get('debug')
self.api_key = SETTINGS.get('api_key', '') self.api_key = SETTINGS.get('api_key', '')
self.ignore = SETTINGS.get('ignore', []) self.ignore = SETTINGS.get('ignore', [])
self.last_action = LAST_ACTION.copy() self.last_heartbeat = LAST_HEARTBEAT.copy()
self.cursorpos = view.sel()[0].begin() if view.sel() else None
self.view = view self.view = view
def run(self): def run(self):
with self.lock: with self.lock:
if self.target_file: if self.target_file:
self.timestamp = time.time() self.timestamp = time.time()
if self.force or (self.is_write and not self.last_action['is_write']) or self.target_file != self.last_action['file'] or enough_time_passed(self.timestamp, self.last_action['time']): if self.force or self.target_file != self.last_heartbeat['file'] or enough_time_passed(self.timestamp, self.last_heartbeat, self.is_write):
self.send_heartbeat() self.send_heartbeat()
def send_heartbeat(self): def send_heartbeat(self):
@ -224,6 +237,8 @@ class SendActionThread(threading.Thread):
project_name = find_project_from_folders(self.folders, self.target_file) project_name = find_project_from_folders(self.folders, self.target_file)
if project_name: if project_name:
cmd.extend(['--project', project_name]) cmd.extend(['--project', project_name])
if self.cursorpos is not None:
cmd.extend(['--cursorpos', '{0}'.format(self.cursorpos)])
for pattern in self.ignore: for pattern in self.ignore:
cmd.extend(['--ignore', pattern]) cmd.extend(['--ignore', pattern])
if self.debug: if self.debug:
@ -243,15 +258,15 @@ class SendActionThread(threading.Thread):
def sent(self): def sent(self):
sublime.set_timeout(self.set_status_bar, 0) sublime.set_timeout(self.set_status_bar, 0)
sublime.set_timeout(self.set_last_action, 0) sublime.set_timeout(self.set_last_heartbeat, 0)
def set_status_bar(self): def set_status_bar(self):
if SETTINGS.get('status_bar_message'): if SETTINGS.get('status_bar_message'):
self.view.set_status('wakatime', 'WakaTime active {0}'.format(datetime.now().strftime('%I:%M %p'))) self.view.set_status('wakatime', 'WakaTime active {0}'.format(datetime.now().strftime('%I:%M %p')))
def set_last_action(self): def set_last_heartbeat(self):
global LAST_ACTION global LAST_HEARTBEAT
LAST_ACTION = { LAST_HEARTBEAT = {
'file': self.target_file, 'file': self.target_file,
'time': self.timestamp, 'time': self.timestamp,
'is_write': self.is_write, 'is_write': self.is_write,
@ -283,13 +298,15 @@ if ST_VERSION < 3000:
class WakatimeListener(sublime_plugin.EventListener): class WakatimeListener(sublime_plugin.EventListener):
def on_post_save(self, view): def on_post_save(self, view):
handle_action(view, is_write=True) handle_heartbeat(view, is_write=True)
def on_selection_modified(self, view): def on_selection_modified(self, view):
handle_action(view) if is_view_active(view):
handle_heartbeat(view)
def on_modified(self, view): def on_modified(self, view):
handle_action(view) if is_view_active(view):
handle_heartbeat(view)
class WakatimeDashboardCommand(sublime_plugin.ApplicationCommand): class WakatimeDashboardCommand(sublime_plugin.ApplicationCommand):

View File

@ -9,7 +9,7 @@
// Ignore files; Files (including absolute paths) that match one of these // Ignore files; Files (including absolute paths) that match one of these
// POSIX regular expressions will not be logged. // POSIX regular expressions will not be logged.
"ignore": ["^/tmp/", "^/etc/", "^/var/"], "ignore": ["^/tmp/", "^/etc/", "^/var/", "COMMIT_EDITMSG$", "PULLREQ_EDITMSG$", "MERGE_MSG$", "TAG_EDITMSG$"],
// Debug mode. Set to true for verbose logging. Defaults to false. // Debug mode. Set to true for verbose logging. Defaults to false.
"debug": false, "debug": false,

View File

@ -1,7 +1,7 @@
__title__ = 'wakatime' __title__ = 'wakatime'
__description__ = 'Common interface to the WakaTime api.' __description__ = 'Common interface to the WakaTime api.'
__url__ = 'https://github.com/wakatime/wakatime' __url__ = 'https://github.com/wakatime/wakatime'
__version_info__ = ('4', '0', '8') __version_info__ = ('4', '0', '10')
__version__ = '.'.join(__version_info__) __version__ = '.'.join(__version_info__)
__author__ = 'Alan Hamlett' __author__ = 'Alan Hamlett'
__author_email__ = 'alan@wakatime.com' __author_email__ = 'alan@wakatime.com'

View File

@ -147,6 +147,10 @@ def parseArguments(argv):
type=float, type=float,
help='optional floating-point unix epoch timestamp; '+ help='optional floating-point unix epoch timestamp; '+
'uses current time by default') 'uses current time by default')
parser.add_argument('--lineno', dest='lineno',
help='optional line number; current line being edited')
parser.add_argument('--cursorpos', dest='cursorpos',
help='optional cursor position in the current file')
parser.add_argument('--notfile', dest='notfile', action='store_true', parser.add_argument('--notfile', dest='notfile', action='store_true',
help='when set, will accept any value for the file. for example, '+ help='when set, will accept any value for the file. for example, '+
'a domain name or other item you want to log time towards.') 'a domain name or other item you want to log time towards.')
@ -322,6 +326,10 @@ def send_heartbeat(project=None, branch=None, stats={}, key=None, targetFile=Non
data['language'] = stats['language'] data['language'] = stats['language']
if stats.get('dependencies'): if stats.get('dependencies'):
data['dependencies'] = stats['dependencies'] data['dependencies'] = stats['dependencies']
if stats.get('lineno'):
data['lineno'] = stats['lineno']
if stats.get('cursorpos'):
data['cursorpos'] = stats['cursorpos']
if isWrite: if isWrite:
data['is_write'] = isWrite data['is_write'] = isWrite
if project: if project:
@ -424,7 +432,8 @@ def main(argv=None):
if os.path.isfile(args.targetFile) or args.notfile: if os.path.isfile(args.targetFile) or args.notfile:
stats = get_file_stats(args.targetFile, notfile=args.notfile) stats = get_file_stats(args.targetFile, notfile=args.notfile,
lineno=args.lineno, cursorpos=args.cursorpos)
project = None project = None
if not args.notfile: if not args.notfile:

View File

@ -9,6 +9,7 @@
:license: BSD, see LICENSE for more details. :license: BSD, see LICENSE for more details.
""" """
import inspect
import logging import logging
import os import os
import sys import sys
@ -47,14 +48,19 @@ class JsonFormatter(logging.Formatter):
def format(self, record): def format(self, record):
data = OrderedDict([ data = OrderedDict([
('now', self.formatTime(record, self.datefmt)), ('now', self.formatTime(record, self.datefmt)),
('version', self.version),
('plugin', self.plugin),
('time', self.timestamp),
('isWrite', self.isWrite),
('file', self.targetFile),
('level', record.levelname),
('message', record.msg),
]) ])
try:
data['package'] = inspect.stack()[9][0].f_globals.get('__package__')
data['lineno'] = inspect.stack()[9][2]
except:
pass
data['version'] = self.version
data['plugin'] = self.plugin
data['time'] = self.timestamp
data['isWrite'] = self.isWrite
data['file'] = self.targetFile
data['level'] = record.levelname
data['message'] = record.msg
if not self.plugin: if not self.plugin:
del data['plugin'] del data['plugin']
if not self.isWrite: if not self.isWrite:
@ -101,4 +107,5 @@ def setup_logging(args, version):
) )
handler.setFormatter(formatter) handler.setFormatter(formatter)
logger.addHandler(handler) logger.addHandler(handler)
logging.getLogger('py.warnings').addHandler(handler)
return logger return logger

View File

@ -86,12 +86,14 @@ def number_lines_in_file(file_name):
return lines return lines
def get_file_stats(file_name, notfile=False): def get_file_stats(file_name, notfile=False, lineno=None, cursorpos=None):
if notfile: if notfile:
stats = { stats = {
'language': None, 'language': None,
'dependencies': [], 'dependencies': [],
'lines': None, 'lines': None,
'lineno': lineno,
'cursorpos': cursorpos,
} }
else: else:
language, lexer = guess_language(file_name) language, lexer = guess_language(file_name)
@ -101,5 +103,7 @@ def get_file_stats(file_name, notfile=False):
'language': language, 'language': language,
'dependencies': dependencies, 'dependencies': dependencies,
'lines': number_lines_in_file(file_name), 'lines': number_lines_in_file(file_name),
'lineno': lineno,
'cursorpos': cursorpos,
} }
return stats return stats