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