hteppl
786 lines
24 KiB
Python
786 lines
24 KiB
Python
import asyncio
|
||
from datetime import datetime
|
||
from typing import Any
|
||
|
||
from aiogram import F, Router, types
|
||
from aiogram.fsm.context import FSMContext
|
||
from aiogram.fsm.state import State, StatesGroup
|
||
from aiogram.types import CallbackQuery
|
||
|
||
from config import TOTAL_GB
|
||
from database import delete_user_data, get_client_id_by_email, get_servers_from_db, restore_trial, update_key_expiry
|
||
from filters.admin import IsAdminFilter
|
||
from handlers.keys.key_utils import (
|
||
delete_key_from_cluster,
|
||
delete_key_from_db,
|
||
renew_key_in_cluster, update_subscription,
|
||
)
|
||
from handlers.utils import sanitize_key_name
|
||
from keyboards.admin.panel_kb import AdminPanelCallback, build_admin_back_kb
|
||
from keyboards.admin.users_kb import build_user_edit_kb, build_key_edit_kb, build_key_delete_kb, \
|
||
build_user_delete_kb, AdminUserEditorCallback, build_editor_kb, build_users_balance_kb, \
|
||
build_users_balance_change_kb, build_user_key_kb
|
||
from logger import logger
|
||
|
||
router = Router()
|
||
|
||
|
||
class UserEditorState(StatesGroup):
|
||
# search
|
||
waiting_for_user_data = State()
|
||
waiting_for_key_name = State()
|
||
# updating data
|
||
waiting_for_balance = State()
|
||
waiting_for_expiry_time = State()
|
||
waiting_for_message_text = State()
|
||
|
||
|
||
@router.callback_query(
|
||
AdminPanelCallback.filter(F.action == "search_user"),
|
||
IsAdminFilter(),
|
||
)
|
||
async def handle_search_user(
|
||
callback_query: CallbackQuery,
|
||
state: FSMContext
|
||
):
|
||
text = (
|
||
"🔍 Введите ID или Username пользователя для поиска:"
|
||
"\n\n🆔 ID - числовой айди"
|
||
"\n📝 Username - юзернейм пользователя"
|
||
)
|
||
|
||
await state.set_state(UserEditorState.waiting_for_user_data)
|
||
await callback_query.message.edit_text(
|
||
text=text,
|
||
reply_markup=build_admin_back_kb()
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminPanelCallback.filter(F.action == "search_key"),
|
||
IsAdminFilter(),
|
||
)
|
||
async def handle_search_key(
|
||
callback_query: CallbackQuery,
|
||
state: FSMContext
|
||
):
|
||
await state.set_state(UserEditorState.waiting_for_key_name)
|
||
await callback_query.message.edit_text(
|
||
text="🔑 Введите имя ключа для поиска:",
|
||
reply_markup=build_admin_back_kb()
|
||
)
|
||
|
||
|
||
@router.message(
|
||
UserEditorState.waiting_for_user_data,
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_user_data_input(
|
||
message: types.Message,
|
||
state: FSMContext,
|
||
session: Any
|
||
):
|
||
kb = build_admin_back_kb()
|
||
|
||
if not message.text:
|
||
await message.answer(
|
||
text="🚫 Пожалуйста, отправьте текстовое сообщение.",
|
||
reply_markup=kb
|
||
)
|
||
return
|
||
|
||
if message.text.isdigit():
|
||
tg_id = int(message.text)
|
||
else:
|
||
# Удаление '@' символа в начале сообщения
|
||
username = message.text.strip().lstrip('@')
|
||
# Удаление начала ссылки на профиль
|
||
username = username.replace('https://t.me/', '')
|
||
|
||
user = await session.fetchrow(
|
||
"SELECT tg_id FROM users WHERE username = $1", username
|
||
)
|
||
|
||
if not user:
|
||
await message.answer(
|
||
text="🚫 Пользователь с указанным Username не найден!",
|
||
reply_markup=kb,
|
||
)
|
||
return
|
||
|
||
tg_id = user["tg_id"]
|
||
|
||
await process_user_search(message, state, session, tg_id)
|
||
|
||
|
||
@router.message(
|
||
UserEditorState.waiting_for_key_name,
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_key_name_input(
|
||
message: types.Message,
|
||
state: FSMContext,
|
||
session: Any
|
||
):
|
||
kb = build_admin_back_kb()
|
||
|
||
if not message.text:
|
||
await message.answer(
|
||
text="🚫 Пожалуйста, отправьте текстовое сообщение.",
|
||
reply_markup=kb
|
||
)
|
||
return
|
||
|
||
key_name = sanitize_key_name(message.text)
|
||
key_details = await get_key_details(key_name, session)
|
||
|
||
if not key_details:
|
||
await message.answer(
|
||
text="🚫 Пользователь с указанным именем ключа не найден.",
|
||
reply_markup=kb
|
||
)
|
||
return
|
||
|
||
await process_user_search(message, state, session, key_details["tg_id"])
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_send_message"),
|
||
IsAdminFilter(),
|
||
)
|
||
async def handle_send_message(
|
||
callback_query: types.CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
state: FSMContext
|
||
):
|
||
tg_id = callback_data.tg_id
|
||
|
||
await callback_query.message.edit_text(
|
||
text="✉️ Введите текст сообщения, которое вы хотите отправить пользователю:",
|
||
reply_markup=build_editor_kb(tg_id)
|
||
)
|
||
|
||
await state.update_data(tg_id=tg_id)
|
||
await state.set_state(UserEditorState.waiting_for_message_text)
|
||
|
||
|
||
@router.message(
|
||
UserEditorState.waiting_for_message_text,
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_message_text_input(
|
||
message: types.Message,
|
||
state: FSMContext
|
||
):
|
||
data = await state.get_data()
|
||
tg_id = data.get("tg_id")
|
||
|
||
try:
|
||
await message.bot.send_message(
|
||
chat_id=tg_id,
|
||
text=message.text
|
||
)
|
||
await message.answer(
|
||
text="✅ Сообщение успешно отправлено.",
|
||
reply_markup=build_editor_kb(tg_id)
|
||
)
|
||
except Exception as e:
|
||
await message.answer(
|
||
text=f"❌ Не удалось отправить сообщение: {e}",
|
||
reply_markup=build_editor_kb(tg_id)
|
||
)
|
||
|
||
await state.clear()
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_trial_restore"),
|
||
IsAdminFilter(),
|
||
)
|
||
async def handle_trial_restore(
|
||
callback_query: types.CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
session: Any
|
||
):
|
||
tg_id = callback_data.tg_id
|
||
|
||
await restore_trial(tg_id, session)
|
||
await callback_query.message.edit_text(
|
||
text="✅ Триал успешно восстановлен!",
|
||
reply_markup=build_editor_kb(tg_id)
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_balance_edit"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_balance_change(
|
||
callback_query: CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
session: Any
|
||
):
|
||
tg_id = callback_data.tg_id
|
||
|
||
records = await session.fetch("""
|
||
SELECT amount, payment_system, status, created_at
|
||
FROM payments
|
||
WHERE tg_id = $1
|
||
ORDER BY created_at DESC
|
||
LIMIT 5
|
||
""", tg_id)
|
||
|
||
balance = await get_user_balance(tg_id, session)
|
||
|
||
text = (
|
||
f"<b>💵 Изменение баланса</b>"
|
||
f"\n\n🆔 ID: <b>{tg_id}</b>"
|
||
f"\n💰 Баланс: <b>{balance}Р</b>"
|
||
f"\n📊 Последние операции:"
|
||
)
|
||
|
||
if records:
|
||
for record in records:
|
||
amount = record["amount"]
|
||
payment_system = record["payment_system"]
|
||
status = record["status"]
|
||
date = record["created_at"].strftime("%Y-%m-%d %H:%M:%S")
|
||
text += (
|
||
f"\n\n<blockquote>Сумма: {amount} | {payment_system}"
|
||
f"\nСтатус: {status}"
|
||
f"\nДата: {date}</blockquote>"
|
||
)
|
||
else:
|
||
text += "\n <i>🚫 Отсутствуют</i>"
|
||
|
||
await callback_query.message.edit_text(
|
||
text=text,
|
||
reply_markup=build_users_balance_kb(tg_id)
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_balance_add"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_balance_add(
|
||
callback_query: CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
state: FSMContext,
|
||
session: Any
|
||
):
|
||
tg_id = callback_data.tg_id
|
||
amount = callback_data.data
|
||
|
||
if amount:
|
||
await add_user_balance(tg_id, int(amount), session)
|
||
await handle_balance_change(callback_query, callback_data, session)
|
||
return
|
||
|
||
await state.update_data(tg_id=tg_id, op_type="add")
|
||
await state.set_state(UserEditorState.waiting_for_balance)
|
||
|
||
await callback_query.message.answer(
|
||
text="✍️ Введите сумму, которую хотите добавить на баланс пользователя:",
|
||
reply_markup=build_users_balance_change_kb(tg_id)
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_balance_take"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_balance_add(
|
||
callback_query: CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
state: FSMContext
|
||
):
|
||
tg_id = callback_data.tg_id
|
||
|
||
await state.update_data(tg_id=tg_id, op_type="take")
|
||
await state.set_state(UserEditorState.waiting_for_balance)
|
||
|
||
await callback_query.message.answer(
|
||
text="✍️ Введите сумму, которую хотите добавить на баланс пользователя:",
|
||
reply_markup=build_users_balance_change_kb(tg_id)
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_balance_set"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_balance_add(
|
||
callback_query: CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
state: FSMContext
|
||
):
|
||
tg_id = callback_data.tg_id
|
||
|
||
await state.update_data(tg_id=tg_id, op_type="set")
|
||
await state.set_state(UserEditorState.waiting_for_balance)
|
||
|
||
await callback_query.message.answer(
|
||
text="✍️ Введите баланс, который хотите установить пользователю:",
|
||
reply_markup=build_users_balance_change_kb(tg_id)
|
||
)
|
||
|
||
|
||
@router.message(
|
||
UserEditorState.waiting_for_balance,
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_balance_input(
|
||
message: types.Message,
|
||
state: FSMContext,
|
||
session: Any
|
||
):
|
||
data = await state.get_data()
|
||
tg_id = data.get("tg_id")
|
||
op_type = data.get("op_type")
|
||
|
||
if not message.text.isdigit() or int(message.text) < 0:
|
||
await message.answer(
|
||
text="🚫 Пожалуйста, введите корректную сумму!",
|
||
reply_markup=build_users_balance_change_kb(tg_id)
|
||
)
|
||
return
|
||
|
||
amount = int(message.text)
|
||
|
||
if op_type == "add":
|
||
text = f"✅ К балансу пользователя добавлено <b>{amount}Р</b>"
|
||
await add_user_balance(tg_id, amount, session)
|
||
elif op_type == "take":
|
||
text = f"✅ Из баланса пользователя было вычтено <b>{amount}Р</b>"
|
||
await add_user_balance(tg_id, -amount, session)
|
||
else:
|
||
text = f"✅ Баланс пользователя изменен на <b>{amount}Р</b>"
|
||
await set_user_balance(tg_id, amount, session)
|
||
|
||
await message.answer(
|
||
text=text,
|
||
reply_markup=build_users_balance_change_kb(tg_id)
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_key_edit"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_key_edit(
|
||
callback_query: CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
session: Any
|
||
):
|
||
email = callback_data.data
|
||
key_details = await get_key_details(email, session)
|
||
|
||
if not key_details:
|
||
await callback_query.message.edit_text(
|
||
text="🚫 Информация о ключе не найдена.",
|
||
reply_markup=build_editor_kb(callback_data.tg_id),
|
||
)
|
||
return
|
||
|
||
text = (
|
||
f"<b>🔑 Информация о ключе</b>"
|
||
f"\n\n<code>{key_details['key']}</code>"
|
||
f"\n\n⏰ Дата истечения: <b>{key_details['expiry_date']}</b>"
|
||
f"\n🌐 Кластер: <b>{key_details['server_name']}</b>"
|
||
f"\n🆔 ID клиента: <b>{key_details['tg_id']}</b>"
|
||
)
|
||
|
||
await callback_query.message.edit_text(
|
||
text=text,
|
||
reply_markup=build_key_edit_kb(key_details, email)
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_change_expiry"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_change_expiry(
|
||
callback_query: CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
state: FSMContext
|
||
):
|
||
email = callback_data.data
|
||
|
||
await callback_query.message.edit_text(
|
||
text=f"✍️ Введите новое время истечения для ключа <b>{email}</b> в формате <code>YYYY-MM-DD HH:MM:SS</code>:"
|
||
)
|
||
|
||
await state.update_data(tg_id=callback_data.tg_id, email=email)
|
||
await state.set_state(UserEditorState.waiting_for_expiry_time)
|
||
|
||
|
||
@router.message(
|
||
UserEditorState.waiting_for_expiry_time,
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_expiry_time_input(
|
||
message: types.Message,
|
||
state: FSMContext,
|
||
session: Any
|
||
):
|
||
user_data = await state.get_data()
|
||
email = user_data.get("email")
|
||
|
||
try:
|
||
expiry_time = int(
|
||
datetime.strptime(message.text, "%Y-%m-%d %H:%M:%S").timestamp() * 1000
|
||
)
|
||
|
||
client_id = await get_client_id_by_email(email)
|
||
|
||
if client_id is None:
|
||
await message.edit_text(
|
||
text=f"🚫 Клиент с Email {email} не найден. 🔍",
|
||
reply_markup=build_admin_back_kb(),
|
||
)
|
||
await state.clear()
|
||
return
|
||
|
||
server_id = await session.fetchrow(
|
||
"SELECT server_id FROM keys WHERE client_id = $1", client_id
|
||
)
|
||
|
||
if not server_id:
|
||
await message.edit_text(
|
||
text="🚫 Клиент не найден в базе данных. 🔍",
|
||
reply_markup=build_admin_back_kb(),
|
||
)
|
||
await state.clear()
|
||
return
|
||
|
||
clusters = await get_servers_from_db()
|
||
|
||
async def update_key_on_all_servers():
|
||
tasks = [
|
||
asyncio.create_task(
|
||
renew_key_in_cluster(
|
||
cluster_name,
|
||
email,
|
||
client_id,
|
||
expiry_time,
|
||
total_gb=TOTAL_GB,
|
||
)
|
||
)
|
||
for cluster_name in clusters
|
||
]
|
||
|
||
await asyncio.gather(*tasks)
|
||
|
||
await update_key_on_all_servers()
|
||
await update_key_expiry(client_id, expiry_time)
|
||
|
||
response_message = f"✅ Время истечения ключа для клиента {client_id} ({email}) успешно обновлено на всех серверах."
|
||
|
||
await message.edit_text(
|
||
text=response_message,
|
||
reply_markup=build_admin_back_kb()
|
||
)
|
||
except ValueError:
|
||
tg_id = user_data.get("tg_id")
|
||
await message.edit_text(
|
||
text="❌ Пожалуйста, используйте формат: YYYY-MM-DD HH:MM:SS.",
|
||
reply_markup=build_editor_kb(tg_id),
|
||
)
|
||
except Exception as e:
|
||
logger.error(e)
|
||
await state.clear()
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_update_key"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_update_key(
|
||
callback_query: CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
session: Any
|
||
):
|
||
tg_id = callback_data.tg_id
|
||
email = callback_data.data
|
||
|
||
try:
|
||
await update_subscription(tg_id, email, session)
|
||
await handle_key_edit(callback_query, callback_data, session)
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при обновлении ключа {email} администратором: {e}")
|
||
await callback_query.message.answer(
|
||
text=f"❗ Произошла ошибка при обновлении ключа: {e}",
|
||
reply_markup=build_user_key_kb(tg_id, email)
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_delete_key"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_delete_key(
|
||
callback_query: types.CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
session: Any
|
||
):
|
||
email = callback_data.data
|
||
client_id = await session.fetchval(
|
||
"SELECT client_id FROM keys WHERE email = $1", email
|
||
)
|
||
|
||
if client_id is None:
|
||
await callback_query.message.edit_text(
|
||
text="🚫 Ключ не найден!",
|
||
reply_markup=build_editor_kb(callback_data.tg_id)
|
||
)
|
||
return
|
||
|
||
await callback_query.message.edit_text(
|
||
text="❓ Вы уверены, что хотите удалить ключ?",
|
||
reply_markup=build_key_delete_kb(callback_data.tg_id, client_id)
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_delete_key_confirm"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_delete_key_confirm(
|
||
callback_query: types.CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
session: Any
|
||
):
|
||
client_id = callback_data.data
|
||
record = await session.fetchrow(
|
||
"SELECT email FROM keys WHERE client_id = $1", client_id
|
||
)
|
||
|
||
kb = build_editor_kb(callback_data.tg_id)
|
||
|
||
if record:
|
||
clusters = await get_servers_from_db()
|
||
|
||
async def delete_key_from_servers(email, client_id):
|
||
tasks = []
|
||
for cluster_name, cluster_servers in clusters.items():
|
||
for server in cluster_servers:
|
||
tasks.append(
|
||
delete_key_from_cluster(cluster_name, email, client_id)
|
||
)
|
||
await asyncio.gather(*tasks)
|
||
|
||
await delete_key_from_servers(record["email"], client_id)
|
||
await delete_key_from_db(client_id, session)
|
||
|
||
await callback_query.message.edit_text(
|
||
text="✅ Ключ успешно удален.",
|
||
reply_markup=kb
|
||
)
|
||
else:
|
||
await callback_query.message.edit_text(
|
||
text="🚫 Ключ не найден или уже удален.",
|
||
reply_markup=kb
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_delete_user"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_delete_user(
|
||
callback_query: types.CallbackQuery,
|
||
callback_data: AdminUserEditorCallback
|
||
):
|
||
tg_id = callback_data.tg_id
|
||
await callback_query.message.edit_text(
|
||
text=f"❗️ Вы уверены, что хотите удалить пользователя с ID {tg_id}?",
|
||
reply_markup=build_user_delete_kb(tg_id)
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_delete_user_confirm"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_delete_user_confirm(
|
||
callback_query: types.CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
session: Any
|
||
):
|
||
tg_id = callback_data.tg_id
|
||
key_records = await session.fetch("SELECT email, client_id FROM keys WHERE tg_id = $1", tg_id)
|
||
|
||
async def delete_keys_from_servers():
|
||
try:
|
||
tasks = []
|
||
for email, client_id in key_records:
|
||
servers = await get_servers_from_db()
|
||
for cluster_id, cluster in servers.items():
|
||
tasks.append(delete_key_from_cluster(cluster_id, email, client_id))
|
||
await asyncio.gather(*tasks)
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при удалении ключей с серверов для пользователя {tg_id}: {e}")
|
||
|
||
await delete_keys_from_servers()
|
||
|
||
try:
|
||
await delete_user_data(session, tg_id)
|
||
await callback_query.message.edit_text(
|
||
text=f"🗑️ Пользователь с ID {tg_id} был удален.",
|
||
reply_markup=build_editor_kb(callback_data.tg_id)
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при удалении данных из базы данных для пользователя {tg_id}: {e}")
|
||
await callback_query.message.edit_text(
|
||
text=f"❌ Произошла ошибка при удалении пользователя с ID {tg_id}. Попробуйте снова."
|
||
)
|
||
|
||
|
||
@router.callback_query(
|
||
AdminUserEditorCallback.filter(F.action == "users_editor"),
|
||
IsAdminFilter()
|
||
)
|
||
async def handle_editor(
|
||
callback_query: types.CallbackQuery,
|
||
callback_data: AdminUserEditorCallback,
|
||
state: FSMContext,
|
||
session: Any
|
||
):
|
||
await process_user_search(
|
||
callback_query.message,
|
||
state,
|
||
session,
|
||
callback_data.tg_id,
|
||
callback_data.edit
|
||
)
|
||
|
||
|
||
async def process_user_search(
|
||
message: types.Message,
|
||
state: FSMContext,
|
||
session: Any,
|
||
tg_id: int,
|
||
edit: bool = False
|
||
) -> None:
|
||
await state.clear()
|
||
|
||
balance = await session.fetchval(
|
||
"SELECT balance FROM connections WHERE tg_id = $1", tg_id
|
||
)
|
||
|
||
if balance is None:
|
||
await message.answer(
|
||
text="🚫 Пользователь с указанным ID не найден!",
|
||
reply_markup=build_admin_back_kb(),
|
||
)
|
||
return
|
||
|
||
username = await session.fetchval(
|
||
"SELECT username FROM users WHERE tg_id = $1", tg_id
|
||
)
|
||
key_records = await session.fetch(
|
||
"SELECT email FROM keys WHERE tg_id = $1", tg_id
|
||
)
|
||
referral_count = await session.fetchval(
|
||
"SELECT COUNT(*) FROM referrals WHERE referrer_tg_id = $1", tg_id
|
||
)
|
||
|
||
text = (
|
||
f"<b>📊 Информация о пользователе</b>"
|
||
f"\n\n🆔 ID: <b>{tg_id}</b>"
|
||
f"\n📄 Логин: <b>@{username}</b>"
|
||
f"\n💰 Баланс: <b>{balance}</b>"
|
||
f"\n👥 Количество рефералов: <b>{referral_count}</b>"
|
||
)
|
||
|
||
kb = build_user_edit_kb(tg_id, key_records)
|
||
|
||
if edit:
|
||
await message.edit_text(
|
||
text=text,
|
||
reply_markup=kb
|
||
)
|
||
else:
|
||
await message.answer(
|
||
text=text,
|
||
reply_markup=kb
|
||
)
|
||
|
||
|
||
async def get_key_details(email, session):
|
||
record = await session.fetchrow(
|
||
"""
|
||
SELECT k.key, k.expiry_time, k.server_id, c.tg_id, c.balance
|
||
FROM keys k
|
||
JOIN connections c ON k.tg_id = c.tg_id
|
||
WHERE k.email = $1
|
||
""",
|
||
email,
|
||
)
|
||
|
||
if not record:
|
||
return None
|
||
|
||
servers = await get_servers_from_db()
|
||
|
||
cluster_name = "Неизвестный кластер"
|
||
for cluster_name, cluster_servers in servers.items():
|
||
if any(
|
||
server["inbound_id"] == record["server_id"] for server in cluster_servers
|
||
):
|
||
cluster_name = cluster_name
|
||
break
|
||
|
||
expiry_date = datetime.utcfromtimestamp(record["expiry_time"] / 1000)
|
||
current_date = datetime.utcnow()
|
||
time_left = expiry_date - current_date
|
||
|
||
if time_left.total_seconds() <= 0:
|
||
days_left_message = "<b>Ключ истек.</b>"
|
||
elif time_left.days > 0:
|
||
days_left_message = f"Осталось дней: <b>{time_left.days}</b>"
|
||
else:
|
||
hours_left = time_left.seconds // 3600
|
||
days_left_message = f"Осталось часов: <b>{hours_left}</b>"
|
||
|
||
return {
|
||
"key": record["key"],
|
||
"expiry_date": expiry_date.strftime("%d %B %Y года"),
|
||
"days_left_message": days_left_message,
|
||
"server_name": cluster_name,
|
||
"balance": record["balance"],
|
||
"tg_id": record["tg_id"],
|
||
}
|
||
|
||
|
||
async def get_user_balance(tg_id: int, session: Any) -> float:
|
||
try:
|
||
return await session.fetchval(
|
||
"SELECT balance FROM connections WHERE tg_id = $1", tg_id,
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при получении баланса для пользователя {tg_id}: {e}")
|
||
return -1
|
||
|
||
|
||
async def add_user_balance(tg_id: int, balance: int, session: Any) -> None:
|
||
try:
|
||
await session.execute(
|
||
"UPDATE connections SET balance = balance + $1 WHERE tg_id = $2",
|
||
balance, tg_id,
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при добавлении баланса для пользователя {tg_id}: {e}")
|
||
|
||
|
||
async def set_user_balance(tg_id: int, balance: int, session: Any) -> None:
|
||
try:
|
||
await session.execute(
|
||
"UPDATE connections SET balance = $1 WHERE tg_id = $2",
|
||
balance, tg_id,
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при установке баланса для пользователя {tg_id}: {e}")
|