feat: schedule node add python version
This commit is contained in:
@@ -5,7 +5,7 @@ from typing import Any
|
||||
|
||||
from pydantic import Field, field_validator, model_validator
|
||||
|
||||
from common.db.models.schedules import FailurePolicy, TriggerType
|
||||
from common.db.models.schedules import FailurePolicy, PythonVersion, TriggerType
|
||||
from common.schemas import StrictModel
|
||||
|
||||
|
||||
@@ -114,6 +114,7 @@ class CreateScheduleNodeRequest(StrictModel):
|
||||
position_y: float = Field(default=0, ge=-100_000, le=100_000)
|
||||
arguments_json: dict[str, Any] = Field(default_factory=dict, max_length=100)
|
||||
env_refs_json: dict[str, str] = Field(default_factory=dict, max_length=100)
|
||||
python_version: PythonVersion = Field(default="3.12")
|
||||
|
||||
@field_validator("node_key", "node_name")
|
||||
@classmethod
|
||||
@@ -146,6 +147,7 @@ class UpdateScheduleNodeRequest(StrictModel):
|
||||
default=None,
|
||||
max_length=100,
|
||||
)
|
||||
python_version: PythonVersion | None = Field(default=None)
|
||||
|
||||
@field_validator("node_name")
|
||||
@classmethod
|
||||
|
||||
@@ -151,6 +151,7 @@ def node_payload(
|
||||
"node_key": item.node_key,
|
||||
"node_name": item.node_name,
|
||||
"versions_id": item.versions_id,
|
||||
"python_version": item.python_version,
|
||||
"timeout_seconds": item.timeout_seconds,
|
||||
"retry_count": item.retry_count,
|
||||
"retry_interval_sec": item.retry_interval_sec,
|
||||
@@ -787,6 +788,7 @@ async def create_schedule_node(
|
||||
node_key=payload.node_key,
|
||||
node_name=payload.node_name,
|
||||
versions_id=payload.versions_id,
|
||||
python_version=payload.python_version,
|
||||
timeout_seconds=payload.timeout_seconds,
|
||||
retry_count=payload.retry_count,
|
||||
retry_interval_sec=payload.retry_interval_sec,
|
||||
@@ -837,6 +839,7 @@ async def update_schedule_node(
|
||||
mutable_fields = {
|
||||
"node_name",
|
||||
"versions_id",
|
||||
"python_version",
|
||||
"timeout_seconds",
|
||||
"retry_count",
|
||||
"retry_interval_sec",
|
||||
|
||||
Reference in New Issue
Block a user