124 lines
3.7 KiB
Python
124 lines
3.7 KiB
Python
"""SPA Telegram Bot — main entry point."""
|
|
import asyncio
|
|
import signal
|
|
import sys
|
|
|
|
from aiogram import Bot, Dispatcher
|
|
|
|
from app.config import BOT_TOKEN, USE_POSTGRES
|
|
from app.logger import logger
|
|
|
|
# Import routers
|
|
from app.handlers.start import router as start_router
|
|
from app.handlers.booking import router as booking_router
|
|
from app.handlers.profile import router as profile_router
|
|
from app.handlers.mybookings import router as mybookings_router
|
|
from app.handlers.services import router as services_router
|
|
from app.handlers.about import router as about_router
|
|
from app.handlers.contacts import router as contacts_router
|
|
from app.handlers.admin import router as admin_router
|
|
|
|
bot = Bot(token=BOT_TOKEN)
|
|
dp = Dispatcher()
|
|
|
|
# Register routers
|
|
dp.include_routers(
|
|
start_router,
|
|
booking_router,
|
|
profile_router,
|
|
mybookings_router,
|
|
services_router,
|
|
about_router,
|
|
contacts_router,
|
|
admin_router,
|
|
)
|
|
|
|
_health_runner = None
|
|
|
|
|
|
async def on_startup():
|
|
"""Called when bot starts."""
|
|
# Initialize Postgres pool if available
|
|
if USE_POSTGRES:
|
|
try:
|
|
from app.db.postgres import PostgresDB
|
|
await PostgresDB.init_pool()
|
|
logger.info("PostgreSQL pool initialized")
|
|
except Exception as e:
|
|
logger.warning("PostgreSQL init failed, falling back to JSON: {error}", error=e)
|
|
|
|
me = await bot.get_me()
|
|
logger.info(
|
|
"Bot started | @{username} (id={id}) | Token: {token_preview}...",
|
|
username=me.username,
|
|
id=me.id,
|
|
token_preview=BOT_TOKEN[:10],
|
|
)
|
|
# Start healthcheck server
|
|
try:
|
|
from app.health import start_health_server
|
|
global _health_runner
|
|
_health_runner = await start_health_server(port=8080)
|
|
except Exception as e:
|
|
logger.warning("Healthcheck server not started: {error}", error=e)
|
|
|
|
|
|
async def on_shutdown():
|
|
"""Called when bot shuts down."""
|
|
logger.info("Bot shutting down...")
|
|
if USE_POSTGRES:
|
|
try:
|
|
from app.db.postgres import PostgresDB
|
|
await PostgresDB.close()
|
|
logger.info("PostgreSQL pool closed")
|
|
except Exception as e:
|
|
logger.warning("PostgreSQL pool close error: {error}", error=e)
|
|
if _health_runner:
|
|
await _health_runner.cleanup()
|
|
logger.info("Healthcheck server stopped")
|
|
pending = asyncio.all_tasks()
|
|
if pending:
|
|
logger.debug("Cancelling {count} pending tasks", count=len(pending))
|
|
logger.info("Bot shutdown complete")
|
|
|
|
|
|
async def main():
|
|
logger.info("Initializing bot...")
|
|
|
|
dp.startup.register(on_startup)
|
|
dp.shutdown.register(on_shutdown)
|
|
|
|
# Handle SIGTERM/SIGINT gracefully
|
|
stop_event = asyncio.Event()
|
|
|
|
def _signal_handler(sig, frame):
|
|
sig_name = signal.Signals(sig).name
|
|
logger.warning("Received {sig} signal, shutting down...", sig=sig_name)
|
|
stop_event.set()
|
|
|
|
loop = asyncio.get_event_loop()
|
|
for sig in (signal.SIGTERM, signal.SIGINT):
|
|
try:
|
|
loop.add_signal_handler(sig, lambda s=sig: _signal_handler(s, None))
|
|
except NotImplementedError:
|
|
signal.signal(sig, _signal_handler)
|
|
|
|
try:
|
|
await dp.start_polling(bot, stop_signal=stop_event)
|
|
except asyncio.CancelledError:
|
|
logger.info("Polling cancelled")
|
|
except Exception as e:
|
|
logger.exception("Fatal error in polling: {error}", error=e)
|
|
finally:
|
|
await bot.session.close()
|
|
logger.info("Bot session closed")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
asyncio.run(main())
|
|
except KeyboardInterrupt:
|
|
logger.info("Bot stopped by user (Ctrl+C)")
|
|
except Exception as e:
|
|
logger.exception("Unhandled exception: {error}", error=e)
|