diff --git a/api/src/database/collection.py b/api/src/database/collection.py index f645396..e3e24aa 100644 --- a/api/src/database/collection.py +++ b/api/src/database/collection.py @@ -1,6 +1,7 @@ from src.database.document import Document from src.database.models.user_models import UserModel from src.database.models.project_models import ProjectModel +from src.database.models.case_models import CaseModel def project_document() -> Document: @@ -8,6 +9,11 @@ def project_document() -> Document: return Document(collection="projects", model=ProjectModel, unique_field="name") +def case_document() -> Document: + """Get case document.""" + return Document(collection="cases", model=CaseModel, unique_field="name") + + def user_document() -> Document: """Get user document.""" return Document(collection="users", model=UserModel, unique_field="email") diff --git a/api/src/database/models/case_models.py b/api/src/database/models/case_models.py new file mode 100644 index 0000000..5d3d001 --- /dev/null +++ b/api/src/database/models/case_models.py @@ -0,0 +1,10 @@ +from typing import Any +from src.database.models import BaseDBModel + + +class CaseModel(BaseDBModel): + """Case Model.""" + + name: str + project_id: str + diagram_data: dict[Any, Any] diff --git a/api/src/routes/payload_models/case_ models.py b/api/src/routes/payload_models/case_ models.py new file mode 100644 index 0000000..fecb3b0 --- /dev/null +++ b/api/src/routes/payload_models/case_ models.py @@ -0,0 +1,7 @@ +from pydantic import BaseModel + + +class CaseModel(BaseModel): + """Case model.""" + + name: str