151 lines
3.4 KiB
Python
151 lines
3.4 KiB
Python
"""Pydantic schemas for API request/response models."""
|
|
from datetime import datetime
|
|
from typing import Generic, TypeVar
|
|
|
|
from pydantic import BaseModel, EmailStr
|
|
|
|
|
|
class UserCredentials(BaseModel):
|
|
"""Base model for user email/password."""
|
|
email: EmailStr
|
|
password: str
|
|
|
|
|
|
UserCreate = UserCredentials
|
|
UserLogin = UserCredentials
|
|
|
|
|
|
class UserResponse(BaseModel):
|
|
"""Response model for authenticated user info."""
|
|
id: int
|
|
email: str
|
|
roles: list[str]
|
|
permissions: list[str]
|
|
|
|
|
|
class RegisterWithInvite(BaseModel):
|
|
"""Request model for registration with invite."""
|
|
email: EmailStr
|
|
password: str
|
|
invite_identifier: str
|
|
|
|
|
|
class SumRequest(BaseModel):
|
|
"""Request model for sum calculation."""
|
|
a: float
|
|
b: float
|
|
|
|
|
|
class SumResponse(BaseModel):
|
|
"""Response model for sum calculation."""
|
|
a: float
|
|
b: float
|
|
result: float
|
|
|
|
|
|
class CounterRecordResponse(BaseModel):
|
|
"""Response model for a counter audit record."""
|
|
id: int
|
|
user_email: str
|
|
value_before: int
|
|
value_after: int
|
|
created_at: datetime
|
|
|
|
|
|
class SumRecordResponse(BaseModel):
|
|
"""Response model for a sum audit record."""
|
|
id: int
|
|
user_email: str
|
|
a: float
|
|
b: float
|
|
result: float
|
|
created_at: datetime
|
|
|
|
|
|
RecordT = TypeVar("RecordT", bound=BaseModel)
|
|
|
|
|
|
class PaginatedResponse(BaseModel, Generic[RecordT]):
|
|
"""Generic paginated response wrapper."""
|
|
records: list[RecordT]
|
|
total: int
|
|
page: int
|
|
per_page: int
|
|
total_pages: int
|
|
|
|
|
|
PaginatedCounterRecords = PaginatedResponse[CounterRecordResponse]
|
|
PaginatedSumRecords = PaginatedResponse[SumRecordResponse]
|
|
|
|
|
|
class ProfileResponse(BaseModel):
|
|
"""Response model for profile data."""
|
|
contact_email: str | None
|
|
telegram: str | None
|
|
signal: str | None
|
|
nostr_npub: str | None
|
|
godfather_email: str | None = None
|
|
|
|
|
|
class ProfileUpdate(BaseModel):
|
|
"""Request model for updating profile."""
|
|
contact_email: str | None = None
|
|
telegram: str | None = None
|
|
signal: str | None = None
|
|
nostr_npub: str | None = None
|
|
|
|
|
|
class InviteCheckResponse(BaseModel):
|
|
"""Response for invite check endpoint."""
|
|
valid: bool
|
|
status: str | None = None
|
|
error: str | None = None
|
|
|
|
|
|
class InviteCreate(BaseModel):
|
|
"""Request model for creating an invite."""
|
|
godfather_id: int
|
|
|
|
|
|
class InviteResponse(BaseModel):
|
|
"""Response model for invite data (admin view)."""
|
|
id: int
|
|
identifier: str
|
|
godfather_id: int
|
|
godfather_email: str
|
|
status: str
|
|
used_by_id: int | None
|
|
used_by_email: str | None
|
|
created_at: datetime
|
|
spent_at: datetime | None
|
|
revoked_at: datetime | None
|
|
|
|
|
|
class UserInviteResponse(BaseModel):
|
|
"""Response model for a user's invite (simpler than admin view)."""
|
|
id: int
|
|
identifier: str
|
|
status: str
|
|
used_by_email: str | None
|
|
created_at: datetime
|
|
spent_at: datetime | None
|
|
|
|
|
|
PaginatedInviteRecords = PaginatedResponse[InviteResponse]
|
|
|
|
|
|
class AdminUserResponse(BaseModel):
|
|
"""Minimal user info for admin dropdowns."""
|
|
id: int
|
|
email: str
|
|
|
|
|
|
# =============================================================================
|
|
# Meta/Constants Schemas
|
|
# =============================================================================
|
|
|
|
class ConstantsResponse(BaseModel):
|
|
"""Response model for shared constants."""
|
|
permissions: list[str]
|
|
roles: list[str]
|
|
invite_statuses: list[str]
|