init

main.py

@@ -0,0 +1,124 @@
+"""
+Telegram News Digest — точка входа.
+
+Использование:
+    python main.py              # Собрать и отправить дайджест
+    python main.py --auth       # Авторизация (первый запуск)
+    python main.py --collect    # Только собрать (без суммаризации)
+    python main.py --dry-run    # Собрать + суммаризировать, но не отправлять
+"""
+
+import argparse
+import asyncio
+import logging
+import sys
+from pathlib import Path
+
+import yaml
+
+from collector import MessageCollector
+from summarizer import Summarizer
+from delivery import DeliveryManager
+
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s [%(levelname)s] %(message)s",
+    datefmt="%H:%M:%S",
+)
+logger = logging.getLogger(__name__)
+
+
+def load_config(path: str = "config.yaml") -> dict:
+    """Загрузить конфигурацию."""
+    config_path = Path(path)
+    if not config_path.exists():
+        logger.error(
+            f"Конфиг не найден: {path}\n"
+            f"Скопируй config.example.yaml → config.yaml и заполни."
+        )
+        sys.exit(1)
+
+    with open(config_path, encoding="utf-8") as f:
+        return yaml.safe_load(f)
+
+
+async def run_auth(config: dict):
+    """Авторизация в Telegram."""
+    logger.info("🔑 Авторизация в Telegram...")
+    collector = MessageCollector(config)
+    await collector.auth()
+    logger.info("✓ Авторизация завершена. Сессия сохранена.")
+
+
+async def run_digest(config: dict, collect_only: bool = False, dry_run: bool = False):
+    """Основной пайплайн: сбор → суммаризация → доставка."""
+
+    # 1. Сбор сообщений
+    logger.info("📥 Сбор сообщений из каналов...")
+    collector = MessageCollector(config)
+    result = await collector.collect()
+
+    if result.errors:
+        logger.warning(f"Ошибки в {len(result.errors)} каналах:")
+        for ch, err in result.errors.items():
+            logger.warning(f"  {ch}: {err}")
+
+    if not result.messages:
+        logger.info("📭 Нет новых сообщений. Выход.")
+        return
+
+    if collect_only:
+        # Вывести сырые сообщения
+        for msg in result.messages[:20]:
+            print(f"\n[{msg.channel_title}] {msg.date.strftime('%H:%M')}")
+            print(msg.text[:200] + ("..." if len(msg.text) > 200 else ""))
+        return
+
+    # 2. Суммаризация
+    logger.info("🤖 Суммаризация через LLM...")
+    summarizer = Summarizer(config)
+    digest = await summarizer.summarize(result.messages)
+
+    if dry_run:
+        print("\n" + "=" * 60)
+        print(digest)
+        print("=" * 60)
+        logger.info("(dry-run: дайджест не отправлен)")
+        return
+
+    # 3. Доставка
+    logger.info("📤 Отправка дайджеста...")
+    delivery = DeliveryManager(config)
+    await delivery.deliver(digest)
+
+    logger.info("✅ Готово!")
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Telegram News Digest")
+    parser.add_argument(
+        "--auth", action="store_true", help="Авторизация в Telegram"
+    )
+    parser.add_argument(
+        "--collect", action="store_true", help="Только сбор (без суммаризации)"
+    )
+    parser.add_argument(
+        "--dry-run", action="store_true", help="Без отправки (вывод в консоль)"
+    )
+    parser.add_argument(
+        "--config", default="config.yaml", help="Путь к конфигу"
+    )
+    args = parser.parse_args()
+
+    config = load_config(args.config)
+
+    if args.auth:
+        asyncio.run(run_auth(config))
+    else:
+        asyncio.run(
+            run_digest(config, collect_only=args.collect, dry_run=args.dry_run)
+        )
+
+
+if __name__ == "__main__":
+    main()