main.py

from summarizer import Summarizer
from tg_bot import TgBot
import os
from argparse import ArgumentParser
from pathlib import Path
from ruamel.yaml import YAML


# Читаем из аргументов путь к конфигу
arg_parser = ArgumentParser(
    description="Справка"
)
arg_parser.add_argument(
    "-c",
    "--config",
    default=Path(__file__).resolve().parent / "config.yaml",
    help="Путь к конфигурационному YAML-файлу"
)
args = arg_parser.parse_args()

# Читаем конфиг
yaml = YAML()
yaml.preserve_quotes = True
with open(args.config, "r", encoding="utf-8") as file:
    config = yaml.load(file)

# Создаём необходимые блоки
summarizer = Summarizer(config)
tg_bot = TgBot(
    str(config["telegram_bot"]["token"]),
    int(config["telegram_bot"]["recepient"])
)
summarizer.add_consumer(tg_bot.print)

# Формируем сводки
summarizer.summarize()

# Сохраняем конфиг (поменялись метки в чатах)
with open(args.config, "w", encoding="utf-8") as file:
    yaml.dump(config, file)