Cheng Zhou
73 lines
1.8 KiB
Python
73 lines
1.8 KiB
Python
import enum
|
|
import uuid
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from app.schemas.document import DocumentScopeType, DocumentSummary
|
|
|
|
|
|
class EtmfNodeStatus(str, enum.Enum):
|
|
NOT_REQUIRED = "NOT_REQUIRED"
|
|
MISSING = "MISSING"
|
|
UPLOADED = "UPLOADED"
|
|
EFFECTIVE = "EFFECTIVE"
|
|
INACTIVE = "INACTIVE"
|
|
|
|
|
|
class EtmfNodeBase(BaseModel):
|
|
parent_id: Optional[uuid.UUID] = None
|
|
code: str
|
|
name: str
|
|
description: Optional[str] = None
|
|
scope_type: DocumentScopeType = DocumentScopeType.GLOBAL
|
|
required: bool = False
|
|
expected_doc_type: Optional[str] = None
|
|
sort_order: int = 0
|
|
is_active: bool = True
|
|
|
|
|
|
class EtmfNodeCreate(EtmfNodeBase):
|
|
study_id: uuid.UUID
|
|
|
|
|
|
class EtmfNodeUpdate(BaseModel):
|
|
parent_id: Optional[uuid.UUID] = None
|
|
code: Optional[str] = None
|
|
name: Optional[str] = None
|
|
description: Optional[str] = None
|
|
scope_type: Optional[DocumentScopeType] = None
|
|
required: Optional[bool] = None
|
|
expected_doc_type: Optional[str] = None
|
|
sort_order: Optional[int] = None
|
|
is_active: Optional[bool] = None
|
|
|
|
|
|
class EtmfNodeRead(EtmfNodeBase):
|
|
id: uuid.UUID
|
|
study_id: uuid.UUID
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
class EtmfNodeStatusSummary(BaseModel):
|
|
status: EtmfNodeStatus
|
|
document_count: int = 0
|
|
effective_document_count: int = 0
|
|
missing_count: int = 0
|
|
|
|
|
|
class EtmfTreeNode(EtmfNodeRead):
|
|
status: EtmfNodeStatus = EtmfNodeStatus.NOT_REQUIRED
|
|
document_count: int = 0
|
|
effective_document_count: int = 0
|
|
children: list["EtmfTreeNode"] = Field(default_factory=list)
|
|
|
|
|
|
class EtmfNodeDocuments(BaseModel):
|
|
node: EtmfNodeRead
|
|
documents: list[DocumentSummary] = Field(default_factory=list)
|