lavatech/rana-cli
2
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

use WAKATIME_HOME env variable for session and offline caching

Browse source
This commit is contained in:
Alan Hamlett 2017-10-29 11:16:23 -07:00
parent c8237c30c5
commit 6cd300a30b
6 changed files with 101 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

7
wakatime/offlinequeue.py
View file

@ -27,11 +27,14 @@ log = logging.getLogger('WakaTime')
class Queue(object):
db_file = os.path.join(os.path.expanduser('~'), '.wakatime.db')
db_file = '.wakatime.db'
table_name = 'heartbeat_1'
def get_db_file(self):
return self.db_file
home = '~'
if os.environ.get('WAKATIME_HOME'):
home = os.environ.get('WAKATIME_HOME')
return os.path.join(os.path.expanduser(home), '.wakatime.db')
def connect(self):
conn = sqlite3.connect(self.get_db_file(), isolation_level=None)

23
wakatime/session_cache.py
View file

@ -30,14 +30,21 @@ log = logging.getLogger('WakaTime')
class SessionCache(object):
DB_FILE = os.path.join(os.path.expanduser('~'), '.wakatime.db')
db_file = '.wakatime.db'
table_name = 'session'
def get_db_file(self):
home = '~'
if os.environ.get('WAKATIME_HOME'):
home = os.environ.get('WAKATIME_HOME')
return os.path.join(os.path.expanduser(home), '.wakatime.db')
def connect(self):
conn = sqlite3.connect(self.DB_FILE, isolation_level=None)
conn = sqlite3.connect(self.get_db_file(), isolation_level=None)
c = conn.cursor()
c.execute('''CREATE TABLE IF NOT EXISTS session (
c.execute('''CREATE TABLE IF NOT EXISTS {0} (
value BLOB)
''')
'''.format(self.table_name))
return (conn, c)
def save(self, session):
@ -48,11 +55,11 @@ class SessionCache(object):
return
try:
conn, c = self.connect()
c.execute('DELETE FROM session')
c.execute('DELETE FROM {0}'.format(self.table_name))
values = {
'value': sqlite3.Binary(pickle.dumps(session, protocol=2)),
}
c.execute('INSERT INTO session VALUES (:value)', values)
c.execute('INSERT INTO {0} VALUES (:value)'.format(self.table_name), values)
conn.commit()
conn.close()
except: # pragma: nocover
@ -76,7 +83,7 @@ class SessionCache(object):
session = None
try:
c.execute('BEGIN IMMEDIATE')
c.execute('SELECT value FROM session LIMIT 1')
c.execute('SELECT value FROM {0} LIMIT 1'.format(self.table_name))
row = c.fetchone()
if row is not None:
session = pickle.loads(row[0])
@ -98,7 +105,7 @@ class SessionCache(object):
return
try:
conn, c = self.connect()
c.execute('DELETE FROM session')
c.execute('DELETE FROM {0}'.format(self.table_name))
conn.commit()
conn.close()
except: