60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Self
|
|
from uuid import UUID
|
|
|
|
from pydantic import model_validator
|
|
from sqlmodel import SQLModel
|
|
|
|
|
|
class BoardBase(SQLModel):
|
|
name: str
|
|
slug: str
|
|
gateway_id: UUID | None = None
|
|
board_group_id: UUID | None = None
|
|
board_type: str = "goal"
|
|
objective: str | None = None
|
|
success_metrics: dict[str, object] | None = None
|
|
target_date: datetime | None = None
|
|
goal_confirmed: bool = False
|
|
goal_source: str | None = None
|
|
|
|
|
|
class BoardCreate(BoardBase):
|
|
gateway_id: UUID
|
|
|
|
@model_validator(mode="after")
|
|
def validate_goal_fields(self) -> Self:
|
|
if self.board_type == "goal" and self.goal_confirmed:
|
|
if not self.objective or not self.success_metrics:
|
|
raise ValueError("Confirmed goal boards require objective and success_metrics")
|
|
return self
|
|
|
|
|
|
class BoardUpdate(SQLModel):
|
|
name: str | None = None
|
|
slug: str | None = None
|
|
gateway_id: UUID | None = None
|
|
board_group_id: UUID | None = None
|
|
board_type: str | None = None
|
|
objective: str | None = None
|
|
success_metrics: dict[str, object] | None = None
|
|
target_date: datetime | None = None
|
|
goal_confirmed: bool | None = None
|
|
goal_source: str | None = None
|
|
|
|
@model_validator(mode="after")
|
|
def validate_gateway_id(self) -> Self:
|
|
# Treat explicit null like "unset" is invalid for patch updates.
|
|
if "gateway_id" in self.model_fields_set and self.gateway_id is None:
|
|
raise ValueError("gateway_id is required")
|
|
return self
|
|
|
|
|
|
class BoardRead(BoardBase):
|
|
id: UUID
|
|
organization_id: UUID
|
|
created_at: datetime
|
|
updated_at: datetime
|