2025-03-27 19:51:47 +00:00

48 lines
1.4 KiB
Python

from pydantic import BaseModel, Field, EmailStr
from typing import Optional
from datetime import datetime
from enum import Enum
class UserRoles(str, Enum):
admin = "admin"
lawyer = "lawyer"
client = "client"
class UserBase(BaseModel):
first_name: str = Field(..., min_length=1, max_length=100)
last_name: str = Field(..., min_length=1, max_length=100)
email: EmailStr = Field(..., max_length=255)
phone_number: str = Field(..., max_length=20)
role: UserRoles
bar_number: str = Field(..., max_length=50)
is_active: bool = True
address: Optional[str] = None
profile_picture: str = Field(..., max_length=255)
class UserCreate(UserBase):
password: str = Field(..., min_length=8, max_length=255)
class Config:
schema_extra = {
"example": {
"first_name": "John",
"last_name": "Doe",
"email": "john@example.com",
"password": "securepass123",
"phone_number": "+1234567890",
"role": "lawyer",
"bar_number": "BAR123456",
"is_active": True,
"address": "123 Main St, City",
"profile_picture": "http://example.com/profile.jpg"
}
}
class UserResponse(UserBase):
id: int
last_login: Optional[datetime] = None
created_at: datetime
updated_at: datetime
class Config:
orm_mode = True