57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Any
|
|
from uuid import UUID
|
|
|
|
from pydantic import field_validator
|
|
from sqlmodel import SQLModel
|
|
|
|
|
|
class GatewayBase(SQLModel):
|
|
name: str
|
|
url: str
|
|
main_session_key: str
|
|
workspace_root: str
|
|
skyll_enabled: bool = False
|
|
|
|
|
|
class GatewayCreate(GatewayBase):
|
|
token: str | None = None
|
|
|
|
@field_validator("token", mode="before")
|
|
@classmethod
|
|
def normalize_token(cls, value: Any) -> Any:
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, str):
|
|
value = value.strip()
|
|
return value or None
|
|
return value
|
|
|
|
|
|
class GatewayUpdate(SQLModel):
|
|
name: str | None = None
|
|
url: str | None = None
|
|
token: str | None = None
|
|
main_session_key: str | None = None
|
|
workspace_root: str | None = None
|
|
skyll_enabled: bool | None = None
|
|
|
|
@field_validator("token", mode="before")
|
|
@classmethod
|
|
def normalize_token(cls, value: Any) -> Any:
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, str):
|
|
value = value.strip()
|
|
return value or None
|
|
return value
|
|
|
|
|
|
class GatewayRead(GatewayBase):
|
|
id: UUID
|
|
token: str | None = None
|
|
created_at: datetime
|
|
updated_at: datetime
|