menu models

Author: rony1philip

backend/app/alembic/versions/32ba675b8ec6_add_menu_b_class_menu_base_model.py

@@ -0,0 +1,39 @@
+"""Add menu_b class Menu base model
+
+Revision ID: 32ba675b8ec6
+Revises: 1cabfca3d8c8
+Create Date: 2025-03-05 08:58:47.583292
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+
+
+# revision identifiers, used by Alembic.
+revision = '32ba675b8ec6'
+down_revision = '1cabfca3d8c8'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('menu',
+    sa.Column('description', sqlmodel.sql.sqltypes.AutoString(length=255), nullable=True),
+    sa.Column('data', sa.JSON(), nullable=True),
+    sa.Column('current', sa.Boolean(), nullable=False),
+    sa.Column('id', sa.Uuid(), nullable=False),
+    sa.Column('title', sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False),
+    sa.Column('created_datetime', sa.TIMESTAMP(timezone=True), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False),
+    sa.Column('owner_id', sa.Uuid(), nullable=False),
+    sa.ForeignKeyConstraint(['owner_id'], ['patient.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('menu')
+    # ### end Alembic commands ###

backend/app/models.py

@@ -1,8 +1,11 @@
 import uuid
 from typing import Optional
-from pydantic import EmailStr
+from pydantic import BaseModel, EmailStr
+from pydantic import Field as F
 from sqlmodel import Field, Relationship, SQLModel, Column, TIMESTAMP, text, DATE
 from datetime import datetime
+from sqlalchemy import JSON
+from sqlalchemy import Column as Col
 import enum
 # Shared properties
 class UserBase(SQLModel):
@@ -161,6 +164,8 @@ class Patient(PatientBase, table=True):
         nullable=False,
         server_default=text("CURRENT_TIMESTAMP"),
     ))
+    menus: list["Menu"] = Relationship(back_populates="owner", cascade_delete=True)
+
 
 # Properties to return via API, id is always required
 class PatientPublic(PatientBase):
@@ -170,3 +175,61 @@ class PatientPublic(PatientBase):
 class PatientsPublic(SQLModel):
     data: list[PatientPublic]
     count: int
+
+class Menu_B(BaseModel):
+    def __init__(self):
+        self.data: dict = F(default=self.menu_builder())
+
+    def menu_builder(self):
+        days = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday" , "Sunday"]
+        mills = ["breakfast",  "mid-morning", "lunch","mid-afternoon" , "dinner"]
+        new_menu = {}
+        for day in days:
+            new_menu.update({day:{mill:{} for mill in mills}})
+        self.data = new_menu 
+        return self.data
+
+# Shared properties
+class MenuBase(SQLModel,Menu_B):
+    title: str = Field(min_length=1, max_length=255)
+    description: str | None = Field(default=None, max_length=255)
+    data: dict = Field(sa_column=Col(JSON), default_factory=dict)
+    current: bool = Field()
+
+
+# Properties to receive on item creation
+class MenuCreate(MenuBase):
+    pass
+
+
+# Properties to receive on item update
+class MenuUpdate(MenuBase):
+    title: str | None = Field(default=None, min_length=1, max_length=255)  # type: ignore
+    description: str | None = Field(default=None, max_length=255)
+    data: dict = Field(sa_column=Column(JSON), default_factory=dict)
+    current: bool = Field()
+
+# Database model, database table inferred from class name
+class Menu(MenuBase, table=True):
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    title: str = Field(max_length=255)
+    created_datetime: Optional[datetime] = Field(sa_column=Column(
+        TIMESTAMP(timezone=True),
+        nullable=False,
+        server_default=text("CURRENT_TIMESTAMP"),
+    ))
+    owner_id: uuid.UUID = Field(
+        foreign_key="patient.id", nullable=False, ondelete="CASCADE"
+    )
+    owner: Patient | None = Relationship(back_populates="menus")
+
+
+# Properties to return via API, id is always required
+class MenuPublic(MenuBase):
+    pass
+
+
+class MenusPublic(SQLModel):
+    data: list[MenuPublic]
+    count: int
+