main.py

import asyncio
from asyncio.exceptions import CancelledError
import signal
from src.scanner.scanner_impl import ScannerImpl
import logging
import sys
import os

sys.stdout.reconfigure(line_buffering=True)

logging.basicConfig(
    level=logging.DEBUG,
    format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
    handlers=[
        logging.FileHandler(f"logs/root.log"),
        logging.StreamHandler(sys.stdout),
    ],
)


async def main():
    try:
        scanner = ScannerImpl(int(os.getenv("SCANNER_INTERVAL_MIN")))
        scanner.start()

        stop_event = asyncio.Event()

        # Обработка сигналов для корректного завершения
        loop = asyncio.get_running_loop()
        for sig in (signal.SIGINT, signal.SIGTERM):
            loop.add_signal_handler(sig, stop_event.set)

        await stop_event.wait()  # Ждём сигнала завершения
        scanner.stop()
    except CancelledError:
        logging.critical("Exited")


if __name__ == "__main__":
    asyncio.run(main())