Egor
219 lines
9.3 KiB
Python
219 lines
9.3 KiB
Python
import asyncio
|
|
import logging
|
|
import sys
|
|
import os
|
|
import signal
|
|
from pathlib import Path
|
|
|
|
sys.path.append(str(Path(__file__).parent))
|
|
|
|
from app.bot import setup_bot
|
|
from app.config import settings
|
|
from app.database.database import init_db
|
|
from app.services.monitoring_service import monitoring_service
|
|
from app.services.maintenance_service import maintenance_service
|
|
from app.services.payment_service import PaymentService
|
|
from app.external.webhook_server import WebhookServer
|
|
from app.external.yookassa_webhook import start_yookassa_webhook_server
|
|
from app.database.universal_migration import run_universal_migration
|
|
|
|
|
|
class GracefulExit:
|
|
|
|
def __init__(self):
|
|
self.exit = False
|
|
|
|
def exit_gracefully(self, signum, frame):
|
|
logging.getLogger(__name__).info(f"Получен сигнал {signum}. Корректное завершение работы...")
|
|
self.exit = True
|
|
|
|
|
|
async def main():
|
|
logging.basicConfig(
|
|
level=getattr(logging, settings.LOG_LEVEL),
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
handlers=[
|
|
logging.FileHandler(settings.LOG_FILE, encoding='utf-8'),
|
|
logging.StreamHandler(sys.stdout)
|
|
]
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
logger.info("🚀 Запуск Bedolaga Remnawave Bot...")
|
|
|
|
killer = GracefulExit()
|
|
signal.signal(signal.SIGINT, killer.exit_gracefully)
|
|
signal.signal(signal.SIGTERM, killer.exit_gracefully)
|
|
|
|
webhook_server = None
|
|
yookassa_server_task = None
|
|
monitoring_task = None
|
|
maintenance_task = None
|
|
polling_task = None
|
|
|
|
try:
|
|
logger.info("📊 Инициализация базы данных...")
|
|
await init_db()
|
|
|
|
skip_migration = os.getenv('SKIP_MIGRATION', 'false').lower() == 'true'
|
|
|
|
if not skip_migration:
|
|
logger.info("🔧 Выполняем проверку и миграцию базы данных...")
|
|
try:
|
|
migration_success = await run_universal_migration()
|
|
|
|
if migration_success:
|
|
logger.info("✅ Миграция базы данных завершена успешно")
|
|
else:
|
|
logger.warning("⚠️ Миграция завершилась с предупреждениями, но продолжаем запуск")
|
|
|
|
except Exception as migration_error:
|
|
logger.error(f"❌ Ошибка выполнения миграции: {migration_error}")
|
|
logger.warning("⚠️ Продолжаем запуск без миграции")
|
|
else:
|
|
logger.info("ℹ️ Миграция пропущена (SKIP_MIGRATION=true)")
|
|
|
|
logger.info("🤖 Настройка бота...")
|
|
bot, dp = await setup_bot()
|
|
|
|
monitoring_service.bot = bot
|
|
maintenance_service.set_bot(bot)
|
|
logger.info("🔗 Бот подключен к сервисам мониторинга и техработ")
|
|
|
|
payment_service = PaymentService(bot)
|
|
|
|
if settings.TRIBUTE_ENABLED:
|
|
logger.info("🌐 Запуск Tribute webhook сервера...")
|
|
webhook_server = WebhookServer(bot)
|
|
await webhook_server.start()
|
|
else:
|
|
logger.info("ℹ️ Tribute отключен, webhook сервер не запускается")
|
|
|
|
if settings.is_yookassa_enabled():
|
|
logger.info("💳 Запуск YooKassa webhook сервера...")
|
|
yookassa_server_task = asyncio.create_task(
|
|
start_yookassa_webhook_server(payment_service)
|
|
)
|
|
else:
|
|
logger.info("ℹ️ YooKassa отключена, webhook сервер не запускается")
|
|
|
|
logger.info("📊 Запуск службы мониторинга...")
|
|
monitoring_task = asyncio.create_task(monitoring_service.start_monitoring())
|
|
|
|
logger.info("🔧 Проверка службы техработ...")
|
|
if not maintenance_service._check_task or maintenance_service._check_task.done():
|
|
logger.info("🔧 Запуск службы техработ...")
|
|
maintenance_task = asyncio.create_task(maintenance_service.start_monitoring())
|
|
else:
|
|
logger.info("🔧 Служба техработ уже запущена")
|
|
maintenance_task = None
|
|
|
|
logger.info("🔄 Запуск polling...")
|
|
polling_task = asyncio.create_task(dp.start_polling(bot, skip_updates=True))
|
|
|
|
logger.info("=" * 50)
|
|
logger.info("🎯 Активные webhook endpoints:")
|
|
if settings.TRIBUTE_ENABLED:
|
|
logger.info(f" Tribute: {settings.WEBHOOK_URL}:{settings.TRIBUTE_WEBHOOK_PORT}{settings.TRIBUTE_WEBHOOK_PATH}")
|
|
if settings.is_yookassa_enabled():
|
|
logger.info(f" YooKassa: {settings.WEBHOOK_URL}:{settings.YOOKASSA_WEBHOOK_PORT}{settings.YOOKASSA_WEBHOOK_PATH}")
|
|
logger.info("=" * 50)
|
|
|
|
try:
|
|
while not killer.exit:
|
|
await asyncio.sleep(1)
|
|
|
|
if yookassa_server_task and yookassa_server_task.done():
|
|
exception = yookassa_server_task.exception()
|
|
if exception:
|
|
logger.error(f"YooKassa webhook сервер завершился с ошибкой: {exception}")
|
|
logger.info("🔄 Перезапуск YooKassa webhook сервера...")
|
|
yookassa_server_task = asyncio.create_task(
|
|
start_yookassa_webhook_server(payment_service)
|
|
)
|
|
|
|
if monitoring_task.done():
|
|
exception = monitoring_task.exception()
|
|
if exception:
|
|
logger.error(f"Служба мониторинга завершилась с ошибкой: {exception}")
|
|
monitoring_task = asyncio.create_task(monitoring_service.start_monitoring())
|
|
|
|
if maintenance_task and maintenance_task.done():
|
|
exception = maintenance_task.exception()
|
|
if exception:
|
|
logger.error(f"Служба техработ завершилась с ошибкой: {exception}")
|
|
maintenance_task = asyncio.create_task(maintenance_service.start_monitoring())
|
|
|
|
if polling_task.done():
|
|
exception = polling_task.exception()
|
|
if exception:
|
|
logger.error(f"Polling завершился с ошибкой: {exception}")
|
|
break
|
|
|
|
except Exception as e:
|
|
logger.error(f"Ошибка в основном цикле: {e}")
|
|
|
|
except Exception as e:
|
|
logger.error(f"❌ Критическая ошибка при запуске: {e}")
|
|
raise
|
|
|
|
finally:
|
|
logger.info("🛑 Начинается корректное завершение работы...")
|
|
|
|
if yookassa_server_task and not yookassa_server_task.done():
|
|
logger.info("ℹ️ Остановка YooKassa webhook сервера...")
|
|
yookassa_server_task.cancel()
|
|
try:
|
|
await yookassa_server_task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
|
|
if monitoring_task and not monitoring_task.done():
|
|
logger.info("ℹ️ Остановка службы мониторинга...")
|
|
monitoring_service.stop_monitoring()
|
|
monitoring_task.cancel()
|
|
try:
|
|
await monitoring_task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
|
|
if maintenance_task and not maintenance_task.done():
|
|
logger.info("ℹ️ Остановка службы техработ...")
|
|
await maintenance_service.stop_monitoring()
|
|
maintenance_task.cancel()
|
|
try:
|
|
await maintenance_task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
|
|
if polling_task and not polling_task.done():
|
|
logger.info("ℹ️ Остановка polling...")
|
|
polling_task.cancel()
|
|
try:
|
|
await polling_task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
|
|
if webhook_server:
|
|
logger.info("ℹ️ Остановка Tribute webhook сервера...")
|
|
await webhook_server.stop()
|
|
|
|
if 'bot' in locals():
|
|
try:
|
|
await bot.session.close()
|
|
logger.info("✅ Сессия бота закрыта")
|
|
except Exception as e:
|
|
logger.error(f"Ошибка закрытия сессии бота: {e}")
|
|
|
|
logger.info("✅ Завершение работы бота завершено")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
asyncio.run(main())
|
|
except KeyboardInterrupt:
|
|
print("\n🛑 Бот остановлен пользователем")
|
|
except Exception as e:
|
|
print(f"❌ Критическая ошибка: {e}")
|
|
sys.exit(1)
|