sublime-rana/packages/wakatime/offlinequeue.py

129 lines
4.1 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
"""
2015-05-12 22:03:23 +00:00
wakatime.offlinequeue
~~~~~~~~~~~~~~~~~~~~~
2015-05-12 22:03:23 +00:00
Queue for saving heartbeats while offline.
:copyright: (c) 2014 Alan Hamlett.
:license: BSD, see LICENSE for more details.
"""
import logging
import os
from time import sleep
2014-06-09 20:18:30 +00:00
try:
import sqlite3
HAS_SQL = True
2015-09-29 10:11:25 +00:00
except ImportError: # pragma: nocover
2014-06-09 20:18:30 +00:00
HAS_SQL = False
2015-08-25 07:42:37 +00:00
from .compat import u
log = logging.getLogger('WakaTime')
class Queue(object):
2015-09-29 10:11:25 +00:00
db_file = os.path.join(os.path.expanduser('~'), '.wakatime.db')
table_name = 'heartbeat_1'
def get_db_file(self):
return self.db_file
def connect(self):
2015-09-29 10:11:25 +00:00
conn = sqlite3.connect(self.get_db_file())
c = conn.cursor()
2015-09-29 10:11:25 +00:00
c.execute('''CREATE TABLE IF NOT EXISTS {0} (
entity text,
type text,
time real,
project text,
branch text,
is_write integer,
2014-12-22 07:01:09 +00:00
stats text,
misc text,
plugin text)
2015-09-29 10:11:25 +00:00
'''.format(self.table_name))
return (conn, c)
2014-12-22 07:01:09 +00:00
def push(self, data, stats, plugin, misc=None):
2015-09-29 10:11:25 +00:00
if not HAS_SQL: # pragma: nocover
2014-06-09 20:18:30 +00:00
return
2014-05-26 21:06:36 +00:00
try:
conn, c = self.connect()
2014-12-22 07:01:09 +00:00
heartbeat = {
2015-09-29 10:11:25 +00:00
'entity': u(data.get('entity')),
'type': u(data.get('type')),
2014-05-26 21:06:36 +00:00
'time': data.get('time'),
2015-08-25 07:42:37 +00:00
'project': u(data.get('project')),
'branch': u(data.get('branch')),
2014-05-26 21:06:36 +00:00
'is_write': 1 if data.get('is_write') else 0,
2015-08-25 07:42:37 +00:00
'stats': u(stats),
'misc': u(misc),
'plugin': u(plugin),
2014-05-26 21:06:36 +00:00
}
2015-09-29 10:11:25 +00:00
c.execute('INSERT INTO {0} VALUES (:entity,:type,:time,:project,:branch,:is_write,:stats,:misc,:plugin)'.format(self.table_name), heartbeat)
2014-05-26 21:06:36 +00:00
conn.commit()
conn.close()
2014-06-09 20:18:30 +00:00
except sqlite3.Error:
2016-06-17 08:17:29 +00:00
log.traceback()
def pop(self):
2015-09-29 10:11:25 +00:00
if not HAS_SQL: # pragma: nocover
2014-06-09 20:18:30 +00:00
return None
tries = 3
wait = 0.1
2014-12-22 07:01:09 +00:00
heartbeat = None
2014-05-26 21:06:36 +00:00
try:
conn, c = self.connect()
2014-06-09 20:18:30 +00:00
except sqlite3.Error:
2016-06-17 08:17:29 +00:00
log.traceback('debug')
2014-05-26 21:06:36 +00:00
return None
loop = True
while loop and tries > -1:
try:
c.execute('BEGIN IMMEDIATE')
2015-09-29 10:11:25 +00:00
c.execute('SELECT * FROM {0} LIMIT 1'.format(self.table_name))
row = c.fetchone()
if row is not None:
2014-05-26 22:01:32 +00:00
values = []
clauses = []
index = 0
2015-09-29 10:11:25 +00:00
for row_name in ['entity', 'type', 'time', 'project', 'branch', 'is_write']:
2014-05-26 22:01:32 +00:00
if row[index] is not None:
clauses.append('{0}=?'.format(row_name))
2015-08-25 18:20:12 +00:00
values.append(row[index])
2015-09-29 10:11:25 +00:00
else: # pragma: nocover
2014-05-26 22:01:32 +00:00
clauses.append('{0} IS NULL'.format(row_name))
index += 1
if len(values) > 0:
2015-09-29 10:11:25 +00:00
c.execute('DELETE FROM {0} WHERE {1}'.format(self.table_name, ' AND '.join(clauses)), values)
else: # pragma: nocover
c.execute('DELETE FROM {0} WHERE {1}'.format(self.table_name, ' AND '.join(clauses)))
conn.commit()
if row is not None:
2014-12-22 07:01:09 +00:00
heartbeat = {
2015-09-29 10:11:25 +00:00
'entity': row[0],
'type': row[1],
'time': row[2],
'project': row[3],
'branch': row[4],
'is_write': True if row[5] is 1 else False,
'stats': row[6],
'misc': row[7],
'plugin': row[8],
}
loop = False
2015-09-29 10:11:25 +00:00
except sqlite3.Error: # pragma: nocover
2016-06-17 08:17:29 +00:00
log.traceback('debug')
sleep(wait)
tries -= 1
2014-05-26 21:06:36 +00:00
try:
conn.close()
2015-09-29 10:11:25 +00:00
except sqlite3.Error: # pragma: nocover
2016-06-17 08:17:29 +00:00
log.traceback('debug')
2014-12-22 07:01:09 +00:00
return heartbeat