from sqlalchemy import Column, Integer, String, Text, Boolean, TIMESTAMP, func
from sqlalchemy.dialects.postgresql import BIGINT
from dataclasses import dataclass
from database.database import Base

@dataclass
class TContentDispatch(Base):
    __tablename__ = 't_content_dispatch'

    id: int = Column(BIGINT, primary_key=True, autoincrement=True, comment='自动递增的唯一任务ID')
    creation_date: str = Column(TIMESTAMP(timezone=True), server_default=func.now(), nullable=False, comment='记录数据条目的创建时间')
    is_sent: bool = Column(Boolean, default=False, nullable=False, comment='表示数据条目是否已被发送')
    category: str = Column(String(255), nullable=False, comment='类别')
    title: str = Column(String(255), nullable=False, comment='标题')
    cover_image: str = Column(Text, nullable=True, comment='封面')
    poster_image: str = Column(Text, nullable=True, comment='海报')
    opening_text: str = Column(Text, nullable=True, comment='开头语')
    content: str = Column(Text, nullable=False, comment='内容')
    ai_content: str = Column(Text, nullable=True, comment='AI编辑的内容')
    closing_text: str = Column(Text, nullable=True, comment='结束语')
    is_scheduled: bool = Column(Boolean, default=False, nullable=False, comment='是否设置为定时发送')
    schedule_time: str = Column(TIMESTAMP, nullable=True, comment='定时发送的具体时间')
    format: str = Column(String(50), nullable=True, comment='数据条目的格式')
    ai_generate: int = Column(Integer, default=0, nullable=False, comment='是否AI生成。0否，1部分参与，2是。')

    def __repr__(self):
        return f"<TContentDispatch(id={self.id}, title={self.title}, category={self.category})>"