From 1136d3aeb25502f54d18460e23ba37420b848a7c Mon Sep 17 00:00:00 2001
From: Mubashir78 <mubashirrindbaloch@gmail.com>
Date: Sat, 4 Jul 2026 02:36:06 +0500
Subject: [PATCH] Add due_date field to Task model

---
 models.py  | 4 +++-
 schemas.py | 4 ++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/models.py b/models.py
index d9182d0..3667a20 100644
--- a/models.py
+++ b/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")
\ No newline at end of file
diff --git a/schemas.py b/schemas.py
index de48d02..bcdb1b2 100644
--- a/schemas.py
+++ b/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: