mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-15 11:04:12 +08:00
117 lines
3.7 KiB
Python
117 lines
3.7 KiB
Python
from typing import Optional
|
||
|
||
from sqlalchemy import Boolean, Column, Index, Integer, String, Text
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.db import Base, db_query, db_update, get_id_column
|
||
|
||
|
||
class AgentTask(Base):
|
||
"""
|
||
Agent 自主定时任务表。
|
||
"""
|
||
|
||
id = get_id_column()
|
||
# 任务名称
|
||
name = Column(String, nullable=False)
|
||
# 交给 Agent 执行的完整任务内容
|
||
content = Column(Text, nullable=False)
|
||
# 触发类型：date-单次触发，cron-周期触发
|
||
trigger_type = Column(String, nullable=False)
|
||
# 标准五段 cron 表达式
|
||
cron_expression = Column(String)
|
||
# 单次触发时间，使用带时区的 ISO 8601 格式
|
||
run_at = Column(String)
|
||
# 是否继续接受调度
|
||
enabled = Column(Boolean, nullable=False, default=True)
|
||
# 创建任务的用户与会话上下文
|
||
user_id = Column(String, nullable=False)
|
||
username = Column(String)
|
||
session_id = Column(String, nullable=False)
|
||
channel = Column(String)
|
||
source = Column(String)
|
||
original_chat_id = Column(String)
|
||
# 最近一次执行状态与结果
|
||
last_status = Column(String, nullable=False, default="waiting")
|
||
last_run_at = Column(String)
|
||
last_result = Column(Text)
|
||
# 最新一次真实执行的公开 ID，用于保护 last_* 投影不被旧运行覆盖
|
||
last_run_id = Column(String)
|
||
# 已收口执行次数；进程中断的未完成尝试不计入
|
||
run_count = Column(Integer, nullable=False, default=0)
|
||
created_at = Column(String, nullable=False)
|
||
updated_at = Column(String, nullable=False)
|
||
|
||
__table_args__ = (
|
||
Index("ix_agenttask_enabled", "enabled"),
|
||
Index("ix_agenttask_user_created", "user_id", "created_at", "id"),
|
||
)
|
||
|
||
@classmethod
|
||
@db_update
|
||
def add_task(cls, db: Session, **kwargs: object) -> int:
|
||
"""
|
||
新增 Agent 定时任务并返回任务 ID。
|
||
"""
|
||
task = cls(**kwargs)
|
||
db.add(task)
|
||
db.flush()
|
||
return task.id
|
||
|
||
@classmethod
|
||
@db_query
|
||
def get_for_user(
|
||
cls,
|
||
db: Session,
|
||
task_id: int,
|
||
user_id: Optional[str] = None,
|
||
) -> Optional["AgentTask"]:
|
||
"""
|
||
按任务 ID 和可选用户 ID 查询 Agent 定时任务。
|
||
"""
|
||
query = db.query(cls).filter(cls.id == task_id)
|
||
if user_id is not None:
|
||
query = query.filter(cls.user_id == user_id)
|
||
return query.first()
|
||
|
||
@classmethod
|
||
@db_query
|
||
def list_for_user(
|
||
cls,
|
||
db: Session,
|
||
user_id: Optional[str] = None,
|
||
enabled: Optional[bool] = None,
|
||
) -> list["AgentTask"]:
|
||
"""
|
||
按用户和启用状态查询 Agent 定时任务。
|
||
"""
|
||
query = db.query(cls)
|
||
if user_id is not None:
|
||
query = query.filter(cls.user_id == user_id)
|
||
if enabled is not None:
|
||
query = query.filter(cls.enabled.is_(enabled))
|
||
return query.order_by(cls.created_at.desc(), cls.id.desc()).all()
|
||
|
||
@classmethod
|
||
@db_update
|
||
def update_task(
|
||
cls,
|
||
db: Session,
|
||
task_id: int,
|
||
payload: dict,
|
||
user_id: Optional[str] = None,
|
||
) -> bool:
|
||
"""
|
||
仅在任务未运行时按任务 ID 和可选用户 ID 更新配置。
|
||
|
||
运行状态与配置必须在同一条条件更新中判定，避免执行认领后被并发配置写入
|
||
覆盖回可再次执行的状态。
|
||
"""
|
||
query = db.query(cls).filter(
|
||
cls.id == task_id,
|
||
cls.last_status != "running",
|
||
)
|
||
if user_id is not None:
|
||
query = query.filter(cls.user_id == user_id)
|
||
return bool(query.update(payload))
|