Compare commits

...

7 Commits

8 changed files with 76 additions and 20 deletions

View File

@ -0,0 +1,13 @@
### 2.1.0
- В БД сохраняется UTF-8 строки, вместо `\u`
### 2.0.0
- переход с SQLite3 на PostgreSQL
### 1.1.0
- Добавлен выхлом ошибок `try...catch` в логи
- У игроков с премиумом теперь отрезается символ 💎 из имени
- Добавлен обработчик `KeyboardInterrupt`

View File

@ -5,7 +5,7 @@ include_toc: true
# 🎴 Бот для ведения статистики клана в AniCard # 🎴 Бот для ведения статистики клана в AniCard
![](./assets/icon.png) <img src="https://git.a2s.su/AniCard/ClanStat/raw/branch/master/assets/icon.png" alt="Иконка" height="512">
## 📦 Возможности ## 📦 Возможности
@ -30,7 +30,11 @@ cp env.example .env
- `APP_HASH` — Хэш приложения Telegram. - `APP_HASH` — Хэш приложения Telegram.
- `ACCOUNT` — Имя аккаунта пользователя Telegram (_например `db_o_qp`_). - `ACCOUNT` — Имя аккаунта пользователя Telegram (_например `db_o_qp`_).
- `LOG_PATH` — Путь в файлу логов (_например `/tmp/test.log`_). - `LOG_PATH` — Путь в файлу логов (_например `/tmp/test.log`_).
- `DB_PATH` — Путь к файлу базы данных SQLite3 (_например `./db.sqlite`_). - `DB_HOST` — Хост базы данных PostgreSQL
- `DB_PORT` — Порт сервера базы данных
- `DB_NAME` — Имя базы данных
- `DB_USER` — Имя пользователя
- `DB_PASSWORD` — Пароль пользователя
### 🐦‍🔥 Настройка виртуального окружения и загрузка бибилотек ### 🐦‍🔥 Настройка виртуального окружения и загрузка бибилотек
@ -80,4 +84,5 @@ Enter confirmation code: 14628
- **Pyrogram** — Telegram клиент - **Pyrogram** — Telegram клиент
- **python-dotenv** — Работа с `.env` файлами - **python-dotenv** — Работа с `.env` файлами
- **loguru** — Логирование - **loguru** — Логирование
- **psycopg2** — Работа с PostgreSQL
- **black** — Форматирование кода - **black** — Форматирование кода

View File

@ -1,6 +1,6 @@
# Информация о программе # Информация о программе
__author__ = 'Alexander Popov' __author__ = 'Alexander Popov'
__version__ = (1, 0, 0) __version__ = (2, 1, 0)
# Импорт модулей стандартной библиотеки # Импорт модулей стандартной библиотеки
from os import getenv from os import getenv
@ -18,7 +18,13 @@ load_dotenv() # Выполняет чтение .env
client = Client(getenv('ACCOUNT'), api_id=getenv('APP_ID'), api_hash=getenv('APP_HASH')) client = Client(getenv('ACCOUNT'), api_id=getenv('APP_ID'), api_hash=getenv('APP_HASH'))
"""Клиент Telegram""" """Клиент Telegram"""
db = DataBase(getenv('DB_PATH')) db = DataBase(
getenv('DB_HOST'),
int(getenv('DB_PORT')),
getenv('DB_NAME'),
getenv('DB_USER'),
getenv('DB_PASSWORD'),
)
"""База данных""" """База данных"""
logger.add(getenv('LOG_PATH'), compression='zip') logger.add(getenv('LOG_PATH'), compression='zip')

View File

@ -17,11 +17,13 @@ async def main():
await get_top_wins() # Получает топ клана по победам await get_top_wins() # Получает топ клана по победам
except Exception as e: except Exception as e:
logger.error('Ошибка получения списка побед') logger.error('Ошибка получения списка побед')
logger.error(e)
try: try:
await get_top_donates() # Получает топ клана по пожертвованиям await get_top_donates() # Получает топ клана по пожертвованиям
except Exception as e: except Exception as e:
logger.error('Ошибка получения списка пожертвований') logger.error('Ошибка получения списка пожертвований')
logger.error(e)
# Приостанавливает выполнение цикла # Приостанавливает выполнение цикла
time.sleep(DELAY_TIME) time.sleep(DELAY_TIME)
@ -36,6 +38,10 @@ if __name__ == '__main__':
logger.info('Выполняется подключение к базе данных') logger.info('Выполняется подключение к базе данных')
db.connect() db.connect()
try:
client.run(main()) client.run(main())
except KeyboardInterrupt:
pass
db.close() db.close()
logger.info('Приложение завершило работу') logger.info('Приложение завершило работу')

View File

@ -25,6 +25,7 @@ def parse_wins_top(message: Message) -> None:
# gamer = re.sub(r'\s[⚡⚜]$', '', gamer) # gamer = re.sub(r'\s[⚡⚜]$', '', gamer)
gamer = gamer.replace('', '') gamer = gamer.replace('', '')
gamer = gamer.replace('', '').replace('\ufe0f', '') gamer = gamer.replace('', '').replace('\ufe0f', '')
gamer = gamer.replace(' 💎', '')
battle_count = battle_count.replace('', '') # удаляем эмодзи мечей battle_count = battle_count.replace('', '') # удаляем эмодзи мечей
battle_count = int(re.sub(r'[^\x00-\x7F]', '', battle_count)) # преобразовывает строку в число battle_count = int(re.sub(r'[^\x00-\x7F]', '', battle_count)) # преобразовывает строку в число
@ -65,6 +66,7 @@ def parse_donates_top(message: Message) -> None:
# player = re.sub(r'\s[⚡⚜]$', '', player) # player = re.sub(r'\s[⚡⚜]$', '', player)
player = player.replace('', '') player = player.replace('', '')
player = player.replace('', '').replace('\ufe0f', '') player = player.replace('', '').replace('\ufe0f', '')
player = player.replace(' 💎', '')
donates_count = donates_count.replace(' 💠', '') # удаляем эмодзи пожертвований donates_count = donates_count.replace(' 💠', '') # удаляем эмодзи пожертвований
donates_count = int(re.sub(r'[^\x00-\x7F]', '', donates_count)) # преобразовывает строку в число donates_count = int(re.sub(r'[^\x00-\x7F]', '', donates_count)) # преобразовывает строку в число

View File

@ -1,19 +1,35 @@
# Импорт модулей стандартной библиотеки # Импорт модулей стандартной библиотеки
import sqlite3
import json import json
# Импорт сторонних модулей
import psycopg2
class DataBase(object): class DataBase(object):
"""Класс для работы с базой данных""" """Класс для работы с базой данных"""
def __init__(self, path): def __init__(self, host: str, port: int, name: str, user: str, password: str):
super(DataBase, self).__init__() super(DataBase, self).__init__()
self.path = path self.host = host
"""Путь к файлу базы данных SQLite3""" """Хост PostgreSQL базы данных"""
self.port = port
"""Порт сервера базы данных"""
self.name = name
"""Имя базы данных"""
self.user = user
"""Имя пользователя"""
self.password = password
"""Пароль пользователя"""
def connect(self) -> bool: def connect(self) -> bool:
"""Выполняет подключение к базе данных""" """Выполняет подключение к базе данных"""
self.conn = sqlite3.connect(self.path) self.conn = psycopg2.connect(
dbname=self.name,
host=self.host,
user=self.user,
password=self.password,
port=self.port,
)
return True return True
def close(self) -> bool: def close(self) -> bool:
@ -34,7 +50,15 @@ class DataBase(object):
table = 'donates' table = 'donates'
cur = self.conn.cursor() cur = self.conn.cursor()
cur.execute('INSERT INTO {table} (\'data\') VALUES (\'{data}\')'.format(table=table, data=json.dumps(data))) cur.execute(
'INSERT INTO {table} ("data") VALUES (\'{data}\')'.format(
table=table,
data=json.dumps(
data,
ensure_ascii=False,
),
)
)
self.commit() self.commit()
cur.close() cur.close()

View File

@ -2,4 +2,8 @@ APP_ID=""
APP_HASH="" APP_HASH=""
ACCOUNT="" ACCOUNT=""
LOG_PATH="/tmp/test.log" LOG_PATH="/tmp/test.log"
DB_PATH="./db.sqlite" DB_HOST=""
DB_PORT=""
DB_NAME=""
DB_USER=""
DB_PASSWORD=""

View File

@ -1,12 +1,8 @@
BEGIN; BEGIN;
CREATE TABLE IF NOT EXISTS 'wins' CREATE TABLE "public"."wins" (
( "data" JSON NOT NULL,
'id' Integer PRIMARY KEY AUTOINCREMENT, "timestamp" Timestamp Without Time Zone DEFAULT CURRENT_TIMESTAMP );
'timestamp' DateTime NOT NULL DEFAULT (datetime(CURRENT_TIMESTAMP, 'localtime')), ;
'data' Text NOT NULL
);
CREATE INDEX IF NOT EXISTS 'index_timestamp' ON 'wins' ('timestamp');
COMMIT; COMMIT;