Vladless
42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from core.redis_cache import cache_delete, cache_get, cache_set
|
|
from database.models import Setting
|
|
|
|
|
|
_KEY = "SITE_INITIALIZED"
|
|
_CACHE_KEY = "site_state:initialized"
|
|
_CACHE_TTL_SEC = 300
|
|
|
|
|
|
async def is_site_initialized(session: AsyncSession) -> bool:
|
|
cached = await cache_get(_CACHE_KEY)
|
|
if isinstance(cached, bool):
|
|
return cached
|
|
result = await session.execute(select(Setting).where(Setting.key == _KEY))
|
|
setting = result.scalar_one_or_none()
|
|
value = bool(setting and setting.value is True)
|
|
await cache_set(_CACHE_KEY, value, _CACHE_TTL_SEC)
|
|
return value
|
|
|
|
|
|
async def mark_site_initialized(session: AsyncSession) -> None:
|
|
"""Идемпотентно выставляет флаг инициализации сайта."""
|
|
result = await session.execute(select(Setting).where(Setting.key == _KEY))
|
|
setting = result.scalar_one_or_none()
|
|
if setting is None:
|
|
session.add(Setting(key=_KEY, value=True, description="Сайт прошёл первую настройку админом"))
|
|
elif setting.value is not True:
|
|
setting.value = True
|
|
await cache_delete(_CACHE_KEY)
|
|
|
|
|
|
async def reset_site_initialized(session: AsyncSession) -> None:
|
|
"""Сброс флага — вызывается при полном ресете сайта через TG-бот."""
|
|
result = await session.execute(select(Setting).where(Setting.key == _KEY))
|
|
setting = result.scalar_one_or_none()
|
|
if setting is not None:
|
|
setting.value = False
|
|
await cache_delete(_CACHE_KEY)
|