Vladless
92 lines
2.0 KiB
Python
92 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any, Literal
|
|
|
|
from pydantic import BaseModel, ConfigDict
|
|
from pydantic.alias_generators import to_camel
|
|
|
|
|
|
_FLOW_CONFIG = ConfigDict(populate_by_name=True, alias_generator=to_camel)
|
|
|
|
|
|
class EdgeConditionSchema(BaseModel):
|
|
model_config = _FLOW_CONFIG
|
|
|
|
field: str
|
|
operator: str
|
|
value: Any = None
|
|
|
|
|
|
class EdgeConditionGroupSchema(BaseModel):
|
|
model_config = _FLOW_CONFIG
|
|
|
|
logic: Literal["and", "or"] = "and"
|
|
conditions: list[EdgeConditionSchema] = []
|
|
|
|
|
|
class FlowEdgeSchema(BaseModel):
|
|
model_config = _FLOW_CONFIG
|
|
|
|
id: str
|
|
source: str
|
|
target: str
|
|
condition: EdgeConditionSchema | None = None
|
|
condition_group: EdgeConditionGroupSchema | None = None
|
|
label: str | None = None
|
|
priority: int | None = None
|
|
|
|
|
|
class FlowNodeActionConfigSchema(BaseModel):
|
|
model_config = _FLOW_CONFIG
|
|
|
|
action_type: str
|
|
params: dict[str, Any] = {}
|
|
|
|
|
|
class FlowNodeSchema(BaseModel):
|
|
model_config = _FLOW_CONFIG
|
|
|
|
id: str
|
|
type: str
|
|
label: str
|
|
label_en: str | None = None
|
|
enabled: bool = True
|
|
page_slug: str | None = None
|
|
cabinet_tab: str | None = None
|
|
screen_group: str | None = None
|
|
screen_id: str | None = None
|
|
auto_skip_if: EdgeConditionSchema | EdgeConditionGroupSchema | None = None
|
|
action_config: FlowNodeActionConfigSchema | None = None
|
|
config: dict = {}
|
|
position: dict
|
|
|
|
|
|
class FlowResponse(BaseModel):
|
|
model_config = _FLOW_CONFIG
|
|
|
|
id: str
|
|
name: str
|
|
nodes: list[FlowNodeSchema]
|
|
edges: list[FlowEdgeSchema]
|
|
entry_node_id: str | None
|
|
version: int
|
|
|
|
|
|
class FlowUpdate(BaseModel):
|
|
model_config = _FLOW_CONFIG
|
|
|
|
name: str | None = None
|
|
nodes: list[FlowNodeSchema]
|
|
edges: list[FlowEdgeSchema]
|
|
entry_node_id: str | None = None
|
|
|
|
|
|
class FlowCreate(BaseModel):
|
|
model_config = _FLOW_CONFIG
|
|
|
|
id: str
|
|
name: str
|
|
nodes: list[FlowNodeSchema] = []
|
|
edges: list[FlowEdgeSchema] = []
|
|
entry_node_id: str | None = None
|