45 строки
2.0 KiB
Python
45 строки
2.0 KiB
Python
from aiogram import types
|
|
from aiogram.dispatcher import FSMContext
|
|
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
|
|
from aiogram.dispatcher.filters.state import State, StatesGroup
|
|
import subprocess
|
|
from loader import dp, bot # Убедись, что loader.py существует
|
|
|
|
class AddClient(StatesGroup):
|
|
protocol = State()
|
|
username = State()
|
|
|
|
@dp.message_handler(commands=["add_client"])
|
|
async def cmd_add_client(message: types.Message):
|
|
keyboard = InlineKeyboardMarkup()
|
|
keyboard.add(
|
|
InlineKeyboardButton("WireGuard", callback_data="proto_wg"),
|
|
InlineKeyboardButton("XRay", callback_data="proto_xray")
|
|
)
|
|
await message.answer("Выберите протокол для нового клиента:", reply_markup=keyboard)
|
|
await AddClient.protocol.set()
|
|
|
|
@dp.callback_query_handler(lambda c: c.data.startswith("proto_"), state=AddClient.protocol)
|
|
async def process_protocol_choice(callback_query: types.CallbackQuery, state: FSMContext):
|
|
protocol = callback_query.data.replace("proto_", "")
|
|
await state.update_data(protocol=protocol)
|
|
await bot.send_message(callback_query.from_user.id, f"Выбран протокол: {protocol.upper()}\nВведите имя клиента:")
|
|
await AddClient.username.set()
|
|
|
|
@dp.message_handler(state=AddClient.username)
|
|
async def process_username(message: types.Message, state: FSMContext):
|
|
data = await state.get_data()
|
|
username = message.text.strip()
|
|
protocol = data.get("protocol", "wg")
|
|
|
|
try:
|
|
subprocess.run(
|
|
["amnezia", "client", "add", "--proto", protocol, "--name", username],
|
|
check=True
|
|
)
|
|
await message.answer(f"✅ Клиент `{username}` с протоколом `{protocol.upper()}` успешно создан.", parse_mode="Markdown")
|
|
except subprocess.CalledProcessError as e:
|
|
await message.answer(f"❌ Ошибка при создании клиента:\n```\n{str(e)}\n```", parse_mode="Markdown")
|
|
|
|
await state.finish()
|
© 2026 StackWorks