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

import logging
from datetime import datetime
from typing import Optional

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) -> Optional[PrivacyPolicy]:
    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=True if enable_if_new else False,
        )
        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