Initial commit

backend/app/models/analytics.py

@@ -0,0 +1,58 @@
+"""Pydantic response schemas for analytics endpoints."""
+
+from datetime import date
+from uuid import UUID
+
+from pydantic import BaseModel
+
+from app.services.analytics_service import Age4Projection, WorkspaceOverview
+
+
+class Age4ProjectionResponse(BaseModel):
+    personal_data_uuid: UUID
+    child_id: UUID
+    birth_date: date
+    fourth_birthday: date
+    total_days: int
+    days_used: int
+    planned_days_before_b4: int
+    projected_left_at_b4: int
+    savable_after_b4: int
+    expiring_at_b4: int
+
+    @classmethod
+    def from_entity(cls, p: Age4Projection) -> "Age4ProjectionResponse":
+        return cls(
+            personal_data_uuid=p.personal_data_uuid,
+            child_id=p.child_id,
+            birth_date=p.birth_date,
+            fourth_birthday=p.fourth_birthday,
+            total_days=p.total_days,
+            days_used=p.days_used,
+            planned_days_before_b4=p.planned_days_before_b4,
+            projected_left_at_b4=p.projected_left_at_b4,
+            savable_after_b4=p.savable_after_b4,
+            expiring_at_b4=p.expiring_at_b4,
+        )
+
+
+class WorkspaceOverviewResponse(BaseModel):
+    workspace_id: UUID
+    child_id: UUID
+    parent_projections: list[Age4ProjectionResponse]
+    combined_days_used: int
+    combined_days_left: int
+    combined_expiring_at_b4: int
+
+    @classmethod
+    def from_entity(cls, overview: WorkspaceOverview) -> "WorkspaceOverviewResponse":
+        return cls(
+            workspace_id=overview.workspace_id,
+            child_id=overview.child_id,
+            parent_projections=[
+                Age4ProjectionResponse.from_entity(p) for p in overview.parent_projections
+            ],
+            combined_days_used=overview.combined_days_used,
+            combined_days_left=overview.combined_days_left,
+            combined_expiring_at_b4=overview.combined_expiring_at_b4,
+        )