service.py

#!/usr/bin/env python3

import asyncio
import logging
import signal

from service import config, db
from service.league_service import LeagueService
from service.message_queue_service import MessageQueueService
from service.season_generator import SeasonGenerator


async def main():
    loop = asyncio.get_running_loop()
    done = asyncio.Future()

    def signal_handler(sig: int, _frame):
        logger.info("Received signal %s, shutting down", signal.Signals(sig))
        if not done.done():
            done.set_result(0)

    # Make sure we can shutdown gracefully
    signal.signal(signal.SIGTERM, signal_handler)
    signal.signal(signal.SIGINT, signal_handler)

    database = db.FAFDatabase(
        host=config.DB_SERVER,
        port=int(config.DB_PORT),
        user=config.DB_LOGIN,
        password=config.DB_PASSWORD,
        db=config.DB_NAME,
    )
    logger.info("Database connected.")

    mq_service = MessageQueueService()
    await mq_service.initialize()

    await mq_service.declare_exchange(config.EXCHANGE_NAME)

    league_service = LeagueService(database, mq_service)
    await league_service.initialize()

    season_generator = SeasonGenerator(database)
    season_generator.initialize()

    await done

    # Cleanup

    await league_service.shutdown()
    await mq_service.shutdown()

    logger.info("Closing database connection...")
    await database.close()
    logger.info("All done. Exiting.")


if __name__ == "__main__":
    logger = logging.getLogger()
    stderr_handler = logging.StreamHandler()
    stderr_handler.setFormatter(
        logging.Formatter(
            fmt="%(levelname)-8s %(asctime)s %(name)-30s %(message)s",
            datefmt="%b %d  %H:%M:%S",
        )
    )
    logger.addHandler(stderr_handler)
    logger.setLevel(config.LOG_LEVEL)

    asyncio.run(main())