mirror of
https://github.com/EDeev/chatping_abobot.git
synced 2026-10-07 20:49:45 +03:00
Egor Deev
a2a0adcf8e
База — PostgreSQL (asyncpg) вместо четырёх файлов SQLite с таблицей на каждый чат: chats, users, members, chat_stats и member_stats с полем «период» — месяцы хранятся, а не обнуляются; счётчики растут одним запросом без блокировки бота. scripts/migrate_sqlite.py переносит старые базы со сверкой сумм (проверено на копии рабочих баз: 20 694 пользователя, 45 чатов, 3 142 038 сообщений — совпадает). Учёт бота в чатах: статус и права бота (событие my_chat_member и сверка с Telegram при запуске и раз в 6 часов), история изменений в bot_status_history; из чатов, где бота исключили, данные не удаляются — меняется только статус. Большие чаты (больше 100 участников): /all — только для админов, раз в 5 минут и только писавшие за 30 дней; упоминание одного человека по имени — не чаще раза в минуту. В любом чате /all делится на сообщения до 4096 символов (раньше в чате на 14 тысяч падал). Новое: /settings (упоминания, имена в голосовых, ивенты, удаление служебных сообщений — меняют админы), /top, /month (итоги месяца по команде). /stop_bot раньше ничего не выключал — флаг не проверялся; теперь работает. Участник, вышедший из чата, больше не упоминается, но его статистика сохраняется. Ошибки — в технический чат DEBUG_CHAT_ID. Код: handlers.py разбит на роутеры, Markdown заменён на HTML (экранирование через html.escape), тесты на PostgreSQL (18), CI на Python 3.10 и 3.12.
49 lines
2.4 KiB
Python
49 lines
2.4 KiB
Python
from datetime import date
|
||
|
||
import db
|
||
|
||
|
||
async def test_counters_all_time_and_month(pool):
|
||
await db.count(-1, 10, [1, 3], "маша", "Чат", day=date(2026, 10, 6))
|
||
await db.count(-1, 10, [1], "маша", day=date(2026, 10, 7))
|
||
await db.count(-1, 10, [1], "маша", day=date(2026, 11, 1))
|
||
assert (await db.member_stats(-1, 10, "all"))[:3] == (3, 0, 1)
|
||
assert (await db.member_stats(-1, 10, "2026-10"))[0] == 2
|
||
assert (await db.member_stats(-1, 10, "2026-11"))[0] == 1 # месяцы хранятся, а не обнуляются
|
||
assert (await db.chat_stats(-1, "all"))[0] == 3
|
||
|
||
|
||
async def test_custom_name_survives_and_left_member(pool):
|
||
await db.count(-1, 10, [1], "мария")
|
||
await db.set_custom_name(-1, 10, "маша")
|
||
await db.count(-1, 10, [1], "мария")
|
||
assert await db.chat_names(-1) == {"маша": 10}
|
||
await db.member_left(-1, 10)
|
||
assert await db.chat_names(-1) == {}
|
||
assert (await db.member_stats(-1, 10, "all"))[0] == 2 # статистика ушедшего остаётся
|
||
|
||
|
||
async def test_chat_migration_keeps_stats(pool):
|
||
await db.count(-1, 10, [1], "маша")
|
||
await db.migrate_chat(-1, -1001)
|
||
assert (await db.member_stats(-1001, 10, "all"))[0] == 1
|
||
assert not await db.chat_exists(-1)
|
||
|
||
|
||
async def test_bot_status_history(pool):
|
||
assert await db.set_bot_status(-5, "member", {"can_send_messages": True}, "update", member_count=40)
|
||
assert not await db.set_bot_status(-5, "member", {"can_send_messages": True}, "check") # без изменений
|
||
assert await db.set_bot_status(-5, "administrator", {"can_delete_messages": True}, "update")
|
||
assert await db.set_bot_status(-5, "kicked", None, "check")
|
||
history = await db.pool.fetch("SELECT status, source FROM bot_status_history WHERE chat_id = -5 ORDER BY id")
|
||
assert [(r["status"], r["source"]) for r in history] == [
|
||
("member", "update"), ("administrator", "update"), ("kicked", "check")]
|
||
assert await db.chats_to_check() == [] # из исключённого чата не проверяем
|
||
assert await db.pool.fetchval("SELECT count(*) FROM active_chats") == 0
|
||
|
||
|
||
async def test_top(pool):
|
||
for uid, name, n in ((1, "а", 3), (2, "б", 5), (3, "в", 1)):
|
||
for _ in range(n):
|
||
await db.count(-1, uid, [1], name)
|
||
assert [(n, v) for _, n, v in await db.top(-1, "all", "mes", 2)] == [("б", 5), ("а", 3)]
|