157 lines
3.8 KiB
Python
157 lines
3.8 KiB
Python
from datetime import date, timedelta, datetime, time
|
||
from typing import Annotated
|
||
|
||
from pydantic_core import PydanticCustomError
|
||
from pydantic import BaseModel, EmailStr, field_validator, HttpUrl, StringConstraints
|
||
from fastapi import Query
|
||
|
||
|
||
class SUserLogin(BaseModel):
|
||
email_or_username: EmailStr | str
|
||
password: str
|
||
|
||
|
||
class SUserRegister(BaseModel):
|
||
email: EmailStr
|
||
username: str = Query(None, min_length=2, max_length=30)
|
||
password: str = Query(None, min_length=8)
|
||
password2: str = Query(None, min_length=8)
|
||
date_of_birth: date
|
||
|
||
@field_validator("date_of_birth") # noqa
|
||
@classmethod
|
||
def validate_date_of_birth(cls, v):
|
||
if date.today() - v < timedelta(days=365 * 16):
|
||
date_of_birth = date.today() - timedelta(days=365 * 16)
|
||
raise PydanticCustomError(
|
||
"date_input_error", "date of birth might be earlier than {date_of_birth}", {"date_of_birth": date_of_birth}
|
||
)
|
||
elif datetime.combine(v, time.min).timestamp() < datetime(year=1924, month=1, day=1).timestamp():
|
||
date_of_birth = date(1924, 1, 1)
|
||
raise PydanticCustomError(
|
||
"date_input_error", "date of birth might be later than {date_of_birth}", {"date_of_birth": date_of_birth}
|
||
)
|
||
return v
|
||
|
||
|
||
class SUserResponse(BaseModel):
|
||
id: int
|
||
username: str
|
||
email: EmailStr
|
||
black_phoenix: bool
|
||
avatar_image: HttpUrl
|
||
date_of_birth: date
|
||
date_of_registration: date
|
||
|
||
|
||
class SUsers(BaseModel):
|
||
users: list[SUserResponse] | None
|
||
|
||
|
||
class SUser(BaseModel):
|
||
id: int
|
||
email: str
|
||
username: str
|
||
hashed_password: str
|
||
role: int
|
||
black_phoenix: bool
|
||
avatar_image: HttpUrl
|
||
date_of_birth: date
|
||
date_of_registration: date
|
||
|
||
|
||
class SUserSendConfirmationCode(BaseModel):
|
||
email: EmailStr
|
||
|
||
|
||
class SUserChangeData(BaseModel):
|
||
verification_code: str
|
||
email: EmailStr
|
||
username: str = Query(None, min_length=2, max_length=30)
|
||
new_password: str | None = Query(None, min_length=8)
|
||
avatar_url: HttpUrl | None
|
||
|
||
|
||
class STokenLogin(BaseModel):
|
||
authorization: str
|
||
|
||
|
||
class SUserCode(BaseModel):
|
||
user_code: str
|
||
|
||
|
||
class SCreateInvitationLink(BaseModel):
|
||
invitation_link: HttpUrl
|
||
|
||
|
||
class SUserAvatarUrl(BaseModel):
|
||
avatar_url: HttpUrl
|
||
|
||
|
||
class SUserAvatars(BaseModel):
|
||
user_avatars: list[SUserAvatarUrl] | None
|
||
|
||
|
||
class SUserFilter(BaseModel):
|
||
username: Annotated[str, StringConstraints(min_length=2, max_length=30)] | None = None
|
||
email: EmailStr | None = None
|
||
|
||
|
||
class SUserPassword(BaseModel):
|
||
user_password: str = Query(None, min_length=8)
|
||
|
||
|
||
class SConfirmationData(BaseModel):
|
||
user_id: int
|
||
username: str
|
||
email_to: EmailStr
|
||
confirmation_code: str
|
||
type: str
|
||
|
||
|
||
class SInvitationData(BaseModel):
|
||
chat_id: int
|
||
|
||
|
||
class SGetUsersFilter(BaseModel):
|
||
username: str | None = None
|
||
|
||
@field_validator("username") # noqa
|
||
@classmethod
|
||
def validate_username(cls, v):
|
||
if not v:
|
||
return "%%"
|
||
else:
|
||
return f"%{v}%"
|
||
|
||
|
||
class Responses:
|
||
class STokenMissingException(BaseModel):
|
||
detail: str = "Токен отсутствует"
|
||
|
||
class SIncorrectAuthDataException(BaseModel):
|
||
detail: str = "Введены не верные данные"
|
||
|
||
class SNotAuthenticated(BaseModel):
|
||
detail: str = "Not authenticated"
|
||
|
||
class SUserNotFoundException(BaseModel):
|
||
details: str = "Юзер не найден"
|
||
|
||
class SUserAlreadyExistsException(BaseModel):
|
||
detail: str = "Пользователь с таким ником или почтой уже существует"
|
||
|
||
class SPasswordAlreadyInUseException(BaseModel):
|
||
detail: str = "Пароль уже занят"
|
||
|
||
class SPasswordsMismatchException(BaseModel):
|
||
detail: str = "Пароли не совпадают"
|
||
|
||
class SIncorrectDataException(BaseModel):
|
||
detail: str = "Ты передал какую-то хуйню"
|
||
|
||
class SWrongCodeException(BaseModel):
|
||
detail: str = "Введён не верный код подтверждения"
|
||
|
||
class SBlackPhoenixException(BaseModel):
|
||
detail: str = "Внутренняя ошибка сервера"
|