Refactoring: step 1
- пакет app переименован в src - файл БД теперь должен лежать в data - удалена папка коммандс, все команды вынесены в модуль коммандс или админ_коммандс - нет импортов в ините - убраны все импорты * - теперь в .env можно указать use_reminder="FALSE", тогда не будет стартовать ремайндер - можно указать урл для Правил - в конфиге теперь только конфиг -то, что из енва получаем - константы теперь лежат отдельно - добавлены явные return там, где не было - бот и вебхук разнесены отдельно - при старте проверяем, что Правила доступны - добавлено еще логирование
This commit is contained in:
@@ -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']=}"""
|
||||
Reference in New Issue
Block a user