pictuga/morss
1
1
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

caching: simplify sqlite code

Browse Source
This commit is contained in:
pictuga
2021-11-07 18:14:18 +01:00
parent 1ff7e4103c
commit cbf7b3f77b
1 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
morss/caching.py
View File

@@ -58,8 +58,8 @@ except ImportError:
class SQLiteCache(BaseCache): class SQLiteCache(BaseCache):
def __init__(self, filename=':memory:'): def __init__(self, path=':memory:'):
self.con = sqlite3.connect(filename, detect_types=sqlite3.PARSE_DECLTYPES, check_same_thread=False) self.con = sqlite3.connect(path, detect_types=sqlite3.PARSE_DECLTYPES, check_same_thread=False)
with self.con: with self.con:
self.con.execute('CREATE TABLE IF NOT EXISTS data (ky UNICODE PRIMARY KEY, data BLOB, timestamp INT)') self.con.execute('CREATE TABLE IF NOT EXISTS data (ky UNICODE PRIMARY KEY, data BLOB, timestamp INT)')
@@ -168,13 +168,9 @@ if 'CACHE' in os.environ:
) )
elif os.environ['CACHE'] == 'sqlite': elif os.environ['CACHE'] == 'sqlite':
if 'SQLITE_PATH' in os.environ: default_cache = SQLiteCache(
path = os.getenv('SQLITE_PATH') os.getenv('SQLITE_PATH', ':memory:')
)
else:
path = ':memory:'
default_cache = SQLiteCache(path)
elif os.environ['CACHE'] == 'redis': elif os.environ['CACHE'] == 'redis':
default_cache = RedisCacheHandler( default_cache = RedisCacheHandler(