diff --git a/handlers/user.py b/handlers/user.py index 442dd87..4d57b75 100644 --- a/handlers/user.py +++ b/handlers/user.py @@ -26,6 +26,13 @@ async def format_rating(top_users: List[Any], api: API) -> str: return result +def calculate_total_messages(top_users: List[Any]) -> int: + total = 0 + for user in top_users: + total += user['value'] + return total + + # noinspection SpellCheckingInspection @labeler.chat_message(text="!помощь") async def rules_handler(message: Message): @@ -70,6 +77,7 @@ async def stats_today_handler(message: Message): return response = bold('Статистика за сегодня') + '\n' + response += 'Всего сообщений - {}\n'.format(calculate_total_messages(top_users)) response += await format_rating(top_users, message.ctx_api) await message.answer(response) @@ -88,6 +96,7 @@ async def stats_month_handler(message: Message): return response = bold('Статистика за месяц') + '\n' + response += 'Всего сообщений - {}\n'.format(calculate_total_messages(top_users)) response += await format_rating(top_users, message.ctx_api) await message.answer(response)