monkeybrain-runtime 1.0.0__py3-none-any.whl

services/workorders/models/ipc_result_records.py

@@ -0,0 +1,164 @@
+from datetime import datetime, timezone
+from typing import Any, Literal, Optional
+
+from pydantic import BaseModel, Field, model_validator
+
+
+IpcResultStatus = Literal["Pending", "Pass", "Fail", "Retest Required", "Not Applicable"]
+IpcReviewStatus = Literal["Pending", "Reviewed", "Approved", "Rejected"]
+
+
+def utc_now() -> datetime:
+    return datetime.now(timezone.utc)
+
+
+def ensure_utc(value: Optional[datetime]) -> Optional[datetime]:
+    if value is None:
+        return None
+    if value.tzinfo is None:
+        return value.replace(tzinfo=timezone.utc)
+    return value.astimezone(timezone.utc)
+
+
+class IpcLimitRecord(BaseModel):
+    parameter: str = Field(..., min_length=1)
+    target: Optional[float | str] = None
+    lower_limit: Optional[float] = None
+    upper_limit: Optional[float] = None
+    unit: Optional[str] = None
+    acceptance_criteria: Optional[str] = None
+    cqa_ids: list[str] = Field(default_factory=list)
+    cpp_ids: list[str] = Field(default_factory=list)
+
+
+class IpcMeasuredResult(BaseModel):
+    parameter: str = Field(..., min_length=1)
+    measured_value: Optional[float] = None
+    text_value: Optional[str] = None
+    unit: Optional[str] = None
+    result: IpcResultStatus = "Pending"
+
+    @model_validator(mode="after")
+    def validate_value(self) -> "IpcMeasuredResult":
+        if self.result not in {"Pending", "Not Applicable"} and self.measured_value is None and not self.text_value:
+            raise ValueError("non-pending IPC measured results must include measured_value or text_value.")
+        return self
+
+
+class IpcResultRecord(BaseModel):
+    ipc_result_id: str = Field(..., min_length=1)
+    ipc_checkpoint_id: str = Field(..., min_length=1)
+    batch_execution_record_id: str = Field(..., min_length=1)
+    batch_step_execution_id: Optional[str] = None
+    batch_id: str = Field(..., min_length=1)
+
+    process_definition_id: str = Field(..., min_length=1)
+    process_step_id: str = Field(..., min_length=1)
+    stage_id: Optional[str] = None
+    workstation_id: Optional[str] = None
+    product_id: Optional[str] = None
+    bom_id: Optional[str] = None
+
+    method: str = Field(..., min_length=1)
+    method_reference: str = Field(..., min_length=1)
+    sampling_method: Optional[str] = None
+    sample_id: Optional[str] = None
+    sample_location: Optional[str] = None
+    sampled_by: Optional[str] = None
+    sampled_at: Optional[datetime] = None
+    tested_by: str = Field(..., min_length=1)
+    tested_at: datetime
+
+    limits_snapshot: list[IpcLimitRecord] = Field(default_factory=list)
+    measured_results: list[IpcMeasuredResult] = Field(default_factory=list)
+    status: IpcResultStatus = "Pending"
+    review_status: IpcReviewStatus = "Pending"
+    reviewed_by: Optional[str] = None
+    reviewed_at: Optional[datetime] = None
+    approved_by: Optional[str] = None
+    approved_at: Optional[datetime] = None
+
+    deviation_id: Optional[str] = None
+    evidence_document_ids: list[str] = Field(default_factory=list)
+    signature_ids: list[str] = Field(default_factory=list)
+    audit_event_ids: list[str] = Field(default_factory=list)
+    metadata: dict[str, Any] = Field(default_factory=dict)
+
+    created_by: Optional[str] = None
+    created_at: datetime = Field(default_factory=utc_now)
+    updated_at: datetime = Field(default_factory=utc_now)
+
+    @model_validator(mode="after")
+    def normalize_and_validate(self) -> "IpcResultRecord":
+        self.sampled_at = ensure_utc(self.sampled_at)
+        self.tested_at = ensure_utc(self.tested_at) or utc_now()
+        self.reviewed_at = ensure_utc(self.reviewed_at)
+        self.approved_at = ensure_utc(self.approved_at)
+        self.created_at = ensure_utc(self.created_at) or utc_now()
+        self.updated_at = ensure_utc(self.updated_at) or utc_now()
+        if self.sampled_at and self.tested_at and self.tested_at < self.sampled_at:
+            raise ValueError("tested_at cannot be before sampled_at for IPC result records.")
+        if self.status in {"Pass", "Fail", "Retest Required"}:
+            if not self.limits_snapshot:
+                raise ValueError("completed IPC result records must include limits_snapshot.")
+            if not self.measured_results:
+                raise ValueError("completed IPC result records must include measured_results.")
+            if not self.evidence_document_ids:
+                raise ValueError("completed IPC result records must include evidence_document_ids.")
+            if not self.signature_ids:
+                raise ValueError("completed IPC result records must include signature_ids.")
+        if self.status == "Fail" and not self.deviation_id:
+            raise ValueError("failed IPC result records must include deviation_id.")
+        if self.review_status in {"Reviewed", "Approved"} and (not self.reviewed_by or not self.reviewed_at):
+            raise ValueError("reviewed/approved IPC result records must include reviewed_by and reviewed_at.")
+        if self.review_status == "Approved" and (not self.approved_by or not self.approved_at):
+            raise ValueError("approved IPC result records must include approved_by and approved_at.")
+        return self
+
+
+class IpcResultRecordCreate(IpcResultRecord):
+    pass
+
+
+class IpcResultRecordUpdate(BaseModel):
+    sample_id: Optional[str] = None
+    sample_location: Optional[str] = None
+    sampled_by: Optional[str] = None
+    sampled_at: Optional[datetime] = None
+    tested_by: Optional[str] = None
+    tested_at: Optional[datetime] = None
+    limits_snapshot: Optional[list[IpcLimitRecord]] = None
+    measured_results: Optional[list[IpcMeasuredResult]] = None
+    status: Optional[IpcResultStatus] = None
+    review_status: Optional[IpcReviewStatus] = None
+    reviewed_by: Optional[str] = None
+    reviewed_at: Optional[datetime] = None
+    approved_by: Optional[str] = None
+    approved_at: Optional[datetime] = None
+    deviation_id: Optional[str] = None
+    evidence_document_ids: Optional[list[str]] = None
+    signature_ids: Optional[list[str]] = None
+    audit_event_ids: Optional[list[str]] = None
+    metadata: Optional[dict[str, Any]] = None
+    updated_at: datetime = Field(default_factory=utc_now)
+
+    @model_validator(mode="after")
+    def normalize_datetimes(self) -> "IpcResultRecordUpdate":
+        self.sampled_at = ensure_utc(self.sampled_at)
+        self.tested_at = ensure_utc(self.tested_at)
+        self.reviewed_at = ensure_utc(self.reviewed_at)
+        self.approved_at = ensure_utc(self.approved_at)
+        self.updated_at = ensure_utc(self.updated_at) or utc_now()
+        return self
+
+
+class IpcResultRecordResponse(IpcResultRecord):
+    class Config:
+        from_attributes = True
+
+
+class PaginatedIpcResultRecordResponse(BaseModel):
+    total: int
+    page: int
+    page_size: int
+    results: list[IpcResultRecordResponse]

services/workorders/models/production_batches.py

@@ -0,0 +1,181 @@
+from datetime import datetime, timezone
+from typing import Any, Literal, Optional
+
+from pydantic import BaseModel, Field, model_validator
+
+
+ProductionBatchStatus = Literal[
+    "Planned",
+    "Created",
+    "Released",
+    "Dispensing",
+    "In Production",
+    "On Hold",
+    "Completed",
+    "Under QA Review",
+    "Approved",
+    "Rejected",
+    "Cancelled",
+    "Closed",
+]
+
+BatchLifecycleEventType = Literal[
+    "Created",
+    "Released",
+    "Started",
+    "Step Started",
+    "Step Completed",
+    "Hold Placed",
+    "Hold Released",
+    "Completed",
+    "QA Review Started",
+    "Approved",
+    "Rejected",
+    "Closed",
+    "Cancelled",
+]
+
+
+def utc_now() -> datetime:
+    return datetime.now(timezone.utc)
+
+
+def ensure_utc(value: Optional[datetime]) -> Optional[datetime]:
+    if value is None:
+        return None
+    if value.tzinfo is None:
+        return value.replace(tzinfo=timezone.utc)
+    return value.astimezone(timezone.utc)
+
+
+class BatchLifecycleEvent(BaseModel):
+    event_id: str = Field(..., min_length=1)
+    event_type: BatchLifecycleEventType
+    status_from: Optional[ProductionBatchStatus] = None
+    status_to: ProductionBatchStatus
+    occurred_at: datetime = Field(default_factory=utc_now)
+    performed_by: Optional[str] = None
+    reason: Optional[str] = None
+    batch_execution_record_id: Optional[str] = None
+    process_step_id: Optional[str] = None
+    evidence_document_ids: list[str] = Field(default_factory=list)
+    signature_ids: list[str] = Field(default_factory=list)
+    metadata: dict[str, Any] = Field(default_factory=dict)
+
+    @model_validator(mode="after")
+    def normalize_datetimes(self) -> "BatchLifecycleEvent":
+        self.occurred_at = ensure_utc(self.occurred_at) or utc_now()
+        return self
+
+
+class ProductionBatch(BaseModel):
+    batch_id: str = Field(..., min_length=1)
+    batch_number: str = Field(..., min_length=1)
+    product_id: str = Field(..., min_length=1)
+    bom_id: Optional[str] = None
+    process_definition_id: Optional[str] = None
+    process_route_id: Optional[str] = None
+    work_order_id: Optional[str] = None
+
+    plant_id: Optional[str] = None
+    line_id: Optional[str] = None
+    current_stage_id: Optional[str] = None
+    current_workstation_id: Optional[str] = None
+    lot_ids: list[str] = Field(default_factory=list)
+
+    status: ProductionBatchStatus = "Created"
+    lifecycle_events: list[BatchLifecycleEvent] = Field(default_factory=list)
+    batch_execution_record_ids: list[str] = Field(default_factory=list)
+    active_batch_execution_record_id: Optional[str] = None
+
+    planned_start_at: Optional[datetime] = None
+    planned_end_at: Optional[datetime] = None
+    released_at: Optional[datetime] = None
+    started_at: Optional[datetime] = None
+    completed_at: Optional[datetime] = None
+    closed_at: Optional[datetime] = None
+
+    released_by: Optional[str] = None
+    qa_reviewer_id: Optional[str] = None
+    created_by: Optional[str] = None
+    metadata: dict[str, Any] = Field(default_factory=dict)
+    created_at: datetime = Field(default_factory=utc_now)
+    updated_at: datetime = Field(default_factory=utc_now)
+
+    @model_validator(mode="after")
+    def normalize_and_validate(self) -> "ProductionBatch":
+        self.planned_start_at = ensure_utc(self.planned_start_at)
+        self.planned_end_at = ensure_utc(self.planned_end_at)
+        self.released_at = ensure_utc(self.released_at)
+        self.started_at = ensure_utc(self.started_at)
+        self.completed_at = ensure_utc(self.completed_at)
+        self.closed_at = ensure_utc(self.closed_at)
+        self.created_at = ensure_utc(self.created_at) or utc_now()
+        self.updated_at = ensure_utc(self.updated_at) or utc_now()
+
+        if self.planned_start_at and self.planned_end_at and self.planned_end_at < self.planned_start_at:
+            raise ValueError("planned_end_at cannot be before planned_start_at.")
+        if self.completed_at and self.started_at and self.completed_at < self.started_at:
+            raise ValueError("completed_at cannot be before started_at.")
+        if self.closed_at and self.completed_at and self.closed_at < self.completed_at:
+            raise ValueError("closed_at cannot be before completed_at.")
+        if self.status in {"Released", "Dispensing", "In Production", "Completed", "Under QA Review", "Approved", "Closed"} and not self.released_at:
+            raise ValueError("released production batches must include released_at.")
+        if self.status in {"Approved", "Closed"} and not self.qa_reviewer_id:
+            raise ValueError("approved or closed production batches must include qa_reviewer_id.")
+        return self
+
+
+class ProductionBatchCreate(ProductionBatch):
+    pass
+
+
+class ProductionBatchUpdate(BaseModel):
+    batch_number: Optional[str] = None
+    product_id: Optional[str] = None
+    bom_id: Optional[str] = None
+    process_definition_id: Optional[str] = None
+    process_route_id: Optional[str] = None
+    work_order_id: Optional[str] = None
+    plant_id: Optional[str] = None
+    line_id: Optional[str] = None
+    current_stage_id: Optional[str] = None
+    current_workstation_id: Optional[str] = None
+    lot_ids: Optional[list[str]] = None
+    status: Optional[ProductionBatchStatus] = None
+    lifecycle_events: Optional[list[BatchLifecycleEvent]] = None
+    batch_execution_record_ids: Optional[list[str]] = None
+    active_batch_execution_record_id: Optional[str] = None
+    planned_start_at: Optional[datetime] = None
+    planned_end_at: Optional[datetime] = None
+    released_at: Optional[datetime] = None
+    started_at: Optional[datetime] = None
+    completed_at: Optional[datetime] = None
+    closed_at: Optional[datetime] = None
+    released_by: Optional[str] = None
+    qa_reviewer_id: Optional[str] = None
+    metadata: Optional[dict[str, Any]] = None
+    updated_at: datetime = Field(default_factory=utc_now)
+
+    @model_validator(mode="after")
+    def normalize_datetimes(self) -> "ProductionBatchUpdate":
+        self.planned_start_at = ensure_utc(self.planned_start_at)
+        self.planned_end_at = ensure_utc(self.planned_end_at)
+        self.released_at = ensure_utc(self.released_at)
+        self.started_at = ensure_utc(self.started_at)
+        self.completed_at = ensure_utc(self.completed_at)
+        self.closed_at = ensure_utc(self.closed_at)
+        self.updated_at = ensure_utc(self.updated_at) or utc_now()
+        return self
+
+
+class ProductionBatchResponse(ProductionBatch):
+    class Config:
+        from_attributes = True
+
+
+class PaginatedProductionBatchResponse(BaseModel):
+    total: int
+    page: int
+    page_size: int
+    results: list[ProductionBatchResponse]

services/workorders/models/work_orders.py

@@ -0,0 +1,255 @@
+from datetime import datetime, timezone
+from typing import Optional, Literal, List
+from pydantic import BaseModel, Field, model_validator
+
+# ─────────────────────────────────────────────
+# LITERALS
+# ─────────────────────────────────────────────
+
+WorkOrderStatus = Literal[
+    "Todo",
+    "In Progress",
+    "Ready for Next Stage",
+    "Rejected",
+    "Completed",
+    "Done",
+    "Cancelled",
+    "On Hold",
+    "Scheduled",
+]
+
+WorkOrderPriority = Literal[
+    "Low",
+    "Medium",
+    "High",
+    "Critical",
+]
+
+WorkOrderType = Literal[
+    "Maintenance",
+    "Inspection",
+    "Calibration",
+    "Cleaning",
+    "Repair",
+    "Investigation",
+    "Other",
+]
+
+# ─────────────────────────────────────────────
+# UTILS (INLINE SAFE UTC HANDLING)
+# ─────────────────────────────────────────────
+
+def utc_now() -> datetime:
+    return datetime.now(timezone.utc)
+
+
+def ensure_utc(dt: Optional[datetime]) -> Optional[datetime]:
+    if dt is None:
+        return None
+    if dt.tzinfo is None:
+        return dt.replace(tzinfo=timezone.utc)
+    return dt.astimezone(timezone.utc)
+
+# ─────────────────────────────────────────────
+# BASE MODEL
+# ─────────────────────────────────────────────
+
+class WorkOrder(BaseModel):
+    work_order_id: str = Field(..., min_length=1)
+    title: str = Field(..., min_length=1)
+    message: Optional[str] = None
+
+    work_order_type: WorkOrderType = "Other"
+    status: WorkOrderStatus = "Todo"
+    priority: WorkOrderPriority = "Medium"
+
+    # Linkage
+    parent_work_order_id: Optional[str] = None
+    machine_id: Optional[str] = None
+    equipment_id: Optional[str] = None
+    line_id: Optional[str] = None
+    circuit_id: Optional[str] = None
+    workstation_id: Optional[str] = None
+    warehouse_id: Optional[str] = None
+    product_id: Optional[str] = None
+    component_id: Optional[str] = None
+    machine_part_id: Optional[str] = None
+    raw_material_id: Optional[str] = None
+
+    # Assignment
+    assigned_to: Optional[str] = None
+    assigned_user_id: Optional[str] = None
+    assigned_user_name: Optional[str] = None
+    assigned_team_id: Optional[str] = None
+    assigned_team_name: Optional[str] = None
+
+    # Dates
+    created_at: datetime = Field(default_factory=utc_now)
+    expected_completion: Optional[datetime] = None
+    started_at: Optional[datetime] = None
+    completed_at: Optional[datetime] = None
+    total_time_hrs: Optional[int] = None
+    updated_at: datetime = Field(default_factory=utc_now)
+
+    # Computed
+    is_overdue: bool = False
+
+    remarks: Optional[str] = None
+    attachments: List[str] = Field(default_factory=list)
+
+    @model_validator(mode="after")
+    def normalize_and_validate(self):
+        # 1. Normalize all datetimes to UTC-aware
+        self.created_at = ensure_utc(self.created_at)
+        self.updated_at = ensure_utc(self.updated_at)
+        self.expected_completion = ensure_utc(self.expected_completion)
+        self.completed_at = ensure_utc(self.completed_at)
+
+        # 2. Validate business rule: completed_at only if terminal completed
+        if self.completed_at and self.status not in ("Done", "Completed"):
+            raise ValueError("'completed_at' can only be set when status is 'Done' or 'Completed'.")
+
+        # 3. Validate chronological order
+        if self.completed_at and self.completed_at < self.created_at:
+            raise ValueError("'completed_at' cannot be before 'created_at'.")
+
+        # 4. Compute overdue safely
+        now = utc_now()
+
+        if self.status not in ("Done", "Completed", "Cancelled") and self.expected_completion:
+            self.is_overdue = now > self.expected_completion
+        else:
+            self.is_overdue = False
+
+        return self
+
+
+# ─────────────────────────────────────────────
+# CREATE
+# ─────────────────────────────────────────────
+
+class WorkOrderCreate(WorkOrder):
+    pass
+
+
+class WorkOrderSubtaskCreate(BaseModel):
+    work_order_id: str = Field(..., min_length=1)
+    title: str = Field(..., min_length=1)
+    message: Optional[str] = None
+    work_order_type: WorkOrderType = "Other"
+    status: WorkOrderStatus = "Todo"
+    priority: WorkOrderPriority = "Medium"
+    machine_id: Optional[str] = None
+    equipment_id: Optional[str] = None
+    line_id: Optional[str] = None
+    circuit_id: Optional[str] = None
+    workstation_id: Optional[str] = None
+    warehouse_id: Optional[str] = None
+    product_id: Optional[str] = None
+    component_id: Optional[str] = None
+    machine_part_id: Optional[str] = None
+    raw_material_id: Optional[str] = None
+    assigned_to: Optional[str] = None
+    assigned_user_id: Optional[str] = None
+    assigned_user_name: Optional[str] = None
+    assigned_team_id: Optional[str] = None
+    assigned_team_name: Optional[str] = None
+    created_by: Optional[str] = None
+    expected_completion: Optional[datetime] = None
+    remarks: Optional[str] = None
+    attachments: List[str] = Field(default_factory=list)
+    created_at: datetime = Field(default_factory=utc_now)
+    updated_at: datetime = Field(default_factory=utc_now)
+
+    @model_validator(mode="after")
+    def normalize_datetimes(self) -> "WorkOrderSubtaskCreate":
+        self.created_at = ensure_utc(self.created_at)
+        self.updated_at = ensure_utc(self.updated_at)
+        self.expected_completion = ensure_utc(self.expected_completion)
+        return self
+
+
+class WorkOrderSubtaskUpdate(BaseModel):
+    title: Optional[str] = None
+    message: Optional[str] = None
+    work_order_type: Optional[WorkOrderType] = None
+    status: Optional[WorkOrderStatus] = None
+    priority: Optional[WorkOrderPriority] = None
+    machine_id: Optional[str] = None
+    equipment_id: Optional[str] = None
+    assigned_to: Optional[str] = None
+    assigned_team_id: Optional[str] = None
+    assigned_team_name: Optional[str] = None
+    expected_completion: Optional[datetime] = None
+    completed_at: Optional[datetime] = None
+    remarks: Optional[str] = None
+    attachments: Optional[List[str]] = None
+    updated_at: datetime = Field(default_factory=utc_now)
+
+    @model_validator(mode="after")
+    def normalize_datetimes(self) -> "WorkOrderSubtaskUpdate":
+        self.expected_completion = ensure_utc(self.expected_completion)
+        self.completed_at = ensure_utc(self.completed_at)
+        self.updated_at = ensure_utc(self.updated_at)
+        return self
+
+
+# ─────────────────────────────────────────────
+# UPDATE
+# ─────────────────────────────────────────────
+
+class WorkOrderUpdate(BaseModel):
+    title: Optional[str] = None
+    message: Optional[str] = None
+    work_order_type: Optional[WorkOrderType] = None
+    status: Optional[WorkOrderStatus] = None
+    priority: Optional[WorkOrderPriority] = None
+
+    parent_work_order_id: Optional[str] = None
+    machine_id: Optional[str] = None
+    equipment_id: Optional[str] = None
+    line_id: Optional[str] = None
+    circuit_id: Optional[str] = None
+    workstation_id: Optional[str] = None
+    warehouse_id: Optional[str] = None
+    product_id: Optional[str] = None
+    component_id: Optional[str] = None
+    machine_part_id: Optional[str] = None
+    raw_material_id: Optional[str] = None
+
+    assigned_to: Optional[str] = None
+    assigned_user_id: Optional[str] = None
+    assigned_user_name: Optional[str] = None
+    assigned_team_id: Optional[str] = None
+    assigned_team_name: Optional[str] = None
+
+    expected_completion: Optional[datetime] = None
+    completed_at: Optional[datetime] = None
+    remarks: Optional[str] = None
+    attachments: Optional[List[str]] = None
+
+    updated_at: datetime = Field(default_factory=utc_now)
+
+
+# ─────────────────────────────────────────────
+# RESPONSE & PAGINATION
+# ─────────────────────────────────────────────
+
+class WorkOrderResponse(WorkOrder):
+    class Config:
+        from_attributes = True
+
+class PaginatedWorkOrderResponse(BaseModel):
+    total: int
+    page: int
+    page_size: int
+    results: List[WorkOrderResponse]
+
+
+class WorkOrderCountStatsResponse(BaseModel):
+    total: int
+    open: int
+    completed: int
+    overdue: int
+    by_status: dict[str, int]
+    by_priority: dict[str, int]