initial commit

backend/app/schemas.py

@@ -0,0 +1,193 @@
+from __future__ import annotations
+
+from datetime import date, datetime
+from typing import Any, Literal, Optional
+
+from pydantic import BaseModel, Field, field_validator
+
+
+Weekday = Literal[0, 1, 2, 3, 4, 5, 6]  # Mon=0 .. Sun=6
+
+
+class SetupPasswordRequest(BaseModel):
+    password: str = Field(min_length=4, max_length=128)
+
+
+class LoginRequest(BaseModel):
+    password: str
+
+
+class TokenResponse(BaseModel):
+    access_token: str
+    token_type: str = "bearer"
+    needs_setup: bool = False
+    onboarded: bool = False
+
+
+class StatusResponse(BaseModel):
+    has_password: bool
+    onboarded: bool
+    phase: str = "active"
+
+
+class OnboardingRequest(BaseModel):
+    one_rm_kg: float = Field(gt=0)
+    training_weekdays: list[int] = Field(min_length=3, max_length=3)
+    start_date: date
+
+    @field_validator("training_weekdays")
+    @classmethod
+    def validate_days(cls, v: list[int]) -> list[int]:
+        if len(set(v)) != 3:
+            raise ValueError("Нужно выбрать 3 разных дня")
+        for d in v:
+            if d < 0 or d > 6:
+                raise ValueError("День недели должен быть 0–6")
+        sorted_days = sorted(v)
+        # check circular gap of at least 1 day between consecutive selected days
+        for i in range(3):
+            a = sorted_days[i]
+            b = sorted_days[(i + 1) % 3]
+            if i < 2:
+                gap = b - a - 1
+            else:
+                gap = (a + 7) - b - 1
+            if gap < 1:
+                raise ValueError("Между выбранными днями нужен минимум 1 день перерыва")
+        return sorted(v)
+
+
+class ProfileResponse(BaseModel):
+    one_rm_kg: Optional[float]
+    training_weekdays: Optional[list[int]]
+    start_date: Optional[date]
+    onboarded: bool
+    phase: str
+    rest_until: Optional[datetime]
+    probe_available_at: Optional[datetime]
+    cycle_number: int
+    rest_countdown_seconds: Optional[int] = None
+    can_probe: bool = False
+
+
+class SetResponse(BaseModel):
+    id: int
+    set_index: int
+    target_percent: Optional[float]
+    target_reps: Optional[int]
+    target_weight_kg: Optional[float]
+    is_failure: bool
+    completed: bool
+    actual_reps: Optional[int]
+    actual_weight_kg: Optional[float]
+
+
+class ExerciseResponse(BaseModel):
+    id: int
+    order_index: int
+    name: str
+    kind: str
+    prescribed_weight_kg: Optional[float]
+    weight_locked: bool
+    needs_weight_input: bool
+    pending_prompt: Optional[str]
+    weight_step_kg: float = 1.25
+    sets: list[SetResponse]
+
+
+class WorkoutSummary(BaseModel):
+    id: int
+    cycle_number: int
+    week_number: int
+    day_index: int
+    scheduled_date: date
+    status: str
+    exercise_count: int = 0
+
+
+class WorkoutDetail(BaseModel):
+    id: int
+    cycle_number: int
+    week_number: int
+    day_index: int
+    scheduled_date: date
+    status: str
+    started_at: Optional[datetime]
+    paused_at: Optional[datetime]
+    completed_at: Optional[datetime]
+    current_exercise_index: int
+    current_set_index: int
+    rest_started_at: Optional[datetime]
+    rest_elapsed_seconds: int = 0
+    can_start: bool = False
+    can_resume: bool = False
+    pause_expires_at: Optional[datetime] = None
+    exercises: list[ExerciseResponse]
+
+
+class CalendarResponse(BaseModel):
+    year: int
+    month: int
+    workouts: list[WorkoutSummary]
+    profile: ProfileResponse
+
+
+class CompleteSetRequest(BaseModel):
+    actual_reps: Optional[int] = None
+    actual_weight_kg: Optional[float] = None
+
+
+class SetWeightRequest(BaseModel):
+    weight_kg: float = Field(gt=0)
+
+
+class AdjustSetRequest(BaseModel):
+    actual_weight_kg: float = Field(gt=0)
+    actual_reps: int = Field(ge=0)
+
+
+class PromptChoiceRequest(BaseModel):
+    choice: Literal["easy_yes", "easy_no"]
+    weight_kg: Optional[float] = None
+    increase_kg: Optional[float] = None
+    decrease_kg: Optional[float] = None
+
+
+class StartRestRequest(BaseModel):
+    rest_days: int = Field(ge=3, le=5)
+
+
+class ProbeRequest(BaseModel):
+    one_rm_kg: float = Field(gt=0)
+    start_date: date
+    training_weekdays: Optional[list[int]] = None
+
+
+class ExportResponse(BaseModel):
+    version: int = 1
+    exported_at: datetime
+    data: dict[str, Any]
+
+
+class ImportRequest(BaseModel):
+    version: int
+    data: dict[str, Any]
+
+
+class MessageResponse(BaseModel):
+    message: str
+    workout: Optional[WorkoutDetail] = None
+    profile: Optional[ProfileResponse] = None
+
+
+class MoveWorkoutRequest(BaseModel):
+    scheduled_date: date
+
+
+class TrainingDaysRequest(BaseModel):
+    training_weekdays: list[int] = Field(min_length=3, max_length=3)
+
+    @field_validator("training_weekdays")
+    @classmethod
+    def validate_days(cls, v: list[int]) -> list[int]:
+        return OnboardingRequest.validate_days(v)