Pouzor
3afc8ed3d8
Add waypoints JSON column to edges table, include it in all edge schemas (EdgeBase, EdgeUpdate, canvas CanvasEdge) and add the idempotent ALTER TABLE migration so existing databases are upgraded on next startup.
70 lines
1.8 KiB
Python
70 lines
1.8 KiB
Python
from typing import Any
|
|
|
|
from pydantic import BaseModel, field_validator
|
|
|
|
from app.schemas.edges import EdgeResponse
|
|
from app.schemas.nodes import NodeResponse
|
|
from app.schemas.utils import normalize_animated
|
|
|
|
|
|
class NodeSave(BaseModel):
|
|
id: str
|
|
type: str
|
|
label: str
|
|
hostname: str | None = None
|
|
ip: str | None = None
|
|
mac: str | None = None
|
|
os: str | None = None
|
|
status: str = "unknown"
|
|
check_method: str | None = None
|
|
check_target: str | None = None
|
|
services: list[Any] = []
|
|
notes: str | None = None
|
|
parent_id: str | None = None
|
|
container_mode: bool = False
|
|
custom_colors: dict[str, Any] | None = None
|
|
custom_icon: str | None = None
|
|
cpu_count: int | None = None
|
|
cpu_model: str | None = None
|
|
ram_gb: float | None = None
|
|
disk_gb: float | None = None
|
|
show_hardware: bool = False
|
|
width: float | None = None
|
|
height: float | None = None
|
|
bottom_handles: int = 1
|
|
pos_x: float = 0
|
|
pos_y: float = 0
|
|
|
|
|
|
class EdgeSave(BaseModel):
|
|
id: str
|
|
source: str
|
|
target: str
|
|
type: str = "ethernet"
|
|
label: str | None = None
|
|
vlan_id: int | None = None
|
|
speed: str | None = None
|
|
custom_color: str | None = None
|
|
path_style: str | None = None
|
|
animated: str = 'none'
|
|
source_handle: str | None = None
|
|
target_handle: str | None = None
|
|
waypoints: list | None = None
|
|
|
|
@field_validator('animated', mode='before')
|
|
@classmethod
|
|
def validate_animated(cls, v: object) -> str:
|
|
return normalize_animated(v)
|
|
|
|
|
|
class CanvasSaveRequest(BaseModel):
|
|
nodes: list[NodeSave] = []
|
|
edges: list[EdgeSave] = []
|
|
viewport: dict[str, Any] = {}
|
|
|
|
|
|
class CanvasStateResponse(BaseModel):
|
|
nodes: list[NodeResponse]
|
|
edges: list[EdgeResponse]
|
|
viewport: dict[str, Any]
|