miltu-bot refactor

bot/services.py

@@ -0,0 +1,52 @@
+# bot/services.py
+import logging
+from dataclasses import dataclass
+from typing import Optional, Protocol, Tuple, Coroutine, Any, Awaitable
+from django.utils import timezone
+from asgiref.sync import sync_to_async
+from .models import TelegramBot, TelegramChat
+
+logger = logging.getLogger(__name__)
+
+# Протокол (интерфейс) репозитория — удобно мокать в тестах
+from typing import Coroutine
+
+class IChatRepository(Protocol):
+    async def upsert_chat(
+        self, *, bot: TelegramBot, chat_id: int, chat_type: str,
+        title: str = "", username: str = "", is_member: bool = True,
+        touch_last_message: bool = False,
+    ) -> Coroutine[Any, Any, Tuple[TelegramChat, bool]]: ...
+
+# Реализация на Django ORM
+class DjangoChatRepository:
+    @sync_to_async
+    def upsert_chat(
+        self, *, bot: TelegramBot, chat_id: int, chat_type: str,
+        title: str = "", username: str = "", is_member: bool = True,
+        touch_last_message: bool = False,
+    ):
+        defaults = {
+            "type": chat_type,
+            "title": title or "",
+            "username": username or "",
+            "is_member": is_member,
+            "joined_at": timezone.now() if is_member else None,
+            "left_at": None if is_member else timezone.now(),
+        }
+        if touch_last_message:
+            defaults["last_message_at"] = timezone.now()
+
+        obj, created = TelegramChat.objects.update_or_create(
+            bot=bot, chat_id=chat_id, defaults=defaults
+        )
+        return obj, created
+
+@dataclass
+class BotServices:
+    """Контейнер зависимостей для конкретного бота."""
+    bot: TelegramBot
+    chats: IChatRepository
+
+def build_services(bot: TelegramBot) -> BotServices:
+    return BotServices(bot=bot, chats=DjangoChatRepository())