Skip to content

Commit

Permalink
add case document and models
Browse files Browse the repository at this point in the history
  • Loading branch information
itsmostafa committed Feb 12, 2025
1 parent 1773e73 commit 69cfa68
Show file tree
Hide file tree
Showing 3 changed files with 23 additions and 0 deletions.
6 changes: 6 additions & 0 deletions api/src/database/collection.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,19 @@
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:
"""Get project 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")
10 changes: 10 additions & 0 deletions api/src/database/models/case_models.py
Original file line number Diff line number Diff line change
@@ -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]
7 changes: 7 additions & 0 deletions api/src/routes/payload_models/case_ models.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
from pydantic import BaseModel


class CaseModel(BaseModel):
"""Case model."""

name: str

0 comments on commit 69cfa68

Please sign in to comment.