43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
import shelve
|
|
import os
|
|
from .config import DATA
|
|
|
|
def create_report_text(chat_id):
|
|
top_users = ''
|
|
flooders = []
|
|
if os.path.exists(f'{DATA}{chat_id}.db'):
|
|
with shelve.open(f'{DATA}{chat_id}') as s:
|
|
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'] >= 10:
|
|
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}. <a href="tg://user?id={user.id}">{name}</a> ({flooder["Count"]})'
|
|
report = f"<code>Hello, World!</code> 🌍\n"
|
|
if s['Banned'] or s['Deleted']:
|
|
report += f"""
|
|
<b>Заблокировано</b> ⛔
|
|
├ <b>Пользователей: </b>{s['Banned']}
|
|
└ <b>Сообщений: </b>{s['Deleted']}
|
|
"""
|
|
if len(flooders) >= 3:
|
|
report += f"\n<b>Топ</b> 🗣{top_users}"
|
|
return report
|
|
else:
|
|
return f'Невозможно получить статистику.\n{DATA}{chat_id}.db не существует, возможно запрос в приватном чате. '
|
|
|
|
|
|
def reset_report_stats(chat_id):
|
|
with shelve.open(f'{DATA}{chat_id}') as s:
|
|
s['Messages'] = {}
|
|
s['Banned'] = 0
|
|
s['Deleted'] = 0
|
|
return f"""Chat ID: {chat_id}
|
|
|
|
{s['Messages']=}
|
|
{s['Banned']=}
|
|
{s['Deleted']=}"""
|