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.
56 lines
2.7 KiB
Python
56 lines
2.7 KiB
Python
import os
|
|
import sqlite3
|
|
import sys
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "scripts"))
|
|
|
|
import db # noqa: E402
|
|
import migrate_sqlite # noqa: E402
|
|
from conftest import DSN # noqa: E402
|
|
|
|
|
|
def make_old_dbs(path):
|
|
users = sqlite3.connect(path / "users.db")
|
|
users.executescript("""
|
|
CREATE TABLE users (id INTEGER PRIMARY KEY AUTOINCREMENT, user_id INTEGER NOT NULL);
|
|
CREATE TABLE "groups" (id INTEGER PRIMARY KEY AUTOINCREMENT, group_id INTEGER NOT NULL);
|
|
INSERT INTO users (user_id) VALUES (111), (222);
|
|
INSERT INTO "groups" (group_id) VALUES (-100), (-200);
|
|
""")
|
|
users.commit()
|
|
base = sqlite3.connect(path / "base.db")
|
|
base.executescript("""
|
|
CREATE TABLE work (group_id INTEGER NOT NULL, state BOOLEAN NOT NULL DEFAULT (True));
|
|
CREATE TABLE stat (group_id INTEGER NOT NULL, mes INTEGER, rep INTEGER, com INTEGER, url INTEGER,
|
|
med INTEGER, sti INTEGER, voi INTEGER);
|
|
CREATE TABLE edit (user_id INTEGER NOT NULL);
|
|
INSERT INTO work VALUES (1, 1), (2, 0);
|
|
INSERT INTO stat VALUES (1, 50, 5, 1, 0, 2, 3, 4), (2, 7, 0, 0, 0, 0, 0, 0);
|
|
INSERT INTO edit VALUES (2);
|
|
""")
|
|
base.commit()
|
|
groups = sqlite3.connect(path / "groups.db")
|
|
groups.executescript("""
|
|
CREATE TABLE [1] (user_id INTEGER, first_name STRING, mes INTEGER, rep INTEGER, com INTEGER, url INTEGER,
|
|
med INTEGER, sti INTEGER, voi INTEGER);
|
|
CREATE TABLE [2] (user_id INTEGER, first_name STRING, mes INTEGER, rep INTEGER, com INTEGER, url INTEGER,
|
|
med INTEGER, sti INTEGER, voi INTEGER);
|
|
INSERT INTO [1] VALUES (1, 'маша', 30, 5, 1, 0, 2, 3, 4), (2, 'котик', 20, 0, 0, 0, 0, 0, 0);
|
|
INSERT INTO [2] VALUES (2, 'котик', 7, 0, 0, 0, 0, 0, 0), (9, 'призрак', 1, 0, 0, 0, 0, 0, 0);
|
|
""")
|
|
groups.commit()
|
|
|
|
|
|
async def test_migration(pool, tmp_path):
|
|
make_old_dbs(tmp_path)
|
|
await db.close()
|
|
assert await migrate_sqlite.migrate(str(tmp_path), DSN, force=True)
|
|
await db.connect(DSN)
|
|
assert await db.pool.fetchval("SELECT short_id FROM chats WHERE id = -200") == 2
|
|
assert await db.get_setting(-200, "events_enabled") is False
|
|
assert (await db.member_stats(-100, 111, "all"))[0] == 30
|
|
assert await db.chat_names(-100) == {"маша": 111, "котик": 222}
|
|
assert await db.pool.fetchval("SELECT custom_name FROM members WHERE chat_id = -100 AND user_id = 222")
|
|
# новый пользователь получает следующий внутренний номер
|
|
await db.ensure_user(333)
|
|
assert await db.pool.fetchval("SELECT short_id FROM users WHERE id = 333") == 3
|