Add due_date field to Task model

models.py

@@ -1,4 +1,5 @@
-from sqlalchemy import Column, Integer, String, Boolean, ForeignKey
+from datetime import date
+from sqlalchemy import Column, Integer, String, Boolean, Date, ForeignKey
 from sqlalchemy.orm import relationship
 from database import Base 
 
@@ -20,6 +21,7 @@ class Task(Base):
     description = Column(String, nullable=True)
     completed = Column(Boolean, default=False)
     priority = Column(String, default="medium")
+    due_date = Column(Date, nullable=True)
     owner_id = Column(Integer, ForeignKey("users.id"))
 
     owner = relationship("User", back_populates="tasks")

schemas.py

@@ -1,3 +1,4 @@
+from datetime import date
 from pydantic import BaseModel
 from typing import Optional, List
 
@@ -8,17 +9,20 @@ class TaskBase(BaseModel):
 class TaskCreate(TaskBase):
     owner_id: int
     priority: str = "medium"
+    due_date: Optional[date] = None
 
 class TaskUpdate(BaseModel):
     title: Optional[str] = None 
     description: Optional[str] = None 
     completed: Optional[bool] = None
     priority: Optional[str] = None
+    due_date: Optional[date] = None
 
 class TaskResponse(TaskBase):
     id: int 
     completed: bool 
     priority: str
+    due_date: Optional[date] = None
     owner_id: int 
 
     class Config: