feat(db): add platform and scheduled_at to Draft model for scheduling and dashboard

Details:
- Updated Enum with SCHEDULED status.
- Migrated Draft table to include platform and scheduled_at.
- Expanded DraftRepository with get_recent_drafts and get_due_scheduled_drafts methods.
- Generated Alembic migration.

Author: PyDevDeep

backend/api/routes/drafts.py

@@ -30,7 +30,7 @@ async def create_draft(
     2. Відправляє фонову задачу генерації в Taskiq.
     3. Повертає task_id клієнту.
     """
-    draft_in = DraftCreate(topic=request.topic, user_id=request.user_id)
+    draft_in = DraftCreate(topic=request.topic, user_id=request.user_id, platform=request.platform)
     await draft_repo.create(draft_in)
 
     # Викликаємо фонову задачу

backend/models/db_models.py

@@ -35,6 +35,16 @@ class Draft(Base):
     topic: Mapped[str] = mapped_column(String(255), nullable=False)
     content: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
     status: Mapped[str] = mapped_column(String(50), nullable=False)
+
+    # --- НОВІ ПОЛЯ ---
+    platform: Mapped[str] = mapped_column(
+        String(50), nullable=False, server_default="telegram"
+    )
+    scheduled_at: Mapped[Optional[datetime]] = mapped_column(
+        DateTime(timezone=True), nullable=True
+    )
+    # -----------------
+
     created_at: Mapped[datetime] = mapped_column(
         DateTime(timezone=True), server_default=func.now(), nullable=False
     )

backend/models/enums.py

@@ -13,6 +13,7 @@ class DraftStatus(str, Enum):
     GENERATED = "generated"
     FAILED = "failed"
     PUBLISHED = "published"
+    SCHEDULED = "scheduled"
 
 
 class TaskStatus(str, Enum):

backend/models/schemas.py

@@ -32,18 +32,23 @@ class DraftBase(BaseModel):
 
 class DraftCreate(DraftBase):
     user_id: int
+    platform: Platform
 
 
 class DraftUpdate(BaseModel):
     content: Optional[str] = None
     status: Optional[DraftStatus] = None
+    platform: Optional[Platform] = None
+    scheduled_at: Optional[datetime] = None
 
 
 class DraftResponse(DraftBase, ORMModel):
     id: int
     user_id: int
     content: Optional[str]
     status: str
+    platform: str
+    scheduled_at: Optional[datetime]  # ДОДАНО
     created_at: datetime
     updated_at: datetime
 

backend/repositories/draft_repository.py

@@ -1,3 +1,5 @@
+from datetime import datetime, timezone
+
 from sqlalchemy import select, update
 from sqlalchemy.ext.asyncio import AsyncSession
 
@@ -11,7 +13,10 @@ def __init__(self, session: AsyncSession):
 
     async def create(self, draft_in: DraftCreate) -> Draft:
         db_draft = Draft(
-            user_id=draft_in.user_id, topic=draft_in.topic, status="pending"
+            user_id=draft_in.user_id,
+            topic=draft_in.topic,
+            status="pending",
+            platform=draft_in.platform,
         )
         self.session.add(db_draft)
         await self.session.flush()
@@ -36,3 +41,24 @@ async def update(self, draft_id: int, draft_update: DraftUpdate) -> Draft | None
         )
         result = await self.session.execute(stmt)
         return result.scalar_one_or_none()
+
+    async def get_recent_drafts(
+        self, limit: int = 10, platform: str | None = None
+    ) -> list[Draft]:
+        """Витягує останні драфти (для Дашборду в Slack)"""
+        stmt = select(Draft).order_by(Draft.updated_at.desc())
+        if platform:
+            stmt = stmt.where(Draft.platform == platform)
+        stmt = stmt.limit(limit)
+
+        result = await self.session.execute(stmt)
+        return list(result.scalars().all())
+
+    async def get_due_scheduled_drafts(self) -> list[Draft]:
+        """Витягує всі пости, час яких настав, але вони ще не опубліковані (для Планувальника)"""
+        now = datetime.now(timezone.utc)
+        stmt = select(Draft).where(
+            Draft.status == "scheduled", Draft.scheduled_at <= now
+        )
+        result = await self.session.execute(stmt)
+        return list(result.scalars().all())

backend/services/content_generator.py

@@ -65,7 +65,7 @@ async def generate_draft(
         topic: str,
         platform: str,
         source_url: str | None = None,
-        max_retries: int = 2,
+        max_retries: int = 1,
     ) -> str:
         logger.info(
             "draft_generation_started",

backend/workers/broker.py

@@ -2,11 +2,13 @@
 
 from taskiq_redis import ListQueueBroker, RedisAsyncResultBackend
 
+# from .scheduled_post import scheduled_post_task
 from backend.config.settings import settings
 from backend.workers.middlewares.logging import StructlogMiddleware
 from backend.workers.middlewares.metrics import PrometheusMetricsMiddleware
 from backend.workers.middlewares.retry import RetryTrackerMiddleware
 
+# from .scheduled_post import scheduled_post_task
 redis_url = f"redis://{settings.REDIS_HOST}:{settings.REDIS_PORT}/0"
 
 # Ініціалізуємо брокер з явно вказаною чергою (ізоляція від інших проєктів)

database/migrations/env.py

@@ -8,6 +8,8 @@
 from sqlalchemy.engine import Connection
 from sqlalchemy.ext.asyncio import async_engine_from_config
 
+from backend.models.db_models import Base
+
 load_dotenv()
 
 config = context.config
@@ -36,7 +38,7 @@ def get_url():
 # 2. Встановлюємо URL в конфігурацію
 config.set_main_option("sqlalchemy.url", get_url())
 
-target_metadata = None
+target_metadata = Base.metadata
 
 # other values from the config, defined by the needs of env.py,
 # can be acquired:

database/migrations/versions/3ee4fb291951_add_platform_and_scheduled_at_to_drafts.py

@@ -0,0 +1,41 @@
+"""add_platform_and_scheduled_at_to_drafts
+
+Revision ID: 3ee4fb291951
+Revises: 0c23d660630f
+Create Date: 2026-04-05 01:28:32.995752
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "3ee4fb291951"
+down_revision: Union[str, Sequence[str], None] = "0c23d660630f"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column(
+        "drafts",
+        sa.Column(
+            "platform", sa.String(length=50), server_default="telegram", nullable=False
+        ),
+    )
+    op.add_column(
+        "drafts", sa.Column("scheduled_at", sa.DateTime(timezone=True), nullable=True)
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("drafts", "scheduled_at")
+    op.drop_column("drafts", "platform")
+    # ### end Alembic commands ###