80 lines
1.7 KiB
Python
80 lines
1.7 KiB
Python
from datetime import datetime
|
|
from typing import Annotated
|
|
|
|
from bson import ObjectId
|
|
from pydantic import BaseModel, BeforeValidator
|
|
|
|
|
|
def validate_object_id(v) -> str:
|
|
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 Client(BaseModel):
|
|
id: str | None = None
|
|
name: str
|
|
slug: str
|
|
is_active: bool = True
|
|
created_at: datetime | None = None
|
|
updated_at: datetime | None = None
|
|
|
|
|
|
class ClientCreate(BaseModel):
|
|
name: str
|
|
slug: str
|
|
|
|
|
|
class ClientUpdate(BaseModel):
|
|
name: str | None = None
|
|
slug: str | None = None
|
|
is_active: bool | None = None
|
|
|
|
|
|
class Team(BaseModel):
|
|
id: str | None = None
|
|
name: str
|
|
client_id: str
|
|
member_user_ids: list[str] = []
|
|
created_at: datetime | None = None
|
|
updated_at: datetime | None = None
|
|
|
|
|
|
class TeamCreate(BaseModel):
|
|
name: str
|
|
|
|
|
|
class TeamUpdate(BaseModel):
|
|
name: str | None = None
|
|
|
|
|
|
class Project(BaseModel):
|
|
id: str | None = None
|
|
name: str
|
|
client_id: str
|
|
is_active: bool = True
|
|
default_languages: list[str] = []
|
|
default_linguist_id: str | None = None
|
|
default_reviewer_id: str | None = None
|
|
created_at: datetime | None = None
|
|
updated_at: datetime | None = None
|
|
|
|
|
|
class ProjectCreate(BaseModel):
|
|
name: str
|
|
default_languages: list[str] = []
|
|
default_linguist_id: str | None = None
|
|
default_reviewer_id: str | None = None
|
|
|
|
|
|
class ProjectUpdate(BaseModel):
|
|
name: str | None = None
|
|
is_active: bool | None = None
|
|
default_languages: list[str] | None = None
|
|
default_linguist_id: str | None = None
|
|
default_reviewer_id: str | None = None
|