format Black

This commit is contained in:
Rykov7
2023-02-22 18:53:12 +03:00
parent 23a93b5dbd
commit d0cf0762c8
13 changed files with 356 additions and 205 deletions
+27 -18
View File
@@ -4,31 +4,40 @@ from .config import DATA, ROLLBACK
def create_report_text(chat_id):
report = ''
top_users = ''
report = ""
top_users = ""
flooders = []
if os.path.exists(f'{DATA}{chat_id}.db'):
with shelve.open(f'{DATA}{chat_id}', writeback=True) as s:
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
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:
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
name = (
f"{user.first_name} {user.last_name}"
if user.last_name
else user.first_name
)
top_users += f'\n {i + 1}. {name} [<code>{flooder["Count"]}</code>]'
if len(flooders) >= 3 or s['Banned'] or s['Deleted']:
if len(flooders) >= 3 or s["Banned"] or s["Deleted"]:
report = f"<code>Hello, World!</code>\n"
if len(flooders) >= 3:
report += f"\n<b>Топ</b> 🐍{top_users}\n"
if s['Banned'] or s['Deleted']:
if s["Banned"] or s["Deleted"]:
report += f"""
<b>Заблокировано</b>
├ Пользователей: <code>{s['Banned']}</code>
@@ -36,14 +45,14 @@ def create_report_text(chat_id):
"""
return report
else:
return f'Невозможно получить статистику.\n{DATA}{chat_id}.db не существует, возможно запрос в приватном чате. '
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
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']=}