update model for todo and subtodo table

Author: blueoc-bachnguyen

backend/app/alembic/versions/9245f5673cb7_update_todo_and_subtodo_table.py

@@ -0,0 +1,63 @@
+"""update todo and subtodo table
+
+Revision ID: 9245f5673cb7
+Revises: 80580dd7587b
+Create Date: 2024-12-06 10:53:59.113398
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+
+
+# revision identifiers, used by Alembic.
+revision = '9245f5673cb7'
+down_revision = '80580dd7587b'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('subtodo', 'desc',
+               existing_type=sa.VARCHAR(length=255),
+               nullable=False)
+    op.alter_column('subtodo', 'status',
+               existing_type=sa.VARCHAR(length=20),
+               type_=sqlmodel.sql.sqltypes.AutoString(length=255),
+               existing_nullable=False)
+    op.add_column('todo', sa.Column('owner_id', sa.Uuid(), nullable=False))
+    op.alter_column('todo', 'desc',
+               existing_type=sa.VARCHAR(length=255),
+               nullable=False)
+    op.alter_column('todo', 'status',
+               existing_type=sa.VARCHAR(length=20),
+               type_=sqlmodel.sql.sqltypes.AutoString(length=255),
+               existing_nullable=False)
+    op.drop_constraint('todo_user_id_fkey', 'todo', type_='foreignkey')
+    op.create_foreign_key(None, 'todo', 'user', ['owner_id'], ['id'], ondelete='CASCADE')
+    op.drop_column('todo', 'user_id')
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('todo', sa.Column('user_id', sa.UUID(), autoincrement=False, nullable=False))
+    op.drop_constraint(None, 'todo', type_='foreignkey')
+    op.create_foreign_key('todo_user_id_fkey', 'todo', 'user', ['user_id'], ['id'], ondelete='CASCADE')
+    op.alter_column('todo', 'status',
+               existing_type=sqlmodel.sql.sqltypes.AutoString(length=255),
+               type_=sa.VARCHAR(length=20),
+               existing_nullable=False)
+    op.alter_column('todo', 'desc',
+               existing_type=sa.VARCHAR(length=255),
+               nullable=True)
+    op.drop_column('todo', 'owner_id')
+    op.alter_column('subtodo', 'status',
+               existing_type=sqlmodel.sql.sqltypes.AutoString(length=255),
+               type_=sa.VARCHAR(length=20),
+               existing_nullable=False)
+    op.alter_column('subtodo', 'desc',
+               existing_type=sa.VARCHAR(length=255),
+               nullable=True)
+    # ### end Alembic commands ###

backend/app/models.py

@@ -1,4 +1,5 @@
 import uuid
+from enum import Enum
 
 from pydantic import EmailStr
 from sqlmodel import Field, Relationship, SQLModel
@@ -44,6 +45,7 @@ class User(UserBase, table=True):
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
     hashed_password: str
     items: list["Item"] = Relationship(back_populates="owner", cascade_delete=True)
+    todos: list["Todo"] = Relationship(back_populates="owner", cascade_delete=True)
 
 
 # Properties to return via API, id is always required
@@ -113,26 +115,71 @@ class NewPassword(SQLModel):
     token: str
     new_password: str = Field(min_length=8, max_length=40)
 
+# Shared properties
+class StatusEnum(str, Enum):
+    pending = "pending"
+    completed = "completed"
+    in_progress = "in_progress"
+    
 class TodoBase(SQLModel):
     title: str = Field(min_length=1, max_length=255)
-    desc: str | None = Field(default=None, max_length=255)
+    desc: str = Field(max_length=255)
 
 # Table Todo
 class Todo(TodoBase, table=True):
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    title: str = Field(max_length=255)
-    desc: str | None = Field(default=None, max_length=255)
-    user_id: uuid.UUID = Field(
+    owner_id: uuid.UUID = Field(
         foreign_key="user.id", nullable=False, ondelete="CASCADE"
     )
-    status: str = Field(max_length=20)
+    status: str = Field(max_length=255)
+    owner: User | None = Relationship(back_populates="todos")
+    subtodos: list["SubTodo"] = Relationship(back_populates="todo")
+
+class TodoCreate(TodoBase):
+    pass
+
+# Properties to receive on item update
+class TodoUpdate(TodoBase):
+    title: str | None = Field(default=None, min_length=1, max_length=255)  # type: ignore
+    desc: str | None = Field(default=None, max_length=255)
+    status: str = Field(max_length=255)
+
+class TodoPublic(TodoBase):
+    id: uuid.UUID
+    owner_id: uuid.UUID
+    status: StatusEnum
+
+class TodosPublic(SQLModel):
+    data: list[TodoPublic]
+    count: int
 
 # Table SubTodo
-class SubTodo(TodoBase, table=True):
+class SubTodoBase(SQLModel):
+    title: str = Field(min_length=1, max_length=255)
+    desc: str = Field(max_length=255)
+
+class SubTodo(SubTodoBase, table=True):
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    title: str = Field(max_length=255)
-    desc: str | None = Field(default=None, max_length=255)
     todo_id: uuid.UUID = Field(
         foreign_key="todo.id", nullable=False, ondelete="CASCADE"
     )
-    status: str = Field(max_length=20)
+    status: str = Field(max_length=255)
+    todo: Todo | None = Relationship(back_populates="subtodos")
+
+class SubTodoCreate(SubTodoBase):
+    pass
+
+# Properties to receive on item update
+class SubTodoUpdate(SubTodoBase):
+    title: str | None = Field(default=None, min_length=1, max_length=255)  # type: ignore
+    desc: str | None = Field(default=None, max_length=255)
+    status: StatusEnum | None = Field(default=None)
+    
+class SubTodoPublic(SubTodoBase):
+    id: uuid.UUID
+    todo_id: uuid.UUID
+    status: StatusEnum  
+
+class SubTodosPublic(SQLModel):
+    data: list[SubTodoPublic]
+    count: int