node_run.py 1.2 KB

12345678910111213141516171819202122232425
  1. from datetime import datetime
  2. from sqlalchemy import DateTime, Integer, String, Text
  3. from sqlalchemy.orm import Mapped, mapped_column
  4. from core_db import AuditMixin, Base, TenantMixin, VersionMixin
  5. class NodeRun(TenantMixin, AuditMixin, VersionMixin, Base):
  6. __tablename__ = "node_run"
  7. run_id: Mapped[str] = mapped_column(String(36), index=True)
  8. parent_node_run_id: Mapped[str | None] = mapped_column(String(36), nullable=True)
  9. node_id: Mapped[str] = mapped_column(String(128))
  10. node_type: Mapped[str] = mapped_column(String(32))
  11. attempt_no: Mapped[int] = mapped_column(Integer, default=1)
  12. status: Mapped[str] = mapped_column(String(32), default="pending", index=True)
  13. worker_key: Mapped[str | None] = mapped_column(String(128), nullable=True)
  14. lease_expire_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
  15. queued_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
  16. started_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
  17. finished_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
  18. error_code: Mapped[str | None] = mapped_column(String(64), nullable=True)
  19. error_message: Mapped[str | None] = mapped_column(Text, nullable=True)