74 lines
1.7 KiB
Python
74 lines
1.7 KiB
Python
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, EmailStr, Field
|
|
|
|
from app.core.enums import OtpPurpose, UserRole
|
|
|
|
class UserBase(BaseModel):
|
|
username: str
|
|
email: EmailStr
|
|
|
|
|
|
class UserCreate(UserBase):
|
|
password: str = Field(..., min_length=8)
|
|
role: UserRole = UserRole.VIEWER
|
|
gatekeeper_groups: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class UserUpdate(BaseModel):
|
|
email: Optional[EmailStr] = None
|
|
role: Optional[UserRole] = None
|
|
gatekeeper_groups: Optional[list[str]] = None
|
|
is_active: Optional[bool] = None
|
|
|
|
|
|
class UserInDB(UserBase):
|
|
id: int
|
|
role: str
|
|
gatekeeper_groups: list[str] = Field(default_factory=list)
|
|
is_active: bool
|
|
last_login_at: Optional[datetime]
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class UserResponse(UserBase):
|
|
id: int
|
|
role: str
|
|
gatekeeper_groups: list[str] = Field(default_factory=list)
|
|
is_active: bool
|
|
email_verified: bool = False
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class UserRegister(BaseModel):
|
|
username: str = Field(..., min_length=3, max_length=50)
|
|
email: EmailStr
|
|
password: str = Field(..., min_length=8, max_length=128)
|
|
|
|
|
|
class VerifyEmailRequest(BaseModel):
|
|
email: EmailStr
|
|
code: str = Field(..., min_length=6, max_length=6)
|
|
|
|
|
|
class ResendCodeRequest(BaseModel):
|
|
email: EmailStr
|
|
purpose: OtpPurpose = OtpPurpose.REGISTER
|
|
|
|
|
|
class ForgotPasswordRequest(BaseModel):
|
|
email: EmailStr
|
|
|
|
|
|
class ResetPasswordRequest(BaseModel):
|
|
email: EmailStr
|
|
code: str = Field(..., min_length=6, max_length=6)
|
|
new_password: str = Field(..., min_length=8, max_length=128)
|