Andrew K. Choi
dd7349bb4c
All checks were successful
continuous-integration/drone/push Build is passing
106 lines
3.3 KiB
Python
106 lines
3.3 KiB
Python
from datetime import date
|
|
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
from pydantic import BaseModel, EmailStr, Field, field_validator
|
|
|
|
|
|
class UserBase(BaseModel):
|
|
email: EmailStr
|
|
username: Optional[str] = None
|
|
phone: Optional[str] = None
|
|
phone_number: Optional[str] = None # Альтернативное поле для phone
|
|
first_name: Optional[str] = ""
|
|
last_name: Optional[str] = ""
|
|
full_name: Optional[str] = None # Будет разделено на first_name и last_name
|
|
date_of_birth: Optional[date] = None
|
|
bio: Optional[str] = Field(None, max_length=500)
|
|
|
|
@field_validator("full_name")
|
|
@classmethod
|
|
def split_full_name(cls, v, info):
|
|
"""Разделяет полное имя на first_name и last_name."""
|
|
if v:
|
|
values = info.data
|
|
parts = v.split(" ", 1)
|
|
if "first_name" in values and not values["first_name"]:
|
|
info.data["first_name"] = parts[0]
|
|
if "last_name" in values and not values["last_name"]:
|
|
info.data["last_name"] = parts[1] if len(parts) > 1 else ""
|
|
return v
|
|
|
|
@field_validator("phone_number")
|
|
@classmethod
|
|
def map_phone_number(cls, v, info):
|
|
"""Копирует phone_number в phone если phone не указан."""
|
|
if v:
|
|
values = info.data
|
|
if "phone" in values and not values["phone"]:
|
|
info.data["phone"] = v
|
|
return v
|
|
|
|
|
|
class UserCreate(UserBase):
|
|
password: str = Field(..., min_length=8, max_length=100)
|
|
|
|
|
|
class UserUpdate(BaseModel):
|
|
first_name: Optional[str] = Field(None, min_length=1, max_length=50)
|
|
last_name: Optional[str] = Field(None, min_length=1, max_length=50)
|
|
phone: Optional[str] = None
|
|
date_of_birth: Optional[date] = None
|
|
bio: Optional[str] = Field(None, max_length=500)
|
|
avatar_url: Optional[str] = None
|
|
|
|
# Emergency contacts
|
|
emergency_contact_1_name: Optional[str] = Field(None, max_length=100)
|
|
emergency_contact_1_phone: Optional[str] = Field(None, max_length=20)
|
|
emergency_contact_2_name: Optional[str] = Field(None, max_length=100)
|
|
emergency_contact_2_phone: Optional[str] = Field(None, max_length=20)
|
|
|
|
# Settings
|
|
location_sharing_enabled: Optional[bool] = None
|
|
emergency_notifications_enabled: Optional[bool] = None
|
|
push_notifications_enabled: Optional[bool] = None
|
|
|
|
|
|
class UserResponse(UserBase):
|
|
id: int
|
|
uuid: str
|
|
avatar_url: Optional[str] = None
|
|
emergency_contact_1_name: Optional[str] = None
|
|
emergency_contact_1_phone: Optional[str] = None
|
|
emergency_contact_2_name: Optional[str] = None
|
|
emergency_contact_2_phone: Optional[str] = None
|
|
location_sharing_enabled: bool
|
|
emergency_notifications_enabled: bool
|
|
push_notifications_enabled: bool
|
|
email_verified: bool
|
|
phone_verified: bool
|
|
is_active: bool
|
|
|
|
@field_validator("uuid", mode="before")
|
|
@classmethod
|
|
def convert_uuid_to_str(cls, v):
|
|
if isinstance(v, UUID):
|
|
return str(v)
|
|
return v
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class UserLogin(BaseModel):
|
|
email: Optional[EmailStr] = None
|
|
username: Optional[str] = None
|
|
password: str
|
|
|
|
|
|
class Token(BaseModel):
|
|
access_token: str
|
|
token_type: str
|
|
|
|
|
|
class TokenData(BaseModel):
|
|
email: Optional[str] = None
|