mirror of
https://github.com/norohind/SquadsActivityMonitor.git
synced 2025-04-12 13:00:02 +03:00
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
import sqlite3
|
|
from EDMCLogging import get_main_logger
|
|
import config
|
|
from typing import Union
|
|
|
|
logger = get_main_logger()
|
|
|
|
|
|
class Cache:
|
|
def __init__(self, disabled: bool = False):
|
|
self.disabled = disabled
|
|
|
|
if disabled:
|
|
return
|
|
|
|
try:
|
|
self.db: sqlite3.Connection = sqlite3.connect(config.model_cache_db_path, check_same_thread=False)
|
|
|
|
except Exception as e:
|
|
logger.warning('Cannot create cache DB due to', exc_info=e)
|
|
logger.warning('Cache is disabled')
|
|
self.disabled = True
|
|
return
|
|
|
|
self.db.row_factory = lambda c, r: dict(zip([col[0] for col in c.description], r))
|
|
|
|
with self.db:
|
|
self.db.execute("create table if not exists cache (key text primary key, value text);")
|
|
|
|
def set(self, key, value) -> None:
|
|
if self.disabled:
|
|
return
|
|
|
|
with self.db:
|
|
if self.db.execute('select count(key) as count from cache where key = ?;', [key]).fetchone()['count'] == 1:
|
|
|
|
# key exists, just need to update value
|
|
self.db.execute('update cache set value = ? where key = ?;', [value, key])
|
|
else:
|
|
|
|
# key doesn't exists, need to insert new row
|
|
self.db.execute('insert into cache (key, value) values (?, ?);', [key, value])
|
|
|
|
def get(self, key, default=None) -> Union[str, None]:
|
|
if self.disabled:
|
|
return
|
|
|
|
res = self.db.execute('select value from cache where key = ?;', [key]).fetchone()
|
|
if res is None:
|
|
return default
|
|
|
|
return res['value']
|
|
|
|
def delete(self, key) -> None:
|
|
if self.disabled:
|
|
return
|
|
|
|
self.db.execute('delete from cache where key = ?;', [key])
|
|
|
|
def delete_all(self) -> None:
|
|
if self.disabled:
|
|
return
|
|
|
|
logger.debug('Dropping cache')
|
|
with self.db:
|
|
self.db.execute('delete from cache;')
|
|
|
|
|
|
cache = Cache(disabled=config.cache_disabled)
|