2d99c52025
- Add is_public field to Tree model (private by default) - Update access control: users see default trees, public trees, or their own - Update all tree endpoints (list, search, get, categories) with new visibility logic - Default/system trees are automatically marked as public - Add migration 004 to add is_public column and update existing defaults - Fix pydantic settings to ignore extra env vars (DATABASE_URL_SYNC) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
from datetime import datetime
|
|
from typing import Optional, Any
|
|
from uuid import UUID
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class TreeBase(BaseModel):
|
|
name: str = Field(..., min_length=1, max_length=255)
|
|
description: Optional[str] = None
|
|
category: Optional[str] = Field(None, max_length=100)
|
|
|
|
|
|
class TreeCreate(TreeBase):
|
|
tree_structure: dict[str, Any] = Field(..., description="The decision tree structure in JSON format")
|
|
is_public: bool = Field(False, description="Make tree visible to all users")
|
|
is_default: bool = Field(False, description="Mark as a default/system tree (admin only)")
|
|
|
|
|
|
class TreeUpdate(BaseModel):
|
|
name: Optional[str] = Field(None, min_length=1, max_length=255)
|
|
description: Optional[str] = None
|
|
category: Optional[str] = Field(None, max_length=100)
|
|
tree_structure: Optional[dict[str, Any]] = None
|
|
is_public: Optional[bool] = None
|
|
is_active: Optional[bool] = None
|
|
|
|
|
|
class TreeResponse(TreeBase):
|
|
id: UUID
|
|
tree_structure: dict[str, Any]
|
|
author_id: Optional[UUID] = None
|
|
team_id: Optional[UUID] = None
|
|
is_active: bool
|
|
is_public: bool
|
|
is_default: bool
|
|
version: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
usage_count: int
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class TreeListResponse(BaseModel):
|
|
id: UUID
|
|
name: str
|
|
description: Optional[str] = None
|
|
category: Optional[str] = None
|
|
is_active: bool
|
|
is_public: bool
|
|
is_default: bool
|
|
version: int
|
|
usage_count: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|