"""发布任务：一次 ImportProductsV3 请求与轮询结果。"""
from __future__ import annotations

import uuid
from datetime import datetime

from sqlalchemy import BigInteger, DateTime, Enum, ForeignKey, Uuid, func
from sqlalchemy.orm import Mapped, mapped_column

from db import Base
from models.enums import PublishStatus
from models.types import JSONType


class PublishTask(Base):
    __tablename__ = "publish_tasks"

    id: Mapped[uuid.UUID] = mapped_column(Uuid(as_uuid=True), primary_key=True, default=uuid.uuid4)
    product_id: Mapped[uuid.UUID] = mapped_column(
        Uuid(as_uuid=True), ForeignKey("products.id", ondelete="CASCADE"), index=True
    )
    shop_id: Mapped[uuid.UUID] = mapped_column(
        Uuid(as_uuid=True), ForeignKey("shops.id", ondelete="CASCADE"), index=True
    )
    ozon_task_id: Mapped[int | None] = mapped_column(BigInteger, nullable=True, index=True)
    status: Mapped[PublishStatus] = mapped_column(
        Enum(PublishStatus, native_enum=False, length=16), default=PublishStatus.pending, index=True
    )
    request_payload: Mapped[dict | None] = mapped_column(JSONType, nullable=True)  # 脱敏后的 items[0]
    response: Mapped[dict | None] = mapped_column(JSONType, nullable=True)         # import/info 原始结果
    errors: Mapped[list | None] = mapped_column(JSONType, nullable=True)
    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
    completed_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)