88 lines
2.2 KiB
Python
88 lines
2.2 KiB
Python
"""Schemas for gateway CRUD and template-sync API payloads."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from uuid import UUID
|
|
|
|
from pydantic import field_validator
|
|
from sqlmodel import Field, SQLModel
|
|
|
|
RUNTIME_ANNOTATION_TYPES = (datetime, UUID)
|
|
|
|
|
|
class GatewayBase(SQLModel):
|
|
"""Shared gateway fields used across create/read payloads."""
|
|
|
|
name: str
|
|
url: str
|
|
workspace_root: str
|
|
|
|
|
|
class GatewayCreate(GatewayBase):
|
|
"""Payload for creating a gateway configuration."""
|
|
|
|
token: str | None = None
|
|
|
|
@field_validator("token", mode="before")
|
|
@classmethod
|
|
def normalize_token(cls, value: object) -> str | None | object:
|
|
"""Normalize empty/whitespace tokens to `None`."""
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, str):
|
|
value = value.strip()
|
|
return value or None
|
|
return value
|
|
|
|
|
|
class GatewayUpdate(SQLModel):
|
|
"""Payload for partial gateway updates."""
|
|
|
|
name: str | None = None
|
|
url: str | None = None
|
|
token: str | None = None
|
|
workspace_root: str | None = None
|
|
|
|
@field_validator("token", mode="before")
|
|
@classmethod
|
|
def normalize_token(cls, value: object) -> str | None | object:
|
|
"""Normalize empty/whitespace tokens to `None`."""
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, str):
|
|
value = value.strip()
|
|
return value or None
|
|
return value
|
|
|
|
|
|
class GatewayRead(GatewayBase):
|
|
"""Gateway payload returned from read endpoints."""
|
|
|
|
id: UUID
|
|
organization_id: UUID
|
|
token: str | None = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
|
|
class GatewayTemplatesSyncError(SQLModel):
|
|
"""Per-agent error entry from a gateway template sync operation."""
|
|
|
|
agent_id: UUID | None = None
|
|
agent_name: str | None = None
|
|
board_id: UUID | None = None
|
|
message: str
|
|
|
|
|
|
class GatewayTemplatesSyncResult(SQLModel):
|
|
"""Summary payload returned by gateway template sync endpoints."""
|
|
|
|
gateway_id: UUID
|
|
include_main: bool
|
|
reset_sessions: bool
|
|
agents_updated: int
|
|
agents_skipped: int
|
|
main_updated: bool
|
|
errors: list[GatewayTemplatesSyncError] = Field(default_factory=list)
|