Capybara-z
28 lines
1.1 KiB
Python
28 lines
1.1 KiB
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 батчами по 500. Вызывать только из основного event loop."""
|
|
if not tg_ids:
|
|
return
|
|
batch_size = 500
|
|
total = 0
|
|
for i in range(0, len(tg_ids), batch_size):
|
|
batch = tg_ids[i : i + batch_size]
|
|
values = [{"tg_id": tg_id} for tg_id in batch]
|
|
stmt = insert(BlockedUser).values(values).on_conflict_do_nothing(index_elements=[BlockedUser.tg_id])
|
|
await session.execute(stmt)
|
|
await session.commit()
|
|
total += len(batch)
|
|
logger.info(f"📝 Добавлено {total} пользователей в blocked_users")
|