92 lines
3.2 KiB
Python
92 lines
3.2 KiB
Python
from datetime import datetime
|
|
from typing import Optional
|
|
from pydantic import BaseModel, EmailStr, Field, field_validator, ConfigDict
|
|
import re
|
|
|
|
class UserBase(BaseModel):
|
|
username: str = Field(..., min_length=3, max_length=50)
|
|
email: EmailStr
|
|
|
|
@field_validator('username')
|
|
@classmethod
|
|
def validate_username(cls, v: str) -> str:
|
|
if not re.match(r'^[a-zA-Z0-9_-]+$', v):
|
|
raise ValueError('Username must contain only alphanumeric characters, underscores, and dashes')
|
|
return v
|
|
|
|
class UserCreate(UserBase):
|
|
password: str = Field(..., min_length=8, max_length=128)
|
|
|
|
@field_validator('password')
|
|
@classmethod
|
|
def validate_password_strength(cls, v: str) -> str:
|
|
if not re.search(r'[A-Z]', v):
|
|
raise ValueError('Password must contain at least one uppercase letter')
|
|
if not re.search(r'[a-z]', v):
|
|
raise ValueError('Password must contain at least one lowercase letter')
|
|
if not re.search(r'\d', v):
|
|
raise ValueError('Password must contain at least one digit')
|
|
return v
|
|
|
|
class User(UserBase):
|
|
id: int
|
|
is_active: bool
|
|
is_superuser: bool
|
|
created_at: datetime
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
class UserCreateByAdmin(UserBase):
|
|
password: str = Field(..., min_length=8, max_length=128)
|
|
is_superuser: bool = False
|
|
|
|
@field_validator('password')
|
|
@classmethod
|
|
def validate_password_strength(cls, v: str) -> str:
|
|
if not re.search(r'[A-Z]', v):
|
|
raise ValueError('Password must contain at least one uppercase letter')
|
|
if not re.search(r'[a-z]', v):
|
|
raise ValueError('Password must contain at least one lowercase letter')
|
|
if not re.search(r'\d', v):
|
|
raise ValueError('Password must contain at least one digit')
|
|
return v
|
|
|
|
class UserUpdate(BaseModel):
|
|
username: Optional[str] = Field(None, min_length=3, max_length=50)
|
|
email: Optional[EmailStr] = None
|
|
password: Optional[str] = Field(None, min_length=8, max_length=128)
|
|
is_active: Optional[bool] = None
|
|
is_superuser: Optional[bool] = None
|
|
|
|
@field_validator('username')
|
|
@classmethod
|
|
def validate_username(cls, v: Optional[str]) -> Optional[str]:
|
|
if v is not None and not re.match(r'^[a-zA-Z0-9_-]+$', v):
|
|
raise ValueError('Username must contain only alphanumeric characters, underscores, and dashes')
|
|
return v
|
|
|
|
@field_validator('password')
|
|
@classmethod
|
|
def validate_password_strength(cls, v: Optional[str]) -> Optional[str]:
|
|
if v is not None:
|
|
if not re.search(r'[A-Z]', v):
|
|
raise ValueError('Password must contain at least one uppercase letter')
|
|
if not re.search(r'[a-z]', v):
|
|
raise ValueError('Password must contain at least one lowercase letter')
|
|
if not re.search(r'\d', v):
|
|
raise ValueError('Password must contain at least one digit')
|
|
return v
|
|
|
|
class UserListResponse(BaseModel):
|
|
users: list[User]
|
|
total: int
|
|
skip: int
|
|
limit: int
|
|
|
|
class Token(BaseModel):
|
|
access_token: str
|
|
token_type: str
|
|
|
|
class TokenData(BaseModel):
|
|
username: Optional[str] = None
|