70 lines
1.7 KiB
Python
70 lines
1.7 KiB
Python
from datetime import datetime
|
|
from enum import Enum
|
|
from typing import Annotated
|
|
|
|
from bson import ObjectId
|
|
from pydantic import BaseModel, BeforeValidator, EmailStr, Field
|
|
|
|
|
|
def validate_object_id(v) -> str:
|
|
"""Convert ObjectId to string"""
|
|
if isinstance(v, ObjectId):
|
|
return str(v)
|
|
if isinstance(v, str):
|
|
return v
|
|
raise ValueError('Invalid ObjectId')
|
|
|
|
|
|
PyObjectId = Annotated[str, BeforeValidator(validate_object_id)]
|
|
|
|
|
|
class UserRole(str, Enum):
|
|
CLIENT = "client"
|
|
REVIEWER = "reviewer"
|
|
LINGUIST = "linguist"
|
|
PRODUCTION = "production"
|
|
PROJECT_MANAGER = "project_manager"
|
|
ADMIN = "admin"
|
|
|
|
|
|
class AuthProvider(str, Enum):
|
|
LOCAL = "local"
|
|
MICROSOFT = "microsoft"
|
|
|
|
|
|
class User(BaseModel):
|
|
id: PyObjectId | None = Field(None, alias="_id")
|
|
email: EmailStr
|
|
hashed_password: str | None = None # Optional for Microsoft users
|
|
full_name: str
|
|
role: UserRole = UserRole.CLIENT
|
|
auth_provider: AuthProvider = AuthProvider.LOCAL
|
|
is_active: bool = True
|
|
pm_client_ids: list[str] = [] # Client IDs where this user is Project Manager (admin-assigned)
|
|
languages: list[str] = [] # BCP-47 language codes the user is competent in (R-8)
|
|
created_at: datetime | None = None
|
|
updated_at: datetime | None = None
|
|
|
|
class Config:
|
|
populate_by_name = True
|
|
use_enum_values = True
|
|
|
|
|
|
class UserInDB(User):
|
|
pass
|
|
|
|
|
|
class UserCreate(BaseModel):
|
|
email: EmailStr
|
|
password: str
|
|
full_name: str
|
|
role: UserRole = UserRole.CLIENT
|
|
|
|
|
|
class UserUpdate(BaseModel):
|
|
email: EmailStr | None = None
|
|
full_name: str | None = None
|
|
role: UserRole | None = None
|
|
is_active: bool | None = None
|
|
pm_client_ids: list[str] | None = None
|
|
languages: list[str] | None = None
|