from .base import BaseModel
from . import db


class CronJob(BaseModel):
    __tablename__ = "cron_jobs"

    agent_id = db.Column(
        db.BigInteger, db.ForeignKey("agents.id"), nullable=False, comment="Agent ID"
    )
    name = db.Column(db.String(100), nullable=False, comment="任务名称")
    cron_expression = db.Column(db.String(50), nullable=False, comment="Cron 表达式")
    prompt = db.Column(db.Text, nullable=False, comment="执行提示词")
    is_active = db.Column(db.Boolean, default=True, nullable=False, comment="是否激活")
    last_run_at = db.Column(db.DateTime, nullable=True, comment="上次运行时间")
    next_run_at = db.Column(db.DateTime, nullable=True, comment="下次运行时间")