Vladless
23 lines
980 B
Python
23 lines
980 B
Python
from sqlalchemy.dialects.postgresql import insert
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from database.models import BlockedUser
|
|
from logger import logger
|
|
|
|
|
|
async def create_blocked_user(session: AsyncSession, tg_id: int):
|
|
stmt = insert(BlockedUser).values(tg_id=tg_id).on_conflict_do_nothing(index_elements=[BlockedUser.tg_id])
|
|
await session.execute(stmt)
|
|
await session.commit()
|
|
|
|
|
|
async def save_blocked_user_ids(session: AsyncSession, tg_ids: list[int]) -> None:
|
|
"""Вставка списка tg_id в таблицу BlockedUser. Вызывать только из основного event loop."""
|
|
if not tg_ids:
|
|
return
|
|
values = [{"tg_id": tg_id} for tg_id in tg_ids]
|
|
stmt = insert(BlockedUser).values(values).on_conflict_do_nothing(index_elements=[BlockedUser.tg_id])
|
|
await session.execute(stmt)
|
|
await session.commit()
|
|
logger.info(f"📝 Добавлено {len(tg_ids)} пользователей в blocked_users")
|