Merge pull request #35 from GenerateNU/qihong/search-filter-items

GET items search and filter

Author: qihongw08

app/api/endpoints/item.py

@@ -3,7 +3,8 @@
 from fastapi import APIRouter, Body, Depends, HTTPException, status
 
 from app.api.endpoints.user import get_current_user
-from app.models.item import item_model
+from app.core.enums import SortOrder
+from app.models.item import ItemSortParam, item_model
 from app.schemas.item import CreateItemRequest, Item, UpdateItemRequest
 from app.schemas.user import User, UserType
 from app.services.item import ItemService
@@ -27,8 +28,13 @@ async def post_item(
 
 
 @router.get("/all", response_model=list[Item])
-async def get_items() -> list[Item]:
-    return await item_model.get_all_items()
+async def get_items(
+    search_text: str | None = None,
+    vendor_id: str | None = None,
+    sort_by: ItemSortParam | None = None,
+    sort_order: SortOrder = SortOrder.ASC,
+) -> list[Item]:
+    return await item_model.get_items(search_text, vendor_id, sort_by, sort_order)
 
 
 @router.get("/{item_id}", response_model=Item)

app/core/enums.py

@@ -0,0 +1,6 @@
+from enum import Enum
+
+
+class SortOrder(str, Enum):
+    ASC = "asc"
+    DESC = "desc"

app/models/item.py

@@ -4,8 +4,9 @@
 from fastapi import HTTPException
 from motor.motor_asyncio import AsyncIOMotorCollection  # noqa: TCH002
 
+from app.core.enums import SortOrder
 from app.database.mongodb import db
-from app.schemas.item import CreateItemRequest, Item, UpdateItemRequest
+from app.schemas.item import CreateItemRequest, Item, ItemSortParam, UpdateItemRequest
 from app.utils.object_id import parse_object_id
 
 
@@ -31,6 +32,36 @@ async def get_all_items(self) -> list[Item]:
 
         return [self.to_item(item) for item in items_list]
 
+    async def get_items(
+        self,
+        search_text: str | None = None,
+        vendor_id: str | None = None,
+        sort_by: ItemSortParam | None = None,
+        sort_order: SortOrder = SortOrder.ASC,
+    ) -> list[Item]:
+        query = {}
+
+        if search_text:
+            query["name"] = {
+                "$regex": search_text,
+                "$options": "i",
+            }  # case-insensitive partial match
+
+        if vendor_id:
+            query["vendor_id"] = ObjectId(vendor_id)
+
+        sort_criteria = []
+        if sort_by:
+            sort_direction = 1 if sort_order == SortOrder.ASC else -1
+            sort_criteria.append((sort_by.field_name, sort_direction))
+
+        if sort_criteria:
+            items_list = await self.collection.find(query).sort(sort_criteria).to_list()
+        else:
+            items_list = await self.collection.find(query).to_list()
+
+        return [self.to_item(item) for item in items_list]
+
     async def get_item(self, item_id: str) -> Item | None:
         item_obj_id = parse_object_id(item_id)
 

app/schemas/item.py

@@ -1,4 +1,5 @@
 from datetime import datetime
+from enum import Enum
 
 from pydantic import BaseModel
 
@@ -16,6 +17,21 @@ class Config:
         from_attributes = True
 
 
+class ItemSortParam(str, Enum):
+    DATE = "date"
+    NAME = "name"
+    COINS = "coins"
+
+    @property
+    def field_name(self) -> str:
+        field_mapping = {
+            ItemSortParam.DATE: "time_posted",
+            ItemSortParam.NAME: "name",
+            ItemSortParam.COINS: "price",
+        }
+        return field_mapping[self]
+
+
 class CreateItemRequest(BaseModel):
     name: str
     expiration: datetime