"""Pydantic schemas for User management endpoints.""" import re import uuid from datetime import datetime from pydantic import BaseModel, ConfigDict, EmailStr, field_validator # ── Password policy ───────────────────────────────────────────────── _MIN_PASSWORD_LENGTH = 12 _PASSWORD_RULES: list[tuple[str, str]] = [ (r"[A-Z]", "at least one uppercase letter"), (r"[a-z]", "at least one lowercase letter"), (r"[0-9]", "at least one digit"), (r"[!@#$%^&*()_+\-=\[\]{};':\"\\|,.<>/?`~]", "at least one special character"), ] def _validate_password_strength(password: str) -> str: """Check that *password* satisfies the complexity policy. Rules: - Minimum 12 characters - At least one uppercase letter - At least one lowercase letter - At least one digit - At least one special character """ errors: list[str] = [] if len(password) < _MIN_PASSWORD_LENGTH: errors.append(f"must be at least {_MIN_PASSWORD_LENGTH} characters long") for pattern, description in _PASSWORD_RULES: if not re.search(pattern, password): errors.append(description) if errors: raise ValueError( "Password does not meet complexity requirements: " + "; ".join(errors) ) return password # ── Create ────────────────────────────────────────────────────────── class UserCreate(BaseModel): """Payload for creating a new user.""" username: str email: str | None = None password: str role: str = "viewer" @field_validator("password") @classmethod def password_strength(cls, v: str) -> str: return _validate_password_strength(v) # ── Update ────────────────────────────────────────────────────────── class UserUpdate(BaseModel): """Payload for partially updating an existing user. Every field is optional so callers send only what changed.""" email: str | None = None role: str | None = None is_active: bool | None = None password: str | None = None @field_validator("password") @classmethod def password_strength(cls, v: str | None) -> str | None: if v is not None: return _validate_password_strength(v) return v # ── Read (full) ───────────────────────────────────────────────────── class UserOut(BaseModel): """Complete representation returned by the API.""" id: uuid.UUID username: str email: str | None = None role: str is_active: bool created_at: datetime | None = None last_login: datetime | None = None model_config = ConfigDict(from_attributes=True)