from sqlalchemy import Column, Integer, DateTime, String, func
from sqlalchemy.orm import declarative_base

Base = declarative_base()

class Task(Base):
    __tablename__ = 'task'

    id = Column(Integer, primary_key=True)
    gmt_create = Column(DateTime(timezone=True), server_default=func.now())
    gmt_modified = Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())
    prompt_id = Column(String, index=True, nullable=False, unique=True)
    result = Column(String, nullable=True)
    job_id = Column(Integer, index=True, nullable=False, unique=True)
    status = Column(Integer)

    def __repr__(self):
        return f"{self.id},{self.gmt_create},{self.gmt_modified},{self.prompt_id},{self.result}"