diff --git a/handlers/__pycache__/keys.cpython-310.pyc b/handlers/__pycache__/keys.cpython-310.pyc index 9a69ce57..028d3eb0 100644 Binary files a/handlers/__pycache__/keys.cpython-310.pyc and b/handlers/__pycache__/keys.cpython-310.pyc differ diff --git a/handlers/__pycache__/notifications.cpython-310.pyc b/handlers/__pycache__/notifications.cpython-310.pyc new file mode 100644 index 00000000..80b1e136 Binary files /dev/null and b/handlers/__pycache__/notifications.cpython-310.pyc differ diff --git a/handlers/__pycache__/pay.cpython-310.pyc b/handlers/__pycache__/pay.cpython-310.pyc index 70f5d749..65cf0116 100644 Binary files a/handlers/__pycache__/pay.cpython-310.pyc and b/handlers/__pycache__/pay.cpython-310.pyc differ diff --git a/handlers/__pycache__/start.cpython-310.pyc b/handlers/__pycache__/start.cpython-310.pyc index c844f7d6..43480704 100644 Binary files a/handlers/__pycache__/start.cpython-310.pyc and b/handlers/__pycache__/start.cpython-310.pyc differ diff --git a/handlers/keys.py b/handlers/keys.py index 424da168..064d3517 100644 --- a/handlers/keys.py +++ b/handlers/keys.py @@ -29,7 +29,8 @@ async def process_callback_view_keys(callback_query: types.CallbackQuery): for record in records: key_name = record['email'] client_id = record['client_id'] - button = types.InlineKeyboardButton(text=key_name, callback_data=f'view_key_{key_name}_{client_id}') + # Заменяем подчеркивание на вертикальную черту в callback_data + button = types.InlineKeyboardButton(text=key_name, callback_data=f'view_key|{key_name}|{client_id}') buttons.append([button]) # Создаем клавиатуру с кнопками @@ -58,10 +59,11 @@ async def process_callback_view_keys(callback_query: types.CallbackQuery): await callback_query.answer() -@router.callback_query(lambda c: c.data.startswith('view_key_')) +@router.callback_query(lambda c: c.data.startswith('view_key|')) async def process_callback_view_key(callback_query: types.CallbackQuery): tg_id = callback_query.from_user.id - key_name, client_id = callback_query.data.split('_')[2], callback_query.data.split('_')[3] + # Разделяем данные по вертикальной черте + key_name, client_id = callback_query.data.split('|')[1], callback_query.data.split('|')[2] try: conn = await asyncpg.connect(DATABASE_URL) @@ -85,9 +87,9 @@ async def process_callback_view_key(callback_query: types.CallbackQuery): f"{days_left_message}") # Кнопки для продления, инструкций и удаления - renew_button = types.InlineKeyboardButton(text='Продлить ключ', callback_data=f'renew_key_{client_id}') + renew_button = types.InlineKeyboardButton(text='Продлить ключ', callback_data=f'renew_key|{client_id}') instructions_button = types.InlineKeyboardButton(text='Инструкции по использованию', callback_data='instructions') - delete_button = types.InlineKeyboardButton(text='Удалить ключ', callback_data=f'delete_key_{client_id}') + delete_button = types.InlineKeyboardButton(text='Удалить ключ', callback_data=f'delete_key|{client_id}') back_button = types.InlineKeyboardButton(text='Назад в профиль', callback_data='view_profile') keyboard = types.InlineKeyboardMarkup(inline_keyboard=[[renew_button], [instructions_button], [delete_button], [back_button]]) @@ -104,23 +106,62 @@ async def process_callback_view_key(callback_query: types.CallbackQuery): await callback_query.answer() # Обработка запроса на удаление ключа -@router.callback_query(lambda c: c.data.startswith('delete_key_')) +@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 = callback_query.data.split('|')[1] # Используем разделитель вертикальная черта confirmation_keyboard = types.InlineKeyboardMarkup(inline_keyboard=[ - [types.InlineKeyboardButton(text='Да, удалить', callback_data=f'confirm_delete_{client_id}')], + [types.InlineKeyboardButton(text='Да, удалить', callback_data=f'confirm_delete|{client_id}')], [types.InlineKeyboardButton(text='Нет, отменить', callback_data='view_keys')] ]) 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_')) + +# Подтверждение удаления +@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('|')[1] # Используем разделитель вертикальная черта + + try: + conn = await asyncpg.connect(DATABASE_URL) + try: + record = await conn.fetchrow('SELECT email, expiry_time FROM keys WHERE client_id = $1', client_id) + + if record: + email = record['email'] + expiry_time = record['expiry_time'] + current_time = datetime.utcnow().timestamp() * 1000 # Получаем текущее время в миллисекундах + + # Убираем проверку на истекший ключ + keyboard = types.InlineKeyboardMarkup(inline_keyboard=[ + [types.InlineKeyboardButton(text='Продлить на 1 месяц (100 руб.)', callback_data=f'renew_plan|1|{client_id}')], + [types.InlineKeyboardButton(text='Продлить на 3 месяца (250 руб.)', callback_data=f'renew_plan|3|{client_id}')], + [types.InlineKeyboardButton(text='Назад', callback_data='view_profile')] + ]) + + balance = await get_balance(tg_id) + response_message = (f"Выберите план продления:\n" + f"Баланс: {balance} руб.\n" + f"Текущая дата истечения ключа: {datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')}") + + 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 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('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 = callback_query.data.split('|')[1] # Используем разделитель вертикальная черта try: conn = await asyncpg.connect(DATABASE_URL) @@ -155,52 +196,11 @@ async def process_callback_confirm_delete(callback_query: types.CallbackQuery): 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] - - try: - conn = await asyncpg.connect(DATABASE_URL) - try: - record = await conn.fetchrow('SELECT email, expiry_time FROM keys WHERE client_id = $1', client_id) - - if record: - email = record['email'] - expiry_time = record['expiry_time'] - current_time = datetime.utcnow().timestamp() * 1000 - - if expiry_time <= current_time: - 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}')], - [types.InlineKeyboardButton(text='Назад', callback_data='view_profile')] - ]) - - balance = await get_balance(tg_id) - response_message = (f"Выберите план продления:\n" - f"Баланс: {balance} руб.\n" - f"Действующий ключ истекает {datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')}") - - 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 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_')) +@router.callback_query(lambda c: c.data.startswith('renew_plan|')) 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 + plan, client_id = callback_query.data.split('|')[1], callback_query.data.split('|')[2] # '1' или '3' и client_id days_to_extend = 30 * int(plan) try: @@ -211,38 +211,42 @@ async def process_callback_renew_plan(callback_query: types.CallbackQuery): if record: email = record['email'] expiry_time = record['expiry_time'] - current_time = datetime.utcnow().timestamp() * 1000 + current_time = datetime.utcnow().timestamp() * 1000 # Текущее время в миллисекундах if expiry_time <= current_time: + # Если ключ уже истек, он не может быть продлен 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') + back_button = types.InlineKeyboardButton(text='Назад', callback_data='view_profile') keyboard = types.InlineKeyboardMarkup(inline_keyboard=[[replenish_button], [back_button]]) 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) success = extend_client_key(session, tg_id, client_id, email, new_expiry_time) if success: + # Обновляем баланс и ключ в базе данных 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') + response_message = f"Ваш ключ был успешно продлен на {days_to_extend // 30} месяц(-а)." + back_button = types.InlineKeyboardButton(text='Назад', callback_data='view_profile') 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) else: await bot.edit_message_text("Ошибка при продлении ключа.", chat_id=tg_id, message_id=callback_query.message.message_id) else: - await bot.edit_message_text("У вас нет ключей для продления.", chat_id=tg_id, message_id=callback_query.message.message_id) + await bot.edit_message_text("Ключ не найден.", chat_id=tg_id, message_id=callback_query.message.message_id) finally: await conn.close() @@ -252,7 +256,5 @@ async def process_callback_renew_plan(callback_query: types.CallbackQuery): await callback_query.answer() -# Обработка ошибок async def handle_error(tg_id, callback_query, message): - await bot.edit_message_text(message, chat_id=tg_id, message_id=callback_query.message.message_id) - + await bot.edit_message_text(message, chat_id=tg_id, message_id=callback_query.message.message_id) \ No newline at end of file diff --git a/handlers/notifications.py b/handlers/notifications.py new file mode 100644 index 00000000..42c57a76 --- /dev/null +++ b/handlers/notifications.py @@ -0,0 +1,41 @@ +import asyncpg +from datetime import datetime, timedelta +from aiogram import Bot + +from config import DATABASE_URL + +async def notify_expiring_keys(bot: Bot): + try: + conn = await asyncpg.connect(DATABASE_URL) + try: + # Получаем все ключи, которые истекают в течение следующих 10 часов + threshold_time = (datetime.utcnow() + timedelta(hours=10)).timestamp() * 1000 # В миллисекундах + records = await conn.fetch(''' + SELECT tg_id, email, expiry_time FROM keys + WHERE expiry_time <= $1 AND expiry_time > $2 + ''', threshold_time, datetime.utcnow().timestamp() * 1000) + + for record in records: + tg_id = record['tg_id'] + email = record['email'] + expiry_time = record['expiry_time'] + expiry_date = datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S') + message = f"Ваш ключ {email} истечет {expiry_date}. Пожалуйста, продлите его." + await bot.send_message(chat_id=tg_id, text=message, parse_mode='HTML') + + # Получаем все истекшие ключи + expired_records = await conn.fetch(''' + SELECT tg_id, email FROM keys + WHERE expiry_time <= $1 + ''', datetime.utcnow().timestamp() * 1000) + + for record in expired_records: + tg_id = record['tg_id'] + email = record['email'] + message = f"Ваш ключ {email} уже истек. Пожалуйста, продлите его." + await bot.send_message(chat_id=tg_id, text=message, parse_mode='HTML') + + finally: + await conn.close() + except Exception as e: + print(f"Ошибка при отправке уведомлений: {e}") diff --git a/handlers/pay.py b/handlers/pay.py index 5c31cdb1..1828e21f 100644 --- a/handlers/pay.py +++ b/handlers/pay.py @@ -217,10 +217,9 @@ async def process_admin_confirmation(callback_query: types.CallbackQuery, state: try: await bot.delete_message(chat_id=user_id, message_id=requisites_message_id) except Exception as e: - print(f"Ошибка при удалении сообщения с реквизитами: {e}") + print(f"Ошибка при удалении сообщения: {e}") finally: - # Закрываем соединение с базой данных await conn.close() elif action == 'decline': @@ -231,14 +230,6 @@ async def process_admin_confirmation(callback_query: types.CallbackQuery, state: await send_message_with_deletion(callback_query.from_user.id, f"Ошибка при пополнении баланса: {e}", state=state, message_key='admin_error_message_id') print(f"Ошибка при пополнении баланса: {e}") - await state.clear() - await callback_query.answer() - - -@router.message(lambda m: m.text and m.text.startswith('Недостаточно средств для продления')) -async def handle_insufficient_funds(message: types.Message): - user_id = message.from_user.id - replenish_keyboard = InlineKeyboardMarkup(inline_keyboard=[ - [InlineKeyboardButton(text='Пополнить баланс', callback_data='replenish_balance')] - ]) - await send_message_with_deletion(user_id, "Ваш баланс недостаточен для продления ключа. Пожалуйста, пополните баланс.", reply_markup=replenish_keyboard, state=None, message_key='insufficient_funds_message_id') + finally: + await state.clear() + await callback_query.answer() diff --git a/handlers/pic.jpg b/handlers/pic.jpg new file mode 100644 index 00000000..cced6e03 Binary files /dev/null and b/handlers/pic.jpg differ diff --git a/handlers/start.py b/handlers/start.py index 6aa6d43c..72e174ed 100644 --- a/handlers/start.py +++ b/handlers/start.py @@ -3,7 +3,8 @@ from aiogram.filters import Command from aiogram.fsm.context import FSMContext from aiogram.fsm.state import State, StatesGroup from aiogram.types import (CallbackQuery, InlineKeyboardButton, - InlineKeyboardMarkup, Message) + InlineKeyboardMarkup, Message, + ReplyKeyboardMarkup, KeyboardButton) from aiogram.types import BufferedInputFile import os @@ -18,20 +19,20 @@ class FeedbackState(StatesGroup): async def send_welcome_message(chat_id: int): # Новый текст приветствия welcome_text = ( - "*SoloNet -- ваш провайдер в бесцензурный интернет!*\n\n" + "*SoloNet — ваш провайдер в бесцензурный интернет!*\n\n" "Получите высокую скорость и самый безопасный протокол VPN, " "который работает даже в Китае." ) # Путь к изображению - image_path = os.path.join(os.path.dirname(__file__), 'solo_pic.png') + image_path = os.path.join(os.path.dirname(__file__), 'pic.jpg') # Проверка существования файла if not os.path.isfile(image_path): await bot.send_message(chat_id, "Файл изображения не найден.") return - # Создаем клавиатуру + # Создаем inline-клавиатуру inline_keyboard = InlineKeyboardMarkup(inline_keyboard=[ [InlineKeyboardButton(text='👤 Мой профиль', callback_data='view_profile')], [InlineKeyboardButton(text='🔒 О VPN', callback_data='about_vpn')], @@ -39,19 +40,31 @@ async def send_welcome_message(chat_id: int): [InlineKeyboardButton(text='📢 Наш канал', url=CHANNEL_URL)] ]) - # Используем BufferedInputFile для отправки изображения + # Создаем Reply-клавиатуру снизу + reply_keyboard = ReplyKeyboardMarkup( + keyboard=[ + [KeyboardButton(text='Меню')] # Кнопка "Меню" + ], + resize_keyboard=True # Автоматическое изменение размера + ) + + # Отправляем изображение с inline-клавиатурой with open(image_path, 'rb') as image_from_buffer: - await bot.send_photo( + message = await bot.send_photo( chat_id, - BufferedInputFile( - image_from_buffer.read(), - filename="solo_pic.png" - ), + BufferedInputFile(image_from_buffer.read(), filename="solo_pic.png"), caption=welcome_text, parse_mode='Markdown', - reply_markup=inline_keyboard + reply_markup=inline_keyboard # Inline-клавиатура ) + # Обновляем сообщение, добавляя к нему Reply-клавиатуру (без создания нового сообщения) + await bot.send_message( + chat_id, + "Нажмите кнопку ниже для доступа к меню:", + reply_markup=reply_keyboard # Reply-клавиатура + ) + @router.message(Command('start')) async def start_command(message: Message): await send_welcome_message(message.chat.id) @@ -125,3 +138,4 @@ async def receive_feedback(message: types.Message, state: FSMContext): print(f"Ошибка при отправке обратной связи: {e}") # Логирование ошибок await state.clear() + diff --git a/key_management.py b/key_management.py index 44feb0bf..0417b7de 100644 --- a/key_management.py +++ b/key_management.py @@ -244,5 +244,4 @@ async def renew_expired_keys(): ]) await bot.send_message(tg_id, "Ваш баланс недостаточен для продления ключа. Пожалуйста, пополните баланс.", reply_markup=replenish_keyboard) - await asyncio.sleep(3600) - + await asyncio.sleep(3600) \ No newline at end of file