кнопка статистики

.gitignore

@@ -3,3 +3,5 @@
 /vpn_users.db
 /config.py
 /database.db
+/bot_old.py
+/bot_old_2.py

auth.py

@@ -62,4 +62,15 @@ def link(session, user_id: str):
     # Создание ссылки для подключения VLESS
     val = f"vless://{client_id}@{DOMEN}?type={tcp}&security={reality}&fp=chrome&pbk=ZIMoEnEd8-qMJkReVU5JxbiEj8CCSrvpm_ckvJ-46TE&sni=yahoo.com&sid=0fb4b595&spx=%2F&flow={flow}#VPN_F-{email}"
     
-    return val
+    return val
+
+def get_statistics(session, tg_id: int):
+    # Пример URL для получения статистики
+    STATS_URL = f"https://vpn.pocomacho.ru:34268/solonet/panel/api/inbounds/stats/{tg_id}"
+    response = session.get(STATS_URL)
+    if response.status_code == 200:
+        stats = response.json()
+        # Форматирование статистики для отображения
+        return f"Загрузка: {stats['upload']} MB\nВывод: {stats['download']} MB"
+    else:
+        raise Exception(f"Ошибка при получении статистики: {response.status_code}, {response.text}")

bot.py

@@ -13,9 +13,9 @@ import uuid
 import aiosqlite
 
 
-
 class Form(StatesGroup):
     waiting_for_key_name = State()
+    waiting_for_statistics = State()
 
 bot = Bot(token=API_TOKEN)
 storage = MemoryStorage()
@@ -24,16 +24,32 @@ router = Router()
 
 @dp.message(Command("start"))
 async def start_command(message: Message):
-    button_create_key = InlineKeyboardButton(text='Создать ключ', callback_data='create_key')
+    # Приветственное сообщение
-    keyboard = InlineKeyboardMarkup(inline_keyboard=[[button_create_key]])
+    welcome_text = "Добро пожаловать! Вы можете создать ключ для подключения VPN или просмотреть статистику использования."
     
-    await message.reply("Выберите действие:", reply_markup=keyboard)
+    # Кнопки для меню
+    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]])
+    
+    await message.reply(welcome_text, reply_markup=keyboard)
 
 @dp.callback_query(F.data == 'create_key')
 async def process_callback_create_key(callback_query: types.CallbackQuery, state: FSMContext):
     await callback_query.message.reply("Введите имя вашего профиля VPN:")
     await state.set_state(Form.waiting_for_key_name)
+    await callback_query.answer()
 
+@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 = "Здесь будет ваша статистика"
+    
+    await callback_query.message.reply(f"Ваша статистика:\n{statistics}")
     await callback_query.answer()
 
 @dp.message()
@@ -42,45 +58,49 @@ async def handle_text(message: types.Message, state: FSMContext):
     
     if current_state == Form.waiting_for_key_name.state:
         try:
+            tg_id = message.from_user.id
+            
+            # Проверка на наличие активного ключа
+            async with aiosqlite.connect(DATABASE_PATH) as db:
+                async with db.execute("SELECT COUNT(*) FROM connections WHERE tg_id = ? AND expiry_time > ?", (tg_id, int(datetime.utcnow().timestamp() * 1000))) as cursor:
+                    count = await cursor.fetchone()
+                    if count[0] > 0:
+                        await message.reply("У вас уже есть активный ключ. Один клиент может иметь только один активный ключ.")
+                        return
+            
             session = login_with_credentials(ADMIN_USERNAME, ADMIN_PASSWORD)
 
-            # Параметры клиента
+            # Создание уникального ID клиента
+            client_id = str(uuid.uuid4())
             email = message.text
-            tg_id = message.from_user.id
+            limit_ip = 1
+            total_gb = 0
+            current_time = datetime.utcnow()
+            expiry_time = int((current_time + timedelta(days=30)).timestamp() * 1000)
+            enable = True
+            flow = "xtls-rprx-vision"
 
-            # Проверяем наличие активного ключа
+            add_client(session, client_id, email, tg_id, limit_ip, total_gb, expiry_time, enable, flow)
-            async with aiosqlite.connect(DATABASE_PATH) as db:
-                async with db.execute('''
-                    SELECT * FROM connections
-                    WHERE tg_id = ? AND expiry_time > ?
-                ''', (tg_id, int(datetime.utcnow().timestamp() * 1000))) as cursor:
-                    existing_key = await cursor.fetchone()
-            
-            if existing_key:
-                await message.reply("У вас уже есть активный ключ. Вы не можете создать больше одного ключа.")
-            else:
-                # Генерация нового ключа
-                client_id = str(uuid.uuid4())
-                limit_ip = 1
-                total_gb = 0
-                current_time = datetime.utcnow()
-                expiry_time = int((current_time + timedelta(days=30)).timestamp() * 1000)
-                enable = True
-                flow = "xtls-rprx-vision"
 
-                # Добавление клиента (функция для создания клиента)
+            # Получение ссылки на подключение
-                result = add_client(session, client_id, email, tg_id, limit_ip, total_gb, expiry_time, enable, flow)
+            connection_link = link(session, email)
 
-                # Сохранение данных в базу данных
+            # Сохранение данных в базу данных
-                await add_connection(tg_id, client_id, email, expiry_time)
+            await add_connection(tg_id, client_id, email, expiry_time)
 
-                # Получение ссылки на подключение
+            # Отправка ключа в виде цитаты
-                connection_link = link(session, email)
+            await message.reply(f"Ключ создан:\n<pre>{connection_link}</pre>", parse_mode="HTML")
-
-                # Отправка ключа в виде цитаты
-                await message.reply(f"Ключ создан:\n<pre>{connection_link}</pre>", parse_mode="HTML")
 
             # Сброс состояния
             await state.clear()
         except Exception as e:
-            await message.reply(f"Произошла ошибка: {e}")
+            await message.reply(f"Ошибка: {e}")
+
+# Запуск бота
+async def main():
+    dp.include_router(router)  # Подключение роутера
+    await dp.start_polling(bot)  # Запуск поллинга
+
+if __name__ == '__main__':
+    import asyncio
+    asyncio.run(main())