54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from pydantic import BaseModel, Field, EmailStr
|
|
from typing import Optional
|
|
from datetime import datetime
|
|
|
|
class UserBase(BaseModel):
|
|
username: str = Field(..., min_length=3, max_length=50)
|
|
email: EmailStr = Field(...)
|
|
company_name: Optional[str] = None
|
|
role: str = Field(default="user")
|
|
phone: Optional[str] = None
|
|
address: Optional[str] = None
|
|
is_active: bool = True
|
|
account_balance: float = 0.0
|
|
|
|
class UserCreate(UserBase):
|
|
password: str = Field(..., min_length=8)
|
|
|
|
class Config:
|
|
schema_extra = {
|
|
"example": {
|
|
"username": "johndoe",
|
|
"email": "john@example.com",
|
|
"password": "securepass123",
|
|
"company_name": "Acme Corp",
|
|
"role": "user",
|
|
"phone": "+1234567890",
|
|
"address": "123 Main St",
|
|
"is_active": True,
|
|
"account_balance": 0.0
|
|
}
|
|
}
|
|
|
|
class UserResponse(UserBase):
|
|
id: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
schema_extra = {
|
|
"example": {
|
|
"id": 1,
|
|
"username": "johndoe",
|
|
"email": "john@example.com",
|
|
"company_name": "Acme Corp",
|
|
"role": "user",
|
|
"phone": "+1234567890",
|
|
"address": "123 Main St",
|
|
"is_active": True,
|
|
"account_balance": 0.0,
|
|
"created_at": "2023-01-01T00:00:00",
|
|
"updated_at": "2023-01-01T00:00:00"
|
|
}
|
|
} |