diff --git a/handlers/__pycache__/keys.cpython-310.pyc b/handlers/__pycache__/keys.cpython-310.pyc
index f452fd24..cebee691 100644
Binary files a/handlers/__pycache__/keys.cpython-310.pyc and b/handlers/__pycache__/keys.cpython-310.pyc differ
diff --git a/handlers/keys.py b/handlers/keys.py
index a3281f4c..e67205ee 100644
--- a/handlers/keys.py
+++ b/handlers/keys.py
@@ -26,7 +26,7 @@ async def process_callback_view_keys(callback_query: types.CallbackQuery):
# Создаем кнопки для каждого ключа
buttons = []
for record in records:
- key_name = record['email'] # Предполагается, что email - это название ключа
+ key_name = record['email']
client_id = record['client_id']
button = types.InlineKeyboardButton(text=key_name, callback_data=f'view_key_{key_name}_{client_id}')
buttons.append([button])
@@ -100,63 +100,63 @@ async def process_callback_view_key(callback_query: types.CallbackQuery):
await callback_query.answer()
-
# Обработка запроса на удаление ключа
@router.callback_query(lambda c: c.data.startswith('delete_key_'))
async def process_callback_delete_key(callback_query: types.CallbackQuery):
tg_id = callback_query.from_user.id
- client_id = callback_query.data.split('_')[2] # Получаем client_id
+ client_id = callback_query.data.split('_')[2]
confirmation_keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
[types.InlineKeyboardButton(text='Да, удалить', callback_data=f'confirm_delete_{client_id}')],
[types.InlineKeyboardButton(text='Нет, отменить', callback_data='view_keys')]
])
- await bot.send_message(tg_id, "Вы уверены, что хотите удалить ключ?", reply_markup=confirmation_keyboard)
+ await bot.edit_message_text("Вы уверены, что хотите удалить ключ?", chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=confirmation_keyboard)
await callback_query.answer()
@router.callback_query(lambda c: c.data.startswith('confirm_delete_'))
async def process_callback_confirm_delete(callback_query: types.CallbackQuery):
tg_id = callback_query.from_user.id
- client_id = callback_query.data.split('_')[2] # Получаем client_id
+ client_id = callback_query.data.split('_')[2]
try:
conn = await asyncpg.connect(DATABASE_URL)
try:
- # Получаем email для удаления через API
record = await conn.fetchrow('SELECT email FROM keys WHERE client_id = $1', client_id)
if record:
email = record['email']
- # Создаем сессию для API-запросов
session = login_with_credentials(ADMIN_USERNAME, ADMIN_PASSWORD)
-
- # Удаляем клиента через API
success = delete_client(session, client_id)
if success:
- # Удаляем ключ из базы данных
await conn.execute('DELETE FROM keys WHERE client_id = $1', client_id)
- await bot.send_message(tg_id, "Ключ был успешно удален.")
+ response_message = "Ключ был успешно удален."
else:
- await bot.send_message(tg_id, "Ошибка при удалении клиента через API.")
+ response_message = "Ошибка при удалении клиента через API."
else:
- await bot.send_message(tg_id, "Ключ не найден или уже удален.")
+ response_message = "Ключ не найден или уже удален."
+
+ back_button = types.InlineKeyboardButton(text='Назад', callback_data='view_keys')
+ keyboard = types.InlineKeyboardMarkup(inline_keyboard=[[back_button]])
+
+ await bot.edit_message_text(response_message, chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=keyboard)
finally:
await conn.close()
except Exception as e:
- await bot.send_message(tg_id, f"Ошибка при удалении ключа: {e}")
+ await bot.edit_message_text(f"Ошибка при удалении ключа: {e}", chat_id=tg_id, message_id=callback_query.message.message_id)
await callback_query.answer()
+
@router.callback_query(lambda c: c.data.startswith('renew_key_'))
async def process_callback_renew_key(callback_query: types.CallbackQuery):
tg_id = callback_query.from_user.id
- client_id = callback_query.data.split('_')[2] # Получаем client_id
+ client_id = callback_query.data.split('_')[2]
try:
conn = await asyncpg.connect(DATABASE_URL)
@@ -169,10 +169,9 @@ async def process_callback_renew_key(callback_query: types.CallbackQuery):
current_time = datetime.utcnow().timestamp() * 1000
if expiry_time <= current_time:
- await callback_query.message.answer("Ваш ключ уже истек и не может быть продлен.")
+ await bot.edit_message_text("Ваш ключ уже истек и не может быть продлен.", chat_id=tg_id, message_id=callback_query.message.message_id)
return
- # Создаем клавиатуру для выбора плана продления
keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
[types.InlineKeyboardButton(text='Продлить на 1 месяц (100 руб.)', callback_data=f'renew_1_month_{client_id}')],
[types.InlineKeyboardButton(text='Продлить на 3 месяца (250 руб.)', callback_data=f'renew_3_months_{client_id}')],
@@ -184,31 +183,27 @@ async def process_callback_renew_key(callback_query: types.CallbackQuery):
f"Баланс: {balance} руб.\n"
f"Действующий ключ истекает {datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')}")
- await delete_previous_message(callback_query)
- await bot.send_message(tg_id, response_message, parse_mode="HTML", reply_markup=keyboard)
+ await bot.edit_message_text(response_message, chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=keyboard, parse_mode="HTML")
finally:
await conn.close()
except Exception as e:
- await callback_query.message.answer(f"Ошибка при выборе плана: {e}")
+ await bot.edit_message_text(f"Ошибка при выборе плана: {e}", chat_id=tg_id, message_id=callback_query.message.message_id)
await callback_query.answer()
-
# Обработка выбора плана продления
@router.callback_query(lambda c: c.data.startswith('renew_'))
async def process_callback_renew_plan(callback_query: types.CallbackQuery):
tg_id = callback_query.from_user.id
plan, client_id = callback_query.data.split('_')[1], callback_query.data.split('_')[3] # '1' или '3' и client_id
- days_to_extend = 30 * int(plan) # 30 дней или 90 дней
- print(plan, client_id, days_to_extend)
+ days_to_extend = 30 * int(plan)
try:
conn = await asyncpg.connect(DATABASE_URL)
try:
record = await conn.fetchrow('SELECT email, expiry_time FROM keys WHERE client_id = $1', client_id)
- print(record)
if record:
email = record['email']
@@ -216,58 +211,45 @@ async def process_callback_renew_plan(callback_query: types.CallbackQuery):
current_time = datetime.utcnow().timestamp() * 1000
if expiry_time <= current_time:
- await callback_query.message.answer("Ваш ключ уже истек и не может быть продлен.")
+ await bot.edit_message_text("Ваш ключ уже истек и не может быть продлен.", chat_id=tg_id, message_id=callback_query.message.message_id)
return
- # Рассчитываем новый срок окончания
new_expiry_time = int(expiry_time + timedelta(days=days_to_extend).total_seconds() * 1000)
-
- # Определяем стоимость продления
cost = 100 if plan == '1' else 250
- # Проверка баланса
balance = await get_balance(tg_id)
if balance < cost:
replenish_button = types.InlineKeyboardButton(text='Пополнить баланс', callback_data='replenish_balance')
back_button = types.InlineKeyboardButton(text='Назад', callback_data='view_keys')
keyboard = types.InlineKeyboardMarkup(inline_keyboard=[[replenish_button], [back_button]])
- await callback_query.message.answer("Недостаточно средств для продления ключа.", reply_markup=keyboard)
+ await bot.edit_message_text("Недостаточно средств для продления ключа.", chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=keyboard)
return
- # Создаем сессию для API-запросов
session = login_with_credentials(ADMIN_USERNAME, ADMIN_PASSWORD)
-
- # Обновляем ключ через API
success = extend_client_key(session, tg_id, client_id, email, new_expiry_time)
if success:
- await update_balance(tg_id, -cost) # Списание средств с баланса
+ await update_balance(tg_id, -cost)
await conn.execute('UPDATE keys SET expiry_time = $1 WHERE client_id = $2', new_expiry_time, client_id)
response_message = f"Ваш ключ был успешно продлен на {days_to_extend // 30} месяц(-)."
back_button = types.InlineKeyboardButton(text='Назад', callback_data='view_keys')
keyboard = types.InlineKeyboardMarkup(inline_keyboard=[[back_button]])
- await bot.send_message(tg_id, response_message, reply_markup=keyboard)
+ await bot.edit_message_text(response_message, chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=keyboard)
else:
- await bot.send_message(tg_id, "Ошибка при продлении ключа.")
+ await bot.edit_message_text("Ошибка при продлении ключа.", chat_id=tg_id, message_id=callback_query.message.message_id)
else:
- await bot.send_message(tg_id, "У вас нет ключей для продления.")
+ await bot.edit_message_text("У вас нет ключей для продления.", chat_id=tg_id, message_id=callback_query.message.message_id)
finally:
await conn.close()
except Exception as e:
- await bot.send_message(tg_id, f"Ошибка при продлении ключа: {e}")
+ await bot.edit_message_text(f"Ошибка при продлении ключа: {e}", chat_id=tg_id, message_id=callback_query.message.message_id)
await callback_query.answer()
-
-# Удаление предыдущего сообщения
-async def delete_previous_message(callback_query: types.CallbackQuery):
- if callback_query.message.message_id:
- await bot.delete_message(chat_id=callback_query.from_user.id, message_id=callback_query.message.message_id)
-
# Обработка ошибок
async def handle_error(tg_id, callback_query, message):
- await delete_previous_message(callback_query)
- await bot.send_message(tg_id, message)
+ await bot.edit_message_text(message, chat_id=tg_id, message_id=callback_query.message.message_id)
+