73 lines
4.1 KiB
Python
73 lines
4.1 KiB
Python
"""ORM models for user scheduled tasks."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
|
|
from sqlalchemy import JSON, Boolean, DateTime, String, Text, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from deerflow.persistence.base import Base
|
|
|
|
|
|
class SchedulerThreadRow(Base):
|
|
__tablename__ = "scheduler_threads"
|
|
|
|
user_id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
|
thread_id: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC))
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC), onupdate=lambda: datetime.now(UTC))
|
|
|
|
|
|
class ScheduledTaskRow(Base):
|
|
__tablename__ = "scheduled_tasks"
|
|
__table_args__ = (UniqueConstraint("user_id", "name", name="uq_scheduled_tasks_user_name"),)
|
|
|
|
task_id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
|
user_id: Mapped[str] = mapped_column(String(64), index=True)
|
|
scheduler_thread_id: Mapped[str] = mapped_column(String(64), index=True)
|
|
source_thread_id: Mapped[str | None] = mapped_column(String(64), index=True)
|
|
source_agent_name: Mapped[str | None] = mapped_column(String(128), index=True)
|
|
execution_agent_name: Mapped[str | None] = mapped_column(String(128), index=True)
|
|
name: Mapped[str] = mapped_column(String(256))
|
|
prompt: Mapped[str] = mapped_column(Text)
|
|
schedule_text: Mapped[str] = mapped_column(String(256))
|
|
cron_expr: Mapped[str] = mapped_column(String(128))
|
|
timezone: Mapped[str] = mapped_column(String(64), default="UTC")
|
|
enabled: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
next_run_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), index=True)
|
|
last_run_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
last_status: Mapped[str | None] = mapped_column(String(32))
|
|
execution_context_json: Mapped[dict] = mapped_column(JSON, default=dict)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC))
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC), onupdate=lambda: datetime.now(UTC))
|
|
|
|
|
|
class ScheduledTaskRunRow(Base):
|
|
__tablename__ = "scheduled_task_runs"
|
|
__table_args__ = (UniqueConstraint("task_id", "scheduled_for", name="uq_scheduled_task_runs_task_scheduled_for"),)
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
|
task_id: Mapped[str] = mapped_column(String(64), index=True)
|
|
user_id: Mapped[str] = mapped_column(String(64), index=True)
|
|
scheduler_thread_id: Mapped[str] = mapped_column(String(64), index=True)
|
|
agent_run_id: Mapped[str | None] = mapped_column(String(64), index=True)
|
|
scheduled_for: Mapped[datetime] = mapped_column(DateTime(timezone=True), index=True)
|
|
started_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
finished_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
status: Mapped[str] = mapped_column(String(32), default="pending")
|
|
error: Mapped[str | None] = mapped_column(Text)
|
|
result_json: Mapped[dict | None] = mapped_column(JSON, default=dict)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC))
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC), onupdate=lambda: datetime.now(UTC))
|
|
|
|
|
|
class ScheduledTaskDeliveryProfileRow(Base):
|
|
__tablename__ = "scheduled_task_delivery_profiles"
|
|
|
|
user_id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
|
element_user_id: Mapped[str | None] = mapped_column(String(256), index=True)
|
|
element_room_id: Mapped[str | None] = mapped_column(String(256), nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC))
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC), onupdate=lambda: datetime.now(UTC))
|