32 lines
1011 B
Python
32 lines
1011 B
Python
# storage.py - storage backend for TeraHz
|
|
# Copyright Kristjan Komloši 2019
|
|
# All code in this file is licensed under the ISC license, provided in LICENSE.txt
|
|
|
|
|
|
import sqlite3
|
|
class jsonStorage:
|
|
def __init__(self, dbFile):
|
|
'''Storage object constructor. Argument is filename'''
|
|
self.db = sqlite3.connect(dbFile)
|
|
|
|
def listJSONs(self):
|
|
c = self.db.cursor()
|
|
c.execute('SELECT * FROM storage')
|
|
result = c.fetchall()
|
|
c.close()
|
|
return result
|
|
|
|
def storeJSON(self, jsonString, comment):
|
|
c = self.db.cursor()
|
|
c.execute('INSERT INTO storage VALUES (datetime(\'now\', \'localtime\'), ?, ?)', (comment, jsonString))
|
|
c.close()
|
|
self.db.commit()
|
|
|
|
def retrieveJSON(self, timestamp):
|
|
'''Retrieves a JSON entry. Takes a timestamp string'''
|
|
c = self.db.cursor()
|
|
c.execute('SELECT * FROM storage WHERE timestamp = ?', (timestamp,))
|
|
result = c.fetchall()
|
|
c.close()
|
|
return result
|