ffb14cd014
Implement three foundational schema features from the design doc: - Tree forking with lineage tracking (migration 022): parent_tree_id, root_tree_id, fork_depth columns with self-referential FKs and composite analytics index - Custom step enhancement: CustomStepSchema with source tracking (ad-hoc, step-library, forked-tree) for backward-compatible JSONB - Session sharing (migration 023): session_shares and session_share_views tables with account-scoped visibility, cryptographic tokens, view tracking, and allow_public_shares account policy Includes 21 new integration tests (9 forking, 12 sharing), SaaS consultant-recommended denormalizations, rate limiting on public share access, and test fixture fix for invite code requirement. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
from datetime import datetime
|
|
from typing import Optional, Any
|
|
from uuid import UUID
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class CategoryInfo(BaseModel):
|
|
"""Embedded category info for tree responses."""
|
|
id: UUID
|
|
name: str
|
|
slug: str
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class TreeBase(BaseModel):
|
|
name: str = Field(..., min_length=1, max_length=255)
|
|
description: Optional[str] = None
|
|
# Legacy category field - kept for backward compatibility
|
|
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)")
|
|
category_id: Optional[UUID] = Field(None, description="Category ID from tree_categories table")
|
|
tags: Optional[list[str]] = Field(None, max_length=10, description="List of tag names to assign")
|
|
|
|
|
|
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)
|
|
category_id: Optional[UUID] = None
|
|
tree_structure: Optional[dict[str, Any]] = None
|
|
is_public: Optional[bool] = None
|
|
is_active: Optional[bool] = None
|
|
tags: Optional[list[str]] = Field(None, max_length=10, description="List of tag names to assign (replaces existing)")
|
|
|
|
|
|
class ForkCreate(BaseModel):
|
|
fork_reason: Optional[str] = Field(None, max_length=255, description="Brief reason for forking")
|
|
name: Optional[str] = Field(None, min_length=1, max_length=255, description="Name for the fork (defaults to 'Fork of {original name}')")
|
|
|
|
|
|
class ForkInfo(BaseModel):
|
|
"""Fork metadata included in tree responses."""
|
|
parent_tree_id: Optional[UUID] = None
|
|
root_tree_id: Optional[UUID] = None
|
|
fork_reason: Optional[str] = None
|
|
fork_depth: int = 0
|
|
parent_updated_at: Optional[datetime] = None
|
|
has_parent_updates: bool = False
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class TreeResponse(TreeBase):
|
|
id: UUID
|
|
tree_structure: dict[str, Any]
|
|
author_id: Optional[UUID] = None
|
|
account_id: Optional[UUID] = None
|
|
category_id: Optional[UUID] = None
|
|
category_info: Optional[CategoryInfo] = None
|
|
tags: list[str] = [] # List of tag names
|
|
fork_info: Optional[ForkInfo] = 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
|
|
category_id: Optional[UUID] = None
|
|
category_info: Optional[CategoryInfo] = None
|
|
tags: list[str] = [] # List of tag names
|
|
author_id: Optional[UUID] = None
|
|
account_id: Optional[UUID] = 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
|