rana-cli/wakatime/offlinequeue.py

123 lines
3.7 KiB
Python
Raw Normal View History

2014-05-26 00:14:38 +00:00
# -*- coding: utf-8 -*-
"""
wakatime.offlinequeue
~~~~~~~~~~~~~~~~~~~~~
2014-05-26 00:14:38 +00:00
Queue for saving heartbeats while offline.
2014-05-26 00:14:38 +00:00
:copyright: (c) 2014 Alan Hamlett.
:license: BSD, see LICENSE for more details.
"""
import logging
import os
import traceback
2014-05-26 00:14:38 +00:00
from time import sleep
try:
import sqlite3
HAS_SQL = True
except ImportError:
HAS_SQL = False
2014-05-26 00:14:38 +00:00
log = logging.getLogger('WakaTime')
2014-05-26 00:14:38 +00:00
class Queue(object):
DB_FILE = os.path.join(os.path.expanduser('~'), '.wakatime.db')
def connect(self):
conn = sqlite3.connect(self.DB_FILE)
c = conn.cursor()
c.execute('''CREATE TABLE IF NOT EXISTS heartbeat (
2014-05-26 00:14:38 +00:00
file text,
time real,
project text,
branch text,
is_write integer,
stats text,
misc text,
2014-05-26 00:14:38 +00:00
plugin text)
''')
return (conn, c)
2014-12-22 06:42:42 +00:00
def push(self, data, stats, plugin, misc=None):
if not HAS_SQL:
return
2014-05-26 21:05:54 +00:00
try:
conn, c = self.connect()
heartbeat = {
'file': data.get('entity'),
2014-05-26 21:05:54 +00:00
'time': data.get('time'),
'project': data.get('project'),
'branch': data.get('branch'),
'is_write': 1 if data.get('is_write') else 0,
2014-12-22 06:42:42 +00:00
'stats': stats,
2014-12-22 06:40:16 +00:00
'misc': misc,
2014-05-26 21:05:54 +00:00
'plugin': plugin,
}
c.execute('INSERT INTO heartbeat VALUES (:file,:time,:project,:branch,:is_write,:stats,:misc,:plugin)', heartbeat)
2014-05-26 21:05:54 +00:00
conn.commit()
conn.close()
2014-05-27 03:02:30 +00:00
except sqlite3.Error:
log.error(traceback.format_exc())
2014-05-26 00:14:38 +00:00
def pop(self):
if not HAS_SQL:
return None
2014-05-26 00:14:38 +00:00
tries = 3
wait = 0.1
heartbeat = None
2014-05-26 21:05:54 +00:00
try:
conn, c = self.connect()
2014-05-27 03:02:30 +00:00
except sqlite3.Error:
log.debug(traceback.format_exc())
2014-05-26 21:05:54 +00:00
return None
2014-05-26 00:14:38 +00:00
loop = True
while loop and tries > -1:
try:
c.execute('BEGIN IMMEDIATE')
c.execute('SELECT * FROM heartbeat LIMIT 1')
2014-05-26 00:14:38 +00:00
row = c.fetchone()
if row is not None:
values = []
clauses = []
index = 0
for row_name in ['file', 'time', 'project', 'branch', 'is_write']:
if row[index] is not None:
clauses.append('{0}=?'.format(row_name))
values.append(row[index])
else:
clauses.append('{0} IS NULL'.format(row_name))
index += 1
if len(values) > 0:
c.execute('DELETE FROM heartbeat WHERE {0}'.format(' AND '.join(clauses)), values)
else:
c.execute('DELETE FROM heartbeat WHERE {0}'.format(' AND '.join(clauses)))
2014-05-26 00:14:38 +00:00
conn.commit()
if row is not None:
heartbeat = {
2014-05-26 00:14:38 +00:00
'file': row[0],
'time': row[1],
'project': row[2],
'branch': row[3],
'is_write': True if row[4] is 1 else False,
'stats': row[5],
'misc': row[6],
2014-05-26 00:14:38 +00:00
'plugin': row[7],
}
loop = False
2014-05-27 03:02:30 +00:00
except sqlite3.Error:
log.debug(traceback.format_exc())
2014-05-26 00:14:38 +00:00
sleep(wait)
tries -= 1
2014-05-26 21:05:54 +00:00
try:
conn.close()
2014-05-27 03:02:30 +00:00
except sqlite3.Error:
log.debug(traceback.format_exc())
return heartbeat