fix(core): resolve Pylance type errors and fix draft persistence

Details:
- Replaced string status literals with DraftStatus Enum to fix Pylance warnings.
- Added user and draft creation to DB on initial generation request so drafts appear on the App Home dashboard.
- Updated generate_draft_task to persist generated and failed content to the database.

Author: PyDevDeep

backend/api/routes/feedback.py

@@ -1,15 +1,16 @@
 import json
-import uuid
 
 import httpx
 import structlog
 from fastapi import APIRouter, Depends, HTTPException, Request, Response
+from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from backend.config.lexicon import SLACK_UI
 from backend.config.settings import settings
-from backend.models.enums import Platform
-from backend.models.schemas import DraftUpdate
+from backend.models.db_models import User
+from backend.models.enums import DraftStatus, Platform
+from backend.models.schemas import DraftCreate, DraftUpdate
 from backend.repositories.draft_repository import DraftRepository
 from backend.workers.dependencies import get_db_session
 from backend.workers.tasks.generate_draft import generate_draft_task
@@ -148,6 +149,13 @@ async def slack_interactions(
 
             async with httpx.AsyncClient() as client:
                 if action_id == "action_publish_draft":
+                    # ДОДАНО: Змінюємо статус на Опубліковано
+                    if draft_id.isdigit():
+                        repo = DraftRepository(session)
+                        await repo.update(
+                            int(draft_id), DraftUpdate(status=DraftStatus.PUBLISHED)
+                        )
+
                     await publish_post_task.kiq(
                         post_id=draft_id, platform=platform, content=draft_text
                     )
@@ -210,20 +218,25 @@ async def slack_interactions(
         callback_id = view.get("callback_id")
         state_values = view.get("state", {}).get("values", {})
 
+        # --- СЦЕНАРІЙ 1: Генерація нового драфту ---
         # --- СЦЕНАРІЙ 1: Генерація нового драфту ---
         if callback_id == "modal_generate_draft":
-            channel_id = view.get("private_metadata")  # Дістаємо збережений канал
+            channel_id = view.get("private_metadata")
             topic = (
                 state_values.get("block_topic_input", {})
                 .get("input_topic", {})
                 .get("value", "")
                 .strip()
             )
-            platform = (
-                state_values.get("block_platform_select", {})
-                .get("input_platform_select", {})
-                .get("selected_option", {})
-                .get("value", "telegram")
+
+            block_state = state_values.get("block_platform_select", {}).get(
+                "input_platform_select", {}
+            )
+            selected_option = block_state.get("selected_option")
+            platform = Platform(
+                selected_option.get("value", "telegram")
+                if selected_option
+                else "telegram"
             )
 
             logger.info(
@@ -233,16 +246,34 @@ async def slack_interactions(
                 platform=platform,
             )
 
-            new_draft_id = str(uuid.uuid4())  # ГЕНЕРУЄМО УНІКАЛЬНИЙ ID
+            # --- ДОДАНО: СТВОРЕННЯ КОРИСТУВАЧА ТА ДРАФТУ В БД ---
+            # 1. Знаходимо або створюємо користувача (Slack ID)
+            user_query = await session.execute(
+                select(User).where(User.username == user_id)
+            )
+            db_user = user_query.scalar_one_or_none()
+            if not db_user:
+                db_user = User(username=user_id)
+                session.add(db_user)
+                await session.flush()
+
+            # 2. Створюємо драфт (статус pending)
+            repo = DraftRepository(session)
+            new_draft = await repo.create(
+                DraftCreate(topic=topic, platform=platform, user_id=db_user.id)
+            )
+            real_draft_id = str(new_draft.id)
+            # ---------------------------------------------------
+
+            # Передаємо РЕАЛЬНИЙ ID з бази замість UUID
             await generate_draft_task.kiq(  # type: ignore[call-overload]
                 topic=topic,
-                platform=platform,
+                platform=platform.value,
                 user_id=user_id,
                 channel_id=channel_id,
-                draft_id=new_draft_id,
+                draft_id=real_draft_id,
             )
 
-            # Відправляємо підтвердження в канал
             async with httpx.AsyncClient() as client:
                 await client.post(
                     "https://slack.com/api/chat.postMessage",

backend/tests/unit/test_tasks.py

@@ -21,8 +21,8 @@ async def test_generate_draft_task_success():
     source_url = "https://pubmed.org/123"
 
     # Act: Прямий виклик функції (в обхід брокера) з ін'єкцією мока
-    result = await generate_draft_task(
-        topic=topic, platform=platform, source_url=source_url, generator=mock_generator
+    result: str = await generate_draft_task(
+        topic=topic, platform=platform, source_url=source_url, generator=mock_generator, session=AsyncMock()
     )
 
     # Assert: Перевіряємо результат та аргументи виклику
@@ -47,7 +47,7 @@ async def test_generate_draft_task_judge_failure():
     # Act & Assert
     with pytest.raises(JudgeFailedError) as exc_info:
         await generate_draft_task(
-            topic="Стрес", platform="twitter", source_url=None, generator=mock_generator
+            topic="Стрес", platform="twitter", source_url=None, generator=mock_generator, session=AsyncMock()
         )
 
     assert exc_info.value.topic == "Стрес"
@@ -73,4 +73,5 @@ async def test_generate_draft_task_unexpected_error():
             platform="threads",
             source_url=None,
             generator=mock_generator,
+            session=AsyncMock(),
         )

backend/workers/tasks/generate_draft.py

@@ -1,12 +1,16 @@
 from typing import Annotated
 
 import structlog
+from sqlalchemy.ext.asyncio import AsyncSession
 from taskiq import TaskiqDepends
 
+from backend.models.enums import DraftStatus
+from backend.models.schemas import DraftUpdate
+from backend.repositories.draft_repository import DraftRepository
 from backend.services.content_generator import ContentGenerator, JudgeFailedError
 from backend.workers.broker import broker
 from backend.workers.callbacks import notify_slack_on_complete, notify_slack_on_failure
-from backend.workers.dependencies import get_content_generator
+from backend.workers.dependencies import get_content_generator, get_db_session
 
 logger = structlog.get_logger()
 
@@ -17,6 +21,7 @@ async def generate_draft_task(
     topic: str,
     platform: str,
     generator: Annotated[ContentGenerator, TaskiqDepends(get_content_generator)],
+    session: Annotated[AsyncSession, TaskiqDepends(get_db_session)],
     source_url: str | None = None,
     user_id: str | None = None,
     channel_id: str | None = None,
@@ -42,7 +47,12 @@ async def generate_draft_task(
         logger.info(
             "background_task_success", task="generate_medical_draft", topic=topic
         )
-
+        # --- ДОДАНО: ЗБЕРЕЖЕННЯ ТЕКСТУ ТА СТАТУСУ В БД ---
+        if draft_id.isdigit():
+            repo = DraftRepository(session)
+            await repo.update(
+                int(draft_id), DraftUpdate(content=result, status=DraftStatus.GENERATED)
+            )
         # ДОДАНО: Відправка готового результату у Slack
         if user_id and channel_id:
             await notify_slack_on_complete(
@@ -63,6 +73,14 @@ async def generate_draft_task(
             topic=e.topic,
             attempts=e.attempts,
         )
+        # --- ДОДАНО: ЗБЕРЕЖЕННЯ "БРУДНОГО" ТЕКСТУ ---
+        if draft_id.isdigit():
+            repo = DraftRepository(session)
+            await repo.update(
+                int(draft_id), DraftUpdate(content=e.draft, status=DraftStatus.FAILED)
+            )
+
+        # ---------------------------------------------
         # ВІДДАЄМО ДРАФТ З КНОПКАМИ, АЛЕ З АЛЕРТОМ (is_valid=False)
         if user_id and channel_id:
             await notify_slack_on_complete(