From 38aeaa9d171fb4888e3f74a839a007e81de5d2b1 Mon Sep 17 00:00:00 2001 From: Vlad Date: Mon, 16 Sep 2024 03:43:25 +0300 Subject: [PATCH] =?UTF-8?q?=D0=BB=D0=BE=D0=B3=D0=B8=D0=BA=D0=B0=20=D1=81?= =?UTF-8?q?=D1=82=D0=B0=D1=82=D0=B8=D1=81=D1=82=D0=B8=D0=BA=D0=B8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- bot.py | 38 +++++++++++++++++++++++++++----------- 1 file changed, 27 insertions(+), 11 deletions(-) diff --git a/bot.py b/bot.py index dba0985d..e98a009e 100644 --- a/bot.py +++ b/bot.py @@ -4,14 +4,15 @@ from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message from aiogram.fsm.storage.memory import MemoryStorage from aiogram.fsm.context import FSMContext from aiogram.fsm.state import State, StatesGroup -from auth import login_with_credentials, link +from auth import login_with_credentials, link, get_clients from client import add_client, generate_client_id from datetime import datetime, timedelta from config import API_TOKEN, ADMIN_PASSWORD, ADMIN_USERNAME from database import add_connection, DATABASE_PATH import uuid import aiosqlite - +import requests +import re class Form(StatesGroup): waiting_for_key_name = State() @@ -22,12 +23,13 @@ storage = MemoryStorage() dp = Dispatcher(bot=bot, storage=storage) router = Router() +# Создаем сессию при старте бота +session = login_with_credentials(ADMIN_USERNAME, ADMIN_PASSWORD) + @dp.message(Command("start")) async def start_command(message: Message): - # Приветственное сообщение welcome_text = "Добро пожаловать! Вы можете создать ключ для подключения VPN или просмотреть статистику использования." - # Кнопки для меню button_create_key = InlineKeyboardButton(text='Создать ключ', callback_data='create_key') button_view_stats = InlineKeyboardButton(text='Посмотреть статистику', callback_data='view_stats') keyboard = InlineKeyboardMarkup(inline_keyboard=[[button_create_key], [button_view_stats]]) @@ -43,11 +45,28 @@ async def process_callback_create_key(callback_query: types.CallbackQuery, state @dp.callback_query(F.data == 'view_stats') async def process_callback_view_stats(callback_query: types.CallbackQuery, state: FSMContext): tg_id = callback_query.from_user.id - # Функция для получения статистики (вы должны реализовать эту функцию в соответствующем модуле) - # statistics = get_statistics(tg_id) - # Для примера - statistics = "Здесь будет ваша статистика" + try: + async with aiosqlite.connect(DATABASE_PATH) as db: + async with db.execute("SELECT email FROM connections WHERE tg_id = ? AND expiry_time > ?", (tg_id, int(datetime.utcnow().timestamp() * 1000))) as cursor: + record = await cursor.fetchone() + if record: + email = record[0] + connection_link = link(session, email) + + # Извлечение данных о загрузке и выгрузке из ссылки + up_match = re.search(r'up=(\d+)', connection_link) + down_match = re.search(r'down=(\d+)', connection_link) + + up = up_match.group(1) if up_match else "Неизвестно" + down = down_match.group(1) if down_match else "Неизвестно" + + statistics = f"Статистика вашего ключа:\nЗагрузка: {up} MB\nВыгрузка: {down} MB" + else: + statistics = "У вас нет активных ключей." + + except Exception as e: + statistics = f"Ошибка при получении статистики: {e}" await callback_query.message.reply(f"Ваша статистика:\n{statistics}") await callback_query.answer() @@ -68,8 +87,6 @@ async def handle_text(message: types.Message, state: FSMContext): await message.reply("У вас уже есть активный ключ. Один клиент может иметь только один активный ключ.") return - session = login_with_credentials(ADMIN_USERNAME, ADMIN_PASSWORD) - # Создание уникального ID клиента client_id = str(uuid.uuid4()) email = message.text @@ -96,7 +113,6 @@ async def handle_text(message: types.Message, state: FSMContext): except Exception as e: await message.reply(f"Ошибка: {e}") -# Запуск бота async def main(): dp.include_router(router) # Подключение роутера await dp.start_polling(bot) # Запуск поллинга