houseassassin/Solo_bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #40 from Vladless/new1

Browse Source 
Автопродление ключа
This commit is contained in:
Vladislav Lisitsyn
2024-10-25 22:21:07 +03:00
committed by GitHub
parent 601024b5bd f0976a2c9c
commit ca1376228d
1 changed files with 149 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files
handlers/notifications.py
+149 -96
View File
@@ -1,5 +1,6 @@
from datetime import datetime, timedelta
import asyncpg
import asyncio
from aiogram import Bot, Router
from aiogram.fsm.state import State, StatesGroup
import logging
@@ -18,107 +19,159 @@ class NotificationStates(StatesGroup):
waiting_for_notification_text = State()
async def notify_expiring_keys(bot: Bot):
conn = None
try:
conn = await asyncpg.connect(DATABASE_URL)
logger.info("Подключение к базе данных успешно.")
try:
current_time = datetime.utcnow().timestamp() * 1000
threshold_time_10h = (datetime.utcnow() + timedelta(hours=10)).timestamp() * 1000
threshold_time_24h = (datetime.utcnow() + timedelta(days=1)).timestamp() * 1000
current_time = datetime.utcnow().timestamp() * 1000
threshold_time_10h = (datetime.utcnow() + timedelta(hours=10)).timestamp() * 1000
threshold_time_24h = (datetime.utcnow() + timedelta(days=1)).timestamp() * 1000
logger.info("Начало обработки уведомлений.")
logger.info("Начало обработки уведомлений.")
# Уведомления за 10 часов до истечения
records = await conn.fetch('''
SELECT tg_id, email, expiry_time, client_id, server_id FROM keys
WHERE expiry_time <= $1 AND expiry_time > $2 AND notified = FALSE
''', threshold_time_10h, current_time)
await notify_10h_keys(bot, conn, current_time, threshold_time_10h)
await asyncio.sleep(1) # Задержка между уведомлениями за 10 часов и 24 часа
await notify_24h_keys(bot, conn, current_time, threshold_time_24h)
await asyncio.sleep(1) # Задержка перед обработкой истекших ключей
await handle_expired_keys(bot, conn, current_time)
logger.info(f"Найдено {len(records)} ключей для уведомления за 10 часов.")
for record in records:
tg_id = record['tg_id']
email = record['email']
expiry_time = record['expiry_time']
server_id = record['server_id']
expiry_date = datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')
message = KEY_EXPIRY_10H.format(server_id=server_id, email=email, expiry_date=expiry_date)
await bot.send_message(tg_id, message)
logger.info(f"Уведомление отправлено пользователю {tg_id}.")
await conn.execute('UPDATE keys SET notified = TRUE WHERE client_id = $1', record['client_id'])
logger.info(f"Обновлено поле notified для клиента {record['client_id']}.")
# Уведомления за 24 часа до истечения
records_24h = await conn.fetch('''
SELECT tg_id, email, expiry_time, client_id, server_id FROM keys
WHERE expiry_time <= $1 AND expiry_time > $2 AND notified_24h = FALSE
''', threshold_time_24h, current_time)
logger.info(f"Найдено {len(records_24h)} ключей для уведомления за 24 часа.")
for record in records_24h:
tg_id = record['tg_id']
email = record['email']
expiry_time = record['expiry_time']
server_id = record['server_id']
time_left = (expiry_time / 1000) - datetime.utcnow().timestamp()
hours_left = max(0, int(time_left // 3600))
expiry_date = datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')
balance = await get_balance(tg_id)
message_24h = KEY_EXPIRY_24H.format(server_id=server_id, email=email, hours_left=hours_left, expiry_date=expiry_date, balance=balance)
await bot.send_message(tg_id, message_24h)
logger.info(f"Уведомление за 24 часа отправлено пользователю {tg_id}.")
await conn.execute('UPDATE keys SET notified_24h = TRUE WHERE client_id = $1', record['client_id'])
logger.info(f"Обновлено поле notified_24h для клиента {record['client_id']}.")
# Обработка ключей по истечению
expiring_keys = await conn.fetch('''
SELECT tg_id, client_id, expiry_time, server_id FROM keys
WHERE expiry_time <= $1
''', current_time)
logger.info(f"Найдено {len(expiring_keys)} истекающих ключей.")
for record in expiring_keys:
tg_id = record['tg_id']
client_id = record['client_id']
balance = await get_balance(tg_id)
server_id = record['server_id']
logger.info(f"Проверка баланса для клиента {tg_id}: {balance}.")
if balance >= 100:
new_expiry_time = int((datetime.utcnow() + timedelta(days=30)).timestamp() * 1000)
await update_key_expiry(client_id, new_expiry_time)
logger.info(f"Ключ для клиента {tg_id} продлен до {datetime.utcfromtimestamp(new_expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')}.")
session = await login_with_credentials(server_id, ADMIN_USERNAME, ADMIN_PASSWORD)
success = await extend_client_key(session, server_id, tg_id, client_id, email, new_expiry_time)
if success:
await bot.send_message(tg_id, KEY_RENEWED)
logger.info(f"Ключ для пользователя {tg_id} успешно продлен на месяц.")
else:
await bot.send_message(tg_id, KEY_RENEWAL_FAILED)
logger.error(f"Не удалось продлить ключ для пользователя {tg_id}.")
else:
await delete_key(client_id)
logger.info(f"Ключ для клиента {tg_id} удален из-за недостаточного баланса.")
session = await login_with_credentials(server_id, ADMIN_USERNAME, ADMIN_PASSWORD)
success = await delete_client(session, server_id, client_id)
if success:
await bot.send_message(tg_id, KEY_DELETED)
logger.info(f"Ключ для пользователя {tg_id} удален.")
else:
await bot.send_message(tg_id, KEY_DELETION_FAILED)
logger.error(f"Не удалось удалить ключ для пользователя {tg_id}.")
finally:
await conn.close()
logger.info("Соединение с базой данных закрыто.")
except Exception as e:
logger.error(f"Ошибка при отправке уведомлений: {e}")
finally:
if conn:
await conn.close()
logger.info("Соединение с базой данных закрыто.")
async def is_bot_blocked(bot: Bot, chat_id: int) -> bool:
try:
member = await bot.get_chat_member(chat_id, bot.id)
return member.status == 'left' # Если бот не в чате, он заблокирован
except Exception as e:
logger.error(f"Ошибка при проверке статуса бота у пользователя {chat_id}: {e}")
return False # Если произошла ошибка, предполагаем, что бот не заблокирован
async def notify_10h_keys(bot: Bot, conn: asyncpg.Connection, current_time: float, threshold_time_10h: float):
records = await conn.fetch('''
SELECT tg_id, email, expiry_time, client_id, server_id FROM keys
WHERE expiry_time <= $1 AND expiry_time > $2 AND notified = FALSE
''', threshold_time_10h, current_time)
logger.info(f"Найдено {len(records)} ключей для уведомления за 10 часов.")
for record in records:
tg_id = record['tg_id']
email = record['email']
expiry_time = record['expiry_time']
server_id = record['server_id']
expiry_date = datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')
message = KEY_EXPIRY_10H.format(server_id=server_id, email=email, expiry_date=expiry_date)
if not await is_bot_blocked(bot, tg_id):
try:
await bot.send_message(tg_id, message)
logger.info(f"Уведомление отправлено пользователю {tg_id}.")
except Exception as e:
logger.error(f"Ошибка при отправке уведомления пользователю {tg_id}: {e}")
continue # Пропустить этого пользователя и продолжить
await conn.execute('UPDATE keys SET notified = TRUE WHERE client_id = $1', record['client_id'])
logger.info(f"Обновлено поле notified для клиента {record['client_id']}.")
await asyncio.sleep(1) # Задержка между уведомлениями
async def notify_24h_keys(bot: Bot, conn: asyncpg.Connection, current_time: float, threshold_time_24h: float):
logger.info("Проверка истекших ключей...")
records_24h = await conn.fetch('''
SELECT tg_id, email, expiry_time, client_id, server_id FROM keys
WHERE expiry_time <= $1 AND expiry_time > $2 AND notified_24h = FALSE
''', threshold_time_24h, current_time)
logger.info(f"Найдено {len(records_24h)} ключей для уведомления за 24 часа.")
for record in records_24h:
tg_id = record['tg_id']
email = record['email']
expiry_time = record['expiry_time']
server_id = record['server_id']
time_left = (expiry_time / 1000) - datetime.utcnow().timestamp()
hours_left = max(0, int(time_left // 3600))
expiry_date = datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')
balance = await get_balance(tg_id)
message_24h = KEY_EXPIRY_24H.format(server_id=server_id, email=email, hours_left=hours_left, expiry_date=expiry_date, balance=balance)
if not await is_bot_blocked(bot, tg_id):
try:
await bot.send_message(tg_id, message_24h)
logger.info(f"Уведомление за 24 часа отправлено пользователю {tg_id}.")
except Exception as e:
logger.error(f"Ошибка при отправке уведомления за 24 часа пользователю {tg_id}: {e}")
continue # Пропустить этого пользователя и продолжить
await conn.execute('UPDATE keys SET notified_24h = TRUE WHERE client_id = $1', record['client_id'])
logger.info(f"Обновлено поле notified_24h для клиента {record['client_id']}.")
await asyncio.sleep(1) # Задержка между уведомлениями
async def handle_expired_keys(bot: Bot, conn: asyncpg.Connection, current_time: float):
logger.info("Проверка истекших ключей...")
current_time = int(current_time) # Убедитесь, что время в миллисекундах
expiring_keys = await conn.fetch('''
SELECT tg_id, client_id, expiry_time, server_id, email FROM keys
WHERE expiry_time <= $1
''', current_time)
logger.info(f"Найдено {len(expiring_keys)} истекающих ключей.")
for record in expiring_keys:
tg_id = record['tg_id']
client_id = record['client_id']
balance = await get_balance(tg_id)
server_id = record['server_id']
email = record['email']
logger.info(f"Проверка баланса для клиента {tg_id}: {balance}.")
if balance >= 100:
new_expiry_time = int((datetime.utcnow() + timedelta(days=30)).timestamp() * 1000)
await update_key_expiry(client_id, new_expiry_time)
logger.info(f"Ключ для клиента {tg_id} продлен до {datetime.utcfromtimestamp(new_expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')}.")
session = await login_with_credentials(server_id, ADMIN_USERNAME, ADMIN_PASSWORD)
success = await extend_client_key(session, server_id, tg_id, client_id, email, new_expiry_time)
if success:
try:
await bot.send_message(tg_id, KEY_RENEWED)
logger.info(f"Ключ для пользователя {tg_id} успешно продлен на месяц.")
except Exception as e:
logger.error(f"Ошибка при отправке уведомления о продлении ключа пользователю {tg_id}: {e}")
else:
try:
await bot.send_message(tg_id, KEY_RENEWAL_FAILED)
logger.error(f"Не удалось продлить ключ для пользователя {tg_id}.")
except Exception as e:
logger.error(f"Ошибка при отправке уведомления о неудачном продлении ключа пользователю {tg_id}: {e}")
else:
await delete_key(client_id)
logger.info(f"Ключ для клиента {tg_id} удален из-за недостаточного баланса.")
session = await login_with_credentials(server_id, ADMIN_USERNAME, ADMIN_PASSWORD)
success = await delete_client(session, server_id, client_id)
if success:
try:
await bot.send_message(tg_id, KEY_DELETED)
logger.info(f"Ключ для пользователя {tg_id} удален.")
except Exception as e:
logger.error(f"Ошибка при отправке уведомления об удалении ключа пользователю {tg_id}: {e}")
else:
try:
await bot.send_message(tg_id, KEY_DELETION_FAILED)
logger.error(f"Не удалось удалить ключ для пользователя {tg_id}.")
except Exception as e:
logger.error(f"Ошибка при отправке уведомления о неудачном удалении ключа пользователю {tg_id}: {e}")
await asyncio.sleep(1) # Задержка между обработками ключей