Изменён алгоритм получения значений побед и пожертвований

ClanStat/__main__.py

@@ -48,7 +48,10 @@ if __name__ == '__main__':
         client.run(main())
     except exceptions.unauthorized_401.AuthKeyUnregistered:
         rich_print('Ошибка Telegram: [bold red][401 AUTH_KEY_UNREGISTERED][/bold red]')
-        rich_print('Удалите файлы авторизации [bold green]Pyrogram[/bold green]:'' [italic yellow]*.session, *.session-journal[/italic yellow]')
+        rich_print(
+            'Удалите файлы авторизации [bold green]Pyrogram[/bold green]:'
+            ' [italic yellow]*.session, *.session-journal[/italic yellow]'
+        )
     except KeyboardInterrupt:
         pass
 

ClanStat/actions.py

@@ -1,3 +1,7 @@
+"""
+🩻 ...
+"""
+
 # Импорт модулей стандартной библиотеки
 import time
 
@@ -8,7 +12,7 @@ from .collect import parse_wins_top, parse_donates_top
 
 async def get_top_wins() -> None:
     """⚔️ Получает топ клана по победам"""
-    logger.info('Выполняется получения списка побед клана')
+    logger.info('⚔️ Выполняется получения списка побед клана')
 
     await client.send_message(BOT_NAME, '🛡 Мой клан')
     time.sleep(1)
@@ -44,7 +48,7 @@ async def get_top_wins() -> None:
 
 async def get_top_donates() -> None:
     """💠 Получает топ клана по пожертвованиям"""
-    logger.info('Выполняется получения списка пожертвований клана')
+    logger.info('💠 Выполняется получения списка пожертвований клана')
 
     await client.send_message(BOT_NAME, '🛡 Мой клан')
     time.sleep(2)

ClanStat/collect.py

@@ -1,7 +1,14 @@
+"""
+🩻 ...
+"""
+
 # Импорт модулей стандартной библиотеки
 import re
+import io
 
 # Импорт сторонних модулей
+from rich import print as rich_print
+from rich.table import Table
 from pyrogram.types import Message
 from pyrogram.enums import MessageEntityType
 
@@ -12,69 +19,75 @@ from .utils import get_telegram_id, get_telegram_data, clean_icons
 
 async def parse_wins_top(message: Message) -> None:
     """⚔️ Выполняет парсинг данных топа побед членов клана"""
-    if message.text.startswith('🏆 Топ по победам'):
-        players = message.text.split('\n')
-        players.pop(0)
-        players.pop(0)
 
-        WINS = list()
+    if not message.text.startswith('🏆 Топ по победам'):
+        return
 
-        for idx, player in enumerate(players):
+    WINS = await parse_tops(message, '⚔')
-            player = re.sub(r'^\d+. ', '', player)  # удаляет нумерацию
-            player, battle_count = player.split(' - ')  # разделяет ник и количество побед в клановых сражениях
-            player = clean_icons(player)
 
-            battle_count = battle_count.replace(' ⚔', '')  # удаляем эмодзи мечей
+    # 📟 Отрисовывает таблицу побед
-            battle_count = int(re.sub(r'[^\x00-\x7F]', '', battle_count))  # преобразовывает строку в число
+    table = Table(show_header=True)
+    table.add_column('ID', style='dim', justify='right', width=12)
+    table.add_column('Username', style='magenta')
+    table.add_column('Count', style='green', width=7)
+    table.add_column('URL', style='yellow')
+    for i in WINS:
+        table.add_row(str(i['telegram_id']), i['username'], str(i['count']), i['url'])
+    rich_print(table)
 
-            player_url, player_telegram_id = await get_telegram_data(message, idx)
+    db.add_data(WINS, True)  # Добавляет данные в базу данных
-
-            WINS.append(
-                {
-                    'telegram_id': player_telegram_id,
-                    'username': player,
-                    'count': battle_count,
-                    'url': player_url,
-                }
-            )
-
-        print(WINS)
-        db.add_data(WINS, True)
-    else:
-        pass
 
 
 async def parse_donates_top(message: Message) -> None:
     """💠 Выполняет парсинг данных топа пожертвований членов клана"""
-    if message.text.startswith('🏆 Топ по пожертвованиям'):
-        players = message.text.split('\n')
 
-        # Удаляет из массива строку с заголовком и последующую пустую строку
+    if not message.text.startswith('🏆 Топ по пожертвованиям'):
-        players.pop(0)
+        return
-        players.pop(0)
 
-        DONATES = list()
+    DONATES = await parse_tops(message, '💠')
 
-        for idx, player in enumerate(players):
+    # 📟 Отрисовывает таблицу побед
-            player = re.sub(r'^\d+. ', '', player)  # удаляет нумерацию
+    table = Table(show_header=True)
-            player, donates_count = player.split(' - ')  # разделяет ник и количество пожертвований в клановую сокровищницу
+    table.add_column('ID', style='dim', justify='right', width=12)
-            player = clean_icons(player)
+    table.add_column('Username', style='magenta')
+    table.add_column('Count', style='green', width=7)
+    table.add_column('URL', style='yellow')
+    for i in DONATES:
+        table.add_row(str(i['telegram_id']), i['username'], str(i['count']), i['url'])
+    rich_print(table)
 
-            donates_count = donates_count.replace(' 💠', '')  # удаляем эмодзи пожертвований
+    db.add_data(DONATES, False)  # Добавляет данные в базу данных
-            donates_count = int(re.sub(r'[^\x00-\x7F]', '', donates_count))  # преобразовывает строку в число
 
-            player_url, player_telegram_id = await get_telegram_data(message, idx)
 
-            DONATES.append(
+async def parse_tops(message: Message, emoji_replace: str) -> list:
-                {
+    """..."""
-                    'telegram_id': player_telegram_id,
-                    'username': player,
-                    'count': donates_count,
-                    'url': player_url,
-                }
-            )
 
-        print(DONATES)
+    top_data = list()
-        db.add_data(DONATES, False)
+
-    else:
+    msg_buffer = io.StringIO(message.text)
-        pass
+    msg_buffer.readline()
+    msg_buffer.readline()
+
+    for e in message.entities:
+        player_line = msg_buffer.readline()
+        player_name, battle_count = player_line.split(' - ')
+        player = clean_icons(message.text[e.offset : e.offset + e.length])
+        player_name = clean_icons(re.sub(r'^\d+. ', '', player_name))
+        battle_count = battle_count.replace(f' {emoji_replace}', '')  # Удаляет эмодзи
+        battle_count = int(re.sub(r'[^\x00-\x7F]', '', battle_count))
+
+        if player not in player_name:
+            top_data.append({'telegram_id': None, 'username': player_name, 'count': battle_count, 'url': None})
+            msg_buffer.readline()
+            continue
+
+        if e.type is MessageEntityType.TEXT_LINK:
+            player_url = e.url
+            player_telegram_id = await get_telegram_id(player_url.replace('http://t.me/', ''))
+        elif e.type is MessageEntityType.TEXT_MENTION:
+            player_url = e.url
+            player_telegram_id = e.user.id
+
+        top_data.append({'telegram_id': player_telegram_id, 'username': player, 'count': battle_count, 'url': player_url})
+
+    return top_data

ClanStat/db.py

@@ -1,3 +1,7 @@
+"""
+🗃️ Работа с базой данных
+"""
+
 # Импорт модулей стандартной библиотеки
 import json
 
@@ -42,8 +46,7 @@ class DataBase(object):
         }
         """Настройки TCP"""
 
-
+        self.conn = psycopg2.connect(**connection_kwargs, **keepalive_kwargs)  # Выполняет подключение к базе данных
-        self.conn = psycopg2.connect(**connection_kwargs, **keepalive_kwargs) # Выполняет подключение к базе данных
         self.conn.autocommit = True  # Активирует автоматический коммит транзакций
 
         return True

ClanStat/utils.py

@@ -1,3 +1,7 @@
+"""
+🧰 Утилиты
+"""
+
 # Импорт сторонних модулей
 from pyrogram.types import Message
 from pyrogram.enums import MessageEntityType
@@ -32,7 +36,7 @@ async def get_telegram_data(message: Message, idx: int) -> tuple():
 
 
 def clean_icons(text: str) -> str:
-    """Удаляет иконки подписок из ников"""
+    """💎 Удаляет иконки подписок из ников"""
 
     # оставляет ники игроков, Python не может в эмодзи в RegExp, по этому пришлось делать через str.replace()
     # player = re.sub(r'\s[⚡⚜]$', '', player)

HISTORY.md

@@ -14,5 +14,7 @@ include_toc: true
 
 ## 2.2.4
 
-* Добавлено сообщение об ошибке **401** `AUTH_KEY_UNREGISTERED`
+* 🟢 Добавлено сообщение об ошибке **401** `AUTH_KEY_UNREGISTERED`
-* Добавлены emoji в приветственное сообщение
+* 🟢 Добавлены emoji в приветственное сообщение
+* 🟢 Результаты в терминале отображаются в таблице
+* 🟢 Изменён алгоритм получения значений побед и пожертвований