model.py 1.7 KB

12345678910111213141516171819202122232425262728
  1. from sqlalchemy import Boolean, Integer, String, Text
  2. from sqlalchemy.orm import Mapped, mapped_column
  3. from app.core.base_model import ModelMixin, UserMixin
  4. class NodeModel(ModelMixin, UserMixin):
  5. """
  6. 节点类型模型 - 动态定义节点类型
  7. """
  8. __tablename__: str = "task_node"
  9. __table_args__: dict[str, str] = {"comment": "节点类型表"}
  10. __loader_options__: list[str] = ["created_by", "updated_by"]
  11. name: Mapped[str] = mapped_column(String(64), nullable=False, comment="节点名称")
  12. code: Mapped[str] = mapped_column(String(32), nullable=False, unique=True, comment="节点编码")
  13. jobstore: Mapped[str | None] = mapped_column(String(64), nullable=True, default="default", comment="存储器")
  14. executor: Mapped[str | None] = mapped_column(String(64), nullable=True, default="default", comment="执行器")
  15. trigger: Mapped[str | None] = mapped_column(String(64), nullable=True, comment="触发器")
  16. trigger_args: Mapped[str | None] = mapped_column(Text, nullable=True, comment="触发器参数")
  17. func: Mapped[str | None] = mapped_column(Text, nullable=True, comment="代码块")
  18. args: Mapped[str | None] = mapped_column(Text, nullable=True, comment="位置参数")
  19. kwargs: Mapped[str | None] = mapped_column(Text, nullable=True, comment="关键字参数")
  20. coalesce: Mapped[bool] = mapped_column(Boolean, nullable=True, default=False, comment="是否合并运行")
  21. max_instances: Mapped[int] = mapped_column(Integer, nullable=True, default=1, comment="最大实例数")
  22. start_date: Mapped[str | None] = mapped_column(String(64), nullable=True, comment="开始时间")
  23. end_date: Mapped[str | None] = mapped_column(String(64), nullable=True, comment="结束时间")