bot dev

bot/handlers.py

@@ -0,0 +1,61 @@
+from datetime import datetime, timezone as py_tz
+from telegram import Update
+from telegram.ext import ContextTypes
+from django.utils import timezone
+from .models import TelegramChat
+
+def _now_aware():
+    return timezone.now()
+
+async def upsert_chat_from_update(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    """
+    Вызывайте в message-хендлерах, чтобы обновлять last_message_at и тайтл.
+    """
+    chat = update.effective_chat
+    if not chat:
+        return
+
+    obj, created = TelegramChat.objects.update_or_create(
+        id=chat.id,
+        defaults={
+            "type": chat.type,
+            "title": chat.title or "",
+            "username": chat.username or "",
+            "is_member": True,  # если есть сообщения — бот, скорее всего, в чате
+            "last_message_at": _now_aware(),
+        }
+    )
+    # created можно игнорировать; при необходимости — логировать
+
+
+async def on_my_chat_member(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    """
+    Ловим изменение статуса бота в чатах (добавили/кикнули).
+    """
+    if not update.my_chat_member:
+        return
+
+    chat = update.my_chat_member.chat
+    new_status = update.my_chat_member.new_chat_member.status  # "member"/"administrator"/"kicked"/"left"/...
+    is_in_chat = new_status in ("member", "administrator")
+
+    defaults = {
+        "type": chat.type,
+        "title": chat.title or "",
+        "username": getattr(chat, "username", "") or "",
+        "is_member": is_in_chat,
+        "last_message_at": _now_aware(),
+    }
+
+    if is_in_chat:
+        defaults["left_at"] = None
+        defaults.setdefault("joined_at", _now_aware())
+
+    obj, _ = TelegramChat.objects.update_or_create(
+        id=chat.id,
+        defaults=defaults
+    )
+
+    if not is_in_chat and obj.left_at is None:
+        obj.left_at = _now_aware()
+        obj.save(update_fields=["is_member", "left_at"])