remnawave-bedolaga-telegram-bot/app/database/crud/privacy_policy.py

import logging
from datetime import datetime

from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from app.database.models import PrivacyPolicy


logger = logging.getLogger(__name__)


async def get_privacy_policy(db: AsyncSession, language: str) -> PrivacyPolicy | None:
    result = await db.execute(select(PrivacyPolicy).where(PrivacyPolicy.language == language))
    return result.scalar_one_or_none()


async def upsert_privacy_policy(
    db: AsyncSession,
    language: str,
    content: str,
    *,
    enable_if_new: bool = True,
) -> PrivacyPolicy:
    policy = await get_privacy_policy(db, language)

    if policy:
        policy.content = content or ''
        policy.updated_at = datetime.utcnow()
    else:
        policy = PrivacyPolicy(
            language=language,
            content=content or '',
            is_enabled=bool(enable_if_new),
        )
        db.add(policy)

    await db.commit()
    await db.refresh(policy)

    logger.info(
        '✅ Политика конфиденциальности для языка %s обновлена (ID: %s)',
        language,
        policy.id,
    )

    return policy


async def set_privacy_policy_enabled(
    db: AsyncSession,
    language: str,
    enabled: bool,
) -> PrivacyPolicy:
    policy = await get_privacy_policy(db, language)

    if policy:
        policy.is_enabled = bool(enabled)
        policy.updated_at = datetime.utcnow()
    else:
        policy = PrivacyPolicy(
            language=language,
            content='',
            is_enabled=bool(enabled),
        )
        db.add(policy)

    await db.commit()
    await db.refresh(policy)

    logger.info(
        '✅ Статус политики конфиденциальности для языка %s обновлен: %s',
        language,
        'enabled' if policy.is_enabled else 'disabled',
    )

    return policy