Refactor

src/applications/twitch_webhook/sent_notifications.py

@@ -0,0 +1,87 @@
+from enum import StrEnum
+from datetime import datetime, timezone
+
+from pydantic import BaseModel
+
+from core.mongo import mongo_manager
+
+from .state import State
+
+
+class SentNotificationType(StrEnum):
+    START_STREAM = "start_stream"
+    CHANGE_CATEGORY = "change_category"
+
+
+class SentResult(BaseModel):
+    success: bool
+    message_id: str | None
+
+
+class SentNotification(BaseModel):
+    notification_type: SentNotificationType
+    twitch_id: int
+    state: State
+    sent_result: dict[str, SentResult]
+    sent_at: datetime
+
+
+class SentNotificationRepository:
+    COLLECTION_NAME = "sent_notifications"
+
+    @classmethod
+    async def add(
+        cls,
+        twitch_id: int,
+        notification_type: SentNotificationType,
+        state: State,
+        sent_result: dict[str, SentResult],
+    ):
+        async with mongo_manager.connect() as client:
+            db = client.get_default_database()
+            collection = db[cls.COLLECTION_NAME]
+
+            await collection.insert_one(
+                SentNotification(
+                    notification_type=notification_type,
+                    twitch_id=twitch_id,
+                    state=state,
+                    sent_at=datetime.now(timezone.utc),
+                    sent_result=sent_result,
+                ).model_dump()
+            )
+
+    @classmethod
+    async def get_penultimate_for_streamer(
+        cls, twitch_id: int
+    ) -> SentNotification | None:
+        async with mongo_manager.connect() as client:
+            db = client.get_default_database()
+            collection = db[cls.COLLECTION_NAME]
+
+            doc = await collection.find_one(
+                {"twitch_id": twitch_id},
+                sort={"sent_at": -1},
+                skip=1,
+            )
+            if doc is None:
+                return None
+
+            return SentNotification(**doc)
+
+    @classmethod
+    async def get_last_for_streamer(
+        cls, twitch_id: int
+    ) -> SentNotification | None:
+        async with mongo_manager.connect() as client:
+            db = client.get_default_database()
+            collection = db[cls.COLLECTION_NAME]
+
+            doc = await collection.find_one(
+                {"twitch_id": twitch_id},
+                sort={"sent_at": -1},
+            )
+            if doc is None:
+                return None
+
+            return SentNotification(**doc)