Refactoring: step 1

- пакет app переименован в src
- файл БД теперь должен лежать в data
- удалена папка коммандс, все команды вынесены в модуль коммандс или админ_коммандс
- нет импортов в ините
- убраны все импорты *
- теперь в .env можно указать use_reminder="FALSE", тогда не будет стартовать ремайндер
- можно указать урл для Правил
- в конфиге теперь только конфиг -то, что из енва получаем
- константы теперь лежат отдельно
- добавлены явные return там, где не было
- бот и вебхук разнесены отдельно
- при старте проверяем, что Правила доступны
- добавлено еще логирование
This commit is contained in:
kotolex
2023-12-22 19:16:57 +05:00
parent 1cd96220a5
commit 9103669cac
19 changed files with 268 additions and 258 deletions
+65
View File
@@ -0,0 +1,65 @@
import html
import os
import random
import shelve
from .config import ROLLBACK
from .constants import DATA
def create_report_text(chat_id):
report = ""
top_users = ""
flooders = []
spammer_icons = ('🐷', '🐒', '🐔')
current_spammer_face = random.choice(spammer_icons)
if os.path.exists(f"{DATA}{chat_id}.db"):
with shelve.open(f"{DATA}{chat_id}", writeback=True) as s:
for user_id in ROLLBACK:
if user_id in s["Messages"] and s["Messages"][user_id]["Count"] > 10:
s["Messages"][user_id]["Count"] //= 3
for n in range(min(3, len(s["Messages"]))):
top_user = s["Messages"][
sorted(
s["Messages"],
key=lambda a: s["Messages"][a]["Count"],
reverse=True,
)[n]
]
if top_user["Count"] >= 5:
flooders.append(top_user)
for i, flooder in enumerate(flooders):
user = flooder["User"]
name = (
f"{user.first_name} {user.last_name}"
if user.last_name
else user.first_name
)
top_users += f'\n{i + 1}. {html.escape(name)} [{flooder["Count"]}]'
if len(flooders) >= 3 or s["Banned"] or s["Deleted"]:
report = f"<b>H</b>ello, <b>W</b>orld!\n"
if len(flooders) >= 3:
report += f"{top_users}\n"
if s["Banned"]:
report += f"""
<b> Криптоидов поймано:</b> {s['Banned']}
└ [ {s['Banned'] * current_spammer_face} ]
"""
return report
else:
return f"Невозможно получить статистику.\n{DATA}{chat_id}.db не существует, возможно запрос в приватном чате. "
def reset_report_stats(chat_id):
with shelve.open(f"{DATA}{chat_id}") as shelve_db:
shelve_db["Messages"] = {}
shelve_db["Banned"] = 0
shelve_db["Deleted"] = 0
return f"""Chat ID: {chat_id}
{shelve_db['Messages']=}
{shelve_db['Banned']=}
{shelve_db['Deleted']=}"""