patch schema

src/api/dependacies/task_dep.py

@@ -1,32 +1,13 @@
 from datetime import date
-from enum import Enum
 from typing import Annotated
 
 from fastapi import Depends, Query
 from fastapi.exceptions import HTTPException
 from pydantic import BaseModel, model_validator
 
+from src.schemas.tasks import TaskFilter
 
-class StatEnum(str, Enum):
-    open = "open"
-    closed = "closed"
-    in_progress = "in_progress"
-    todo = "todo"
-
-
-class Pagination(BaseModel):
-    page: int | None = Query(default=0, ge=0)
-    limit: int | None = Query(default=30, ge=0, le=50)
-
-
-PaginationTasksDep = Annotated[Pagination, Depends()]
-
-
-class Status(BaseModel):
-    status: StatEnum | None = Query(default=None)
-
-
-StatusTaskDep = Annotated[Status, Depends()]
+TaskFilterDep = Annotated[TaskFilter, Depends()]
 
 
 class Date(BaseModel):

src/api/v1/tasks.py

@@ -3,7 +3,7 @@ from typing import Annotated
 from fastapi import APIRouter, Depends
 
 from src.api.dependacies.db_dep import sessionDep
-from src.api.dependacies.task_dep import DateDep, PaginationTasksDep, StatusTaskDep
+from src.api.dependacies.task_dep import TaskFilterDep
 from src.api.dependacies.user_dep import ActiveUser, TaskOwnerDep
 from src.schemas.tasks import TaskADDRequest
 from src.services.tasks import TaskService
@@ -16,17 +16,11 @@ router = APIRouter(prefix="/tasks", tags=["Tasks"])
 async def get_tasks(
     session: sessionDep,
     user: ActiveUser,
-    page: PaginationTasksDep,
-    status: StatusTaskDep,
-    date_filter: DateDep,
-):
+    filter: TaskFilterDep
+    ):
     result = await UserService(session).get_user_with_tasks(
         user_id=user.id,
-        status=status.status,
-        limit=page.limit,
-        offset=page.page,
-        date_to=date_filter.date_to,
-        date_from=date_filter.date_from,
+        **filter.model_dump(exclude_unset=True)
     )
     return result
 

src/schemas/tasks.py

@@ -1,13 +1,27 @@
 from datetime import date
-from typing import Literal
+from enum import Enum
 
 from pydantic import BaseModel, ConfigDict
 
 
+class PriorityEnum(str, Enum):
+    low = "low"
+    medium = "medium"
+    high = "high"
+    critical = "critical"
+
+
+class StatusEnum(str, Enum):
+    open = "open"
+    closed = "closed"
+    in_progress = "in_progress"
+    todo = "todo"
+
+
 class TaskShort(BaseModel):
     title: str
     due_date: date | None = None
-    priority: Literal["low", "medium", "high", "critical"] = "medium"
+    priority: PriorityEnum = PriorityEnum.medium
     model_config = ConfigDict(from_attributes=True)
 
 
@@ -20,10 +34,27 @@ class TaskADDRequest(TaskShort):
     description: str | None = None
 
 
+class TaskPATCHRequest(BaseModel):
+    title: str | None = None
+    description: str | None = None
+    due_date: date | None = None
+    status: StatusEnum | None = None
+    priority: PriorityEnum | None = None
+    time_spent: int | None = None
+
+
 class Task(TaskADDRequest):
     id: int
     user_id: int
-    status: Literal["open", "closed", "in_progress", "todo"]
+    status: StatusEnum
     time_spent: int
 
     model_config = ConfigDict(from_attributes=True)
+
+
+class TaskFilter(BaseModel):
+    status: StatusEnum | None = None
+    limit: int | None = 30
+    offset: int | None = None
+    date_from: date | None = None
+    date_to: date | None = None

src/services/tasks.py

@@ -20,6 +20,8 @@ class TaskService(BaseService):
 
     async def get_task(self, task_id: int):
         task = await self.session.task.get_one_or_none(id=task_id)
+        if task is None:
+            raise HTTPException(status_code=404, detail="Task not found.")
         return Task.model_validate(task)
 
     async def delete_task(self, task_id: int):

src/services/users.py

@@ -49,8 +49,8 @@ class UserService(BaseService):
         offset: int | None,
         date_to: date | None,
         date_from: date | None,
+        
     ):
-        print(type(status), status)
         user = await self.session.user.get_one_with_load(
             user_id=user_id,
             status=status,