42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
import sqlite3
|
||
from loguru import logger
|
||
|
||
|
||
conn = sqlite3.connect("fmn_states.sqlite", check_same_thread=False)
|
||
c = conn.cursor()
|
||
|
||
c.execute(
|
||
'''CREATE TABLE IF NOT EXISTS states(key VARCHAR (500) UNIQUE DEFAULT NULL, value VARCHAR (500) UNIQUE DEFAULT NULL)''')
|
||
|
||
conn.commit()
|
||
|
||
|
||
def add_state(key, value):
|
||
'''Создание стейта, если существует - будет заменен'''
|
||
logger.debug(f'Adding {key}: {value}')
|
||
c.execute("INSERT OR REPLACE INTO states(key, value) VALUES (?, ?)", (key, value))
|
||
conn.commit()
|
||
logger.warning(f'Добавлен стейт {key}: {value}')
|
||
|
||
|
||
def get_state(key):
|
||
'''Получение стейта по ключу'''
|
||
logger.trace(f'Запрошен стейт {key}')
|
||
value = c.execute("SELECT value FROM states WHERE key = (?)", (key,)).fetchone()
|
||
if value:
|
||
return value[0]
|
||
|
||
|
||
def remove_state(key):
|
||
'''Удалить стейт по ключу'''
|
||
c.execute("DELETE FROM states WHERE key = (?)", (key,))
|
||
conn.commit()
|
||
logger.warning(f'Удален стейт {key}')
|
||
|
||
|
||
def clear_all_states():
|
||
c.execute("DELETE FROM states")
|
||
conn.commit()
|
||
logger.warning(f'Все стейты удалены')
|
||
|