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

ClanStat/__init__.py

@@ -1,6 +1,6 @@
 # Информация о программе
 __author__ = 'Alexander Popov'
-__version__ = (2, 2, 3)
+__version__ = (2, 2, 4)
 
 # Импорт модулей стандартной библиотеки
 from os import getenv

ClanStat/__main__.py

@@ -1,6 +1,10 @@
 # Импорт модулей стандартной библиотеки
 import time
 
+# Импорт сторонних модулей
+from pyrogram.errors import exceptions
+from rich import print as rich_print
+
 # Импорт модулей приложения
 from . import __version__, client, db, logger, DELAY_TIME
 from .actions import get_top_wins, get_top_donates
@@ -33,8 +37,8 @@ async def main():
 
 
 if __name__ == '__main__':
-    logger.warning('ClanStat :: {}', '.'.join(map(str, __version__)))
-    logger.warning('URL: {}', 'https://git.a2s.su/AniCard/ClanStat/')
+    logger.warning('🎴 ClanStat 🆔 {}', '.'.join(map(str, __version__)))
+    logger.warning('🌐 URL: {}', 'https://git.a2s.su/AniCard/ClanStat/')
     logger.info('Выполняется запуск приложения')
 
     logger.info('Выполняется подключение к базе данных')
@@ -42,6 +46,12 @@ if __name__ == '__main__':
 
     try:
         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]'
+        )
     except KeyboardInterrupt:
         pass
 

ClanStat/actions.py

@@ -1,3 +1,7 @@
+"""
+🩻 ...
+"""
+
 # Импорт модулей стандартной библиотеки
 import time
 
@@ -7,8 +11,8 @@ 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)
@@ -22,7 +26,7 @@ async def get_top_wins() -> None:
         message_id=message_id,
         callback_data='my_clan:tops:91:0:0:0:0:1',
     )
-    time.sleep(1)
+    time.sleep(2)
 
     async for message in client.get_chat_history(BOT_NAME, limit=1):
         message_id = message.id
@@ -32,22 +36,22 @@ async def get_top_wins() -> None:
         message_id=message_id,
         callback_data='my_clan:top-wins:91:0:0:0:0:1',
     )
-    time.sleep(1)
+    time.sleep(2)
 
     async for message in client.get_chat_history(BOT_NAME, limit=1):
         await parse_wins_top(message)
 
-    await client.read_chat_history(BOT_NAME)
+    await client.read_chat_history(BOT_NAME)  # Помечает сообщения в чате прочитанными
 
     logger.info('Получение списка побед клана завершено')
 
 
 async def get_top_donates() -> None:
-    """Получает топ клана по пожертвованиям"""
-    logger.info('Выполняется получения списка пожертвований клана')
+    """💠 Получает топ клана по пожертвованиям"""
+    logger.info('💠 Выполняется получения списка пожертвований клана')
 
     await client.send_message(BOT_NAME, '🛡 Мой клан')
-    time.sleep(1)
+    time.sleep(2)
 
     message_id = 0
     async for message in client.get_chat_history(BOT_NAME, limit=1):
@@ -58,7 +62,7 @@ async def get_top_donates() -> None:
         message_id=message_id,
         callback_data='my_clan:tops:91:0:0:0:0:1',
     )
-    time.sleep(1)
+    time.sleep(2)
 
     async for message in client.get_chat_history(BOT_NAME, limit=1):
         message_id = message.id
@@ -73,6 +77,6 @@ async def get_top_donates() -> None:
     async for message in client.get_chat_history(BOT_NAME, limit=1):
         await parse_donates_top(message)
 
-    await client.read_chat_history(BOT_NAME)
+    await client.read_chat_history(BOT_NAME)  # Помечает сообщения в чате прочитанными
 
     logger.info('Получение списка пожертвований клана завершено')

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
 
@@ -11,70 +18,76 @@ 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):
-            player = re.sub(r'^\d+. ', '', player)  # удаляет нумерацию
-            player, battle_count = player.split(' - ')  # разделяет ник и количество побед в клановых сражениях
-            player = clean_icons(player)
+    WINS = await parse_tops(message, '⚔')
 
-            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)
-
-            WINS.append(
-                {
-                    'telegram_id': player_telegram_id,
-                    'username': player,
-                    'count': battle_count,
-                    'url': player_url,
-                }
-            )
-
-        print(WINS)
-        db.add_data(WINS, True)
-    else:
-        pass
+    db.add_data(WINS, True)  # Добавляет данные в базу данных
 
 
 async def parse_donates_top(message: Message) -> None:
-    """Выполняет парсинг данных топа пожертвований членов клана"""
-    if message.text.startswith('🏆 Топ по пожертвованиям'):
-        players = message.text.split('\n')
+    """💠 Выполняет парсинг данных топа пожертвований членов клана"""
 
-        # Удаляет из массива строку с заголовком и последующую пустую строку
-        players.pop(0)
-        players.pop(0)
+    if not message.text.startswith('🏆 Топ по пожертвованиям'):
+        return
 
-        DONATES = list()
+    DONATES = await parse_tops(message, '💠')
 
-        for idx, player in enumerate(players):
-            player = re.sub(r'^\d+. ', '', player)  # удаляет нумерацию
-            player, donates_count = player.split(' - ')  # разделяет ник и количество пожертвований в клановую сокровищницу
-            player = clean_icons(player)
+    # 📟 Отрисовывает таблицу побед
+    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 DONATES:
+        table.add_row(str(i['telegram_id']), i['username'], str(i['count']), i['url'])
+    rich_print(table)
 
-            donates_count = donates_count.replace(' 💠', '')  # удаляем эмодзи пожертвований
-            donates_count = int(re.sub(r'[^\x00-\x7F]', '', donates_count))  # преобразовывает строку в число
+    db.add_data(DONATES, False)  # Добавляет данные в базу данных
 
-            player_url, player_telegram_id = await get_telegram_data(message, idx)
 
-            DONATES.append(
-                {
-                    'telegram_id': player_telegram_id,
-                    'username': player,
-                    'count': donates_count,
-                    'url': player_url,
-                }
-            )
+async def parse_tops(message: Message, emoji_replace: str) -> list:
+    """..."""
 
-        print(DONATES)
-        db.add_data(DONATES, False)
-    else:
-        pass
+    top_data = list()
+
+    msg_buffer = io.StringIO(message.text)
+    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
 
@@ -23,13 +27,26 @@ class DataBase(object):
 
     def connect(self) -> bool:
         """Выполняет подключение к базе данных"""
-        self.conn = psycopg2.connect(
-            dbname=self.name,
-            host=self.host,
-            user=self.user,
-            password=self.password,
-            port=self.port,
-        )
+
+        connection_kwargs = {
+            "host": self.host,
+            "port": self.port,
+            "user": self.user,
+            "password": self.password,
+            "database": self.name,
+        }
+        """Настройки соединения"""
+
+        keepalive_kwargs = {
+            "keepalives": 1,
+            "keepalives_idle": 5,
+            "keepalives_interval": 5,
+            "keepalives_count": 2,
+            "tcp_user_timeout": 1000,
+        }
+        """Настройки TCP"""
+
+        self.conn = psycopg2.connect(**connection_kwargs, **keepalive_kwargs)  # Выполняет подключение к базе данных
         self.conn.autocommit = True  # Активирует автоматический коммит транзакций
 
         return True
@@ -39,12 +56,23 @@ class DataBase(object):
         self.conn.close()
         return True
 
+    def check_connect(self) -> None:
+        """Проверяет статус подключения"""
+        try:
+            cur = self.conn.cursor()
+            cur.execute('SELECT 1')
+            cur.close()
+
+            return True
+        except Exception as e:
+            return False
+
     def commit(self) -> None:
         """Выполняет коммит"""
         self.conn.commit()
         return True
 
-    def add_data(self, data, wins: bool) -> bool:  # wins изменит на Enum
+    def add_data(self, data, wins: bool) -> bool:  # wins изменить на Enum
         """Добавляет данные в базу данных"""
         if wins:
             table = 'wins'

ClanStat/utils.py

@@ -1,3 +1,7 @@
+"""
+🧰 Утилиты
+"""
+
 # Импорт сторонних модулей
 from pyrogram.types import Message
 from pyrogram.enums import MessageEntityType
@@ -6,7 +10,7 @@ from . import client
 
 
 async def get_telegram_id(username: str) -> int:
-    """Возвращает Telegram ID"""
+    """🆔 Возвращает Telegram ID"""
 
     try:
         telegram_user = await client.get_users(username)
@@ -17,6 +21,7 @@ async def get_telegram_id(username: str) -> int:
 
 async def get_telegram_data(message: Message, idx: int) -> tuple():
     """..."""
+
     if message.entities[idx].type == MessageEntityType.TEXT_LINK:
         player_url = message.entities[idx].url
         player_telegram_id = await get_telegram_id(player_url.replace('http://t.me/', ''))
@@ -31,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

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

requirements.txt

@@ -1,5 +1,7 @@
 loguru==0.7.2
+psycopg2==2.9.10
 Pyrogram==2.0.106
 python-dotenv==1.0.1
+rich==14.0.0
 TgCrypto==1.2.5
-psycopg2==2.9.10
+zstandard==0.23.0

scripts/make-appimage.sh

@@ -1,7 +1,12 @@
 #!/bin/sh
 
+# author: Alexander Popov
+# mail: iiiypuk {at} fastmail.fm
+# license: Public Domain
+# desc: Собирает AppImage
+
 # Импортирует скрипт с определениями цветов
-source ../3rdparty/colors.sh/colors.sh
+source ./3rdparty/colors.sh/colors.sh
 
 APPDIR="../AppDir"
 GIT_HASH=$(git rev-parse --short HEAD)

scripts/mkdocs.sh

@@ -1,3 +1,15 @@
 #!/bin/sh
 
-pdoc -o docs/ app
+# author: Alexander Popov
+# mail: iiiypuk {at} fastmail.fm
+# license: Public Domain
+# desc: Генерирует документацию Python кода
+
+# Импортирует скрипт с определениями цветов
+source ./3rdparty/colors.sh/colors.sh
+
+echo -e "${CYAN}Генерация документации...${NC}"
+
+pdoc -o docs/ ClanStat
+
+echo -e "${GREEN}Генерация завершена. Смотрите директорию 'docs'.${NC}"