131 lines
2.9 KiB
Python
131 lines
2.9 KiB
Python
import json
|
|
from datetime import date, timedelta, datetime, time
|
|
|
|
from pydantic_core import PydanticCustomError
|
|
from pydantic import BaseModel, EmailStr, ConfigDict, field_validator, model_validator
|
|
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)
|
|
date_of_birth: date
|
|
|
|
@field_validator("date_of_birth")
|
|
@classmethod
|
|
def validate_date_of_birth(cls, input_date):
|
|
if date.today() - input_date < 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(input_date, 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 input_date
|
|
|
|
|
|
class SUser(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
email: EmailStr
|
|
id: int
|
|
username: str
|
|
avatar_image: str
|
|
black_phoenix: bool
|
|
date_of_birth: date
|
|
date_of_registration: date
|
|
avatar_hex: str
|
|
|
|
|
|
class SUserRename(BaseModel):
|
|
username: str = Query(None, min_length=2, max_length=30)
|
|
password: str
|
|
|
|
|
|
class SUserAvatar(BaseModel):
|
|
password: str
|
|
new_avatar_image: str
|
|
avatar_hex: str
|
|
|
|
|
|
class SUserPassword(BaseModel):
|
|
current_password: str = Query(None, min_length=8)
|
|
new_password: str = Query(None, min_length=8)
|
|
new_password2: str = Query(None, min_length=8)
|
|
|
|
|
|
class SUserPasswordRecover(BaseModel):
|
|
email: EmailStr
|
|
|
|
|
|
class SUserCode(BaseModel):
|
|
user_code: str
|
|
|
|
|
|
class SUserPasswordChange(BaseModel):
|
|
user_id: int
|
|
password1: str = Query(None, min_length=8)
|
|
password2: str = Query(None, min_length=8)
|
|
|
|
|
|
class SRecoverEmailSent(BaseModel):
|
|
recover_email_sent: bool
|
|
|
|
|
|
class SEmailVerification(BaseModel):
|
|
email_verification: bool
|
|
|
|
|
|
class SUserToken(BaseModel):
|
|
access_token: str
|
|
|
|
|
|
class SUserName(BaseModel):
|
|
username: str
|
|
|
|
|
|
class SNewAvatar(BaseModel):
|
|
new_avatar_image: str
|
|
avatar_hex: str
|
|
|
|
|
|
class SConfirmPasswordRecovery(BaseModel):
|
|
user_id: int
|
|
|
|
|
|
class SPasswordRecovered(BaseModel):
|
|
username: str
|
|
|
|
|
|
class SCreateInvitationLink(BaseModel):
|
|
invitation_link: str
|
|
|
|
|
|
class SUserAddedToChat(BaseModel):
|
|
user_added_to_chat: bool
|
|
|
|
|
|
class SUserAvatars(BaseModel):
|
|
id: int
|
|
avatar_image: str
|
|
|
|
|
|
class SUsername(BaseModel):
|
|
username: str = Query(None, min_length=2, max_length=30)
|
|
|
|
|
|
class SEmail(BaseModel):
|
|
email: EmailStr
|