main.py

"""
Модуль для запуска бота.
"""

# pylint: skip-file
import asyncio
import logging
import signal
import sys

from bot_init import bot
from commands import (
    echo_command,
    gpt_command,
    help_command,
    ping_command,
    shutdown_command,
    status_command,
    uptime_command,
    user_role_command,
)
from commands.post_admin import (
    restart_command,
    update_command,
    admin_info_command,
    bunker_command,
    admin_presets_command,
    game_rules_command
)
from commands.db_ss import (
    multi_akk_db_command,
)
from commands.adt_team import (
    add_role_command,
    add_vacation_command,
    end_vacation_command,
    extend_vacation_command,
    new_team_command,
    remove_role_command,
    remove_team_command,
    team_help_command,
    tweak_team_command,
)
from commands.dbCommand import info_command, status_command, help_command
from commands.github import (
    achang_command,
    check_workflows,
    forks_command,
    github_processor,
    milestones_command,
    pr_changelog_send,
    review_command,
)
from commands.misc.shutdows_deff import (
    shutdown_def,
)  # Для выполнения завершающих операций
from config import DISCORD_KEY
from events import on_command, on_error, on_message, on_ready, update_status

# Настройка логирования
logging.basicConfig(
    level=logging.INFO,  # Уровень логирования
    format="%(asctime)s - %(levelname)s - %(message)s",  # Формат сообщений
    handlers=[
        logging.FileHandler("bot_logs.log"),  # Логирование в файл
        logging.StreamHandler(sys.stdout),  # Логирование в консоль
    ],
)


async def shutdown(signal_type=None):
    """
    Завершаем работу бота, выполняя все необходимые завершающие операции.

    :param signal_type: Тип сигнала, если он передан
    """
    logging.info(
        "Получен сигнал: %s. Завершаем работу...",
        signal_type.name if signal_type else "Manual Exit",
    )
    try:
        await shutdown_def()  # Выполнение завершающих операций
        await bot.close()  # Закрытие соединения бота
    except Exception as e:
        logging.error("Ошибка при завершении работы бота: %s", e)
    finally:
        sys.exit(0)  # Принудительное завершение процесса


def setup_signal_handlers():
    """
    Настроить обработчики сигналов для корректного завершения работы бота.
    """
    if sys.platform == "win32":
        logging.warning(
            "Сигналы SIGINT и SIGTERM не поддерживаются на Windows."
        )
    else:
        loop = asyncio.get_event_loop()
        for sig in (signal.SIGINT, signal.SIGTERM):
            loop.add_signal_handler(
                sig, lambda s=sig: asyncio.create_task(shutdown(s))
            )


async def main():
    """
    Основная функция для запуска бота.
    """
    try:
        setup_signal_handlers()  # Настройка обработчиков сигналов
        logging.info("Запуск бота...")
        await bot.start(DISCORD_KEY)  # Запуск бота с токеном
    except (KeyboardInterrupt, SystemExit):
        await shutdown()  # Обработка принудительного завершения
    except Exception as e:
        logging.error("Неожиданная ошибка при запуске бота: %s", e)
        await shutdown()


if __name__ == "__main__":
    bot.run(DISCORD_KEY)  # Запуск бота, без asyncio.run()