Kyle Pope
0e94b6e1f7
- C-01: Wrap connection request flush in IntegrityError handler for TOCTOU race on partial unique index - W-02: Extract ntfy config into plain dict before commit to avoid DetachedInstanceError in background tasks - W-04: Add integer range validation (1–2147483647) on notification IDs - W-07: Add typed response models for respond_to_request endpoint - W-09: Document resolved_at requirement for future cancel endpoint - S-02: Use Literal type for ConnectionRequestResponse.status - S-04: Check ntfy master switch in extract_ntfy_config - S-05: Move date import to module level in connection service Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
"""
|
|
Connection schemas — search, request, respond, connection management.
|
|
All input schemas use extra="forbid" to prevent mass-assignment.
|
|
"""
|
|
import re
|
|
from typing import Literal, Optional
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
|
|
|
_UMBRAL_NAME_RE = re.compile(r'^[a-zA-Z0-9_-]{3,50}$')
|
|
|
|
|
|
class UmbralSearchRequest(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
umbral_name: str = Field(..., max_length=50)
|
|
|
|
@field_validator('umbral_name')
|
|
@classmethod
|
|
def validate_umbral_name(cls, v: str) -> str:
|
|
if not _UMBRAL_NAME_RE.match(v):
|
|
raise ValueError('Umbral name must be 3-50 alphanumeric characters, hyphens, or underscores')
|
|
return v
|
|
|
|
|
|
class UmbralSearchResponse(BaseModel):
|
|
found: bool
|
|
|
|
|
|
class SendConnectionRequest(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
umbral_name: str = Field(..., max_length=50)
|
|
|
|
@field_validator('umbral_name')
|
|
@classmethod
|
|
def validate_umbral_name(cls, v: str) -> str:
|
|
if not _UMBRAL_NAME_RE.match(v):
|
|
raise ValueError('Umbral name must be 3-50 alphanumeric characters, hyphens, or underscores')
|
|
return v
|
|
|
|
|
|
class ConnectionRequestResponse(BaseModel):
|
|
id: int
|
|
sender_umbral_name: str
|
|
sender_preferred_name: Optional[str] = None
|
|
receiver_umbral_name: str
|
|
receiver_preferred_name: Optional[str] = None
|
|
status: Literal["pending", "accepted", "rejected", "cancelled"]
|
|
created_at: datetime
|
|
|
|
|
|
class RespondRequest(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
action: Literal["accept", "reject"]
|
|
|
|
|
|
class ConnectionResponse(BaseModel):
|
|
id: int
|
|
connected_user_id: int
|
|
connected_umbral_name: str
|
|
connected_preferred_name: Optional[str] = None
|
|
person_id: Optional[int] = None
|
|
created_at: datetime
|
|
|
|
|
|
class RespondAcceptResponse(BaseModel):
|
|
message: str
|
|
connection_id: int
|
|
|
|
|
|
class RespondRejectResponse(BaseModel):
|
|
message: str
|
|
|
|
|
|
class SharingOverrideUpdate(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
preferred_name: Optional[bool] = None
|
|
email: Optional[bool] = None
|
|
phone: Optional[bool] = None
|
|
mobile: Optional[bool] = None
|
|
birthday: Optional[bool] = None
|
|
address: Optional[bool] = None
|
|
company: Optional[bool] = None
|
|
job_title: Optional[bool] = None
|