diff --git a/endpoints/fruits.get.py b/endpoints/fruits.get.py index 4115255..41c9c56 100644 --- a/endpoints/fruits.get.py +++ b/endpoints/fruits.get.py @@ -2,21 +2,11 @@ from fastapi import APIRouter, Depends from sqlalchemy.orm import Session from typing import List from core.database import get_db -from schemas.fruit import FruitSchema from helpers.fruit_helpers import get_all_fruits router = APIRouter() -@router.get("/fruits", response_model=List[FruitSchema]) -async def get_fruits(db: Session = Depends(get_db)): +@router.get("/fruits", response_model=List[str]) +async def get_fruit_names(db: Session = Depends(get_db)): fruits = get_all_fruits(db) - enhanced_fruits = [] - fancy_prefixes = ["Magnificent", "Splendid", "Extraordinary", "Marvelous", "Spectacular"] - from random import choice - - for fruit in fruits: - fruit_dict = fruit.__dict__ - fruit_dict["name"] = f"{choice(fancy_prefixes)} {fruit_dict['name']}" - enhanced_fruits.append(fruit_dict) - - return enhanced_fruits \ No newline at end of file + return [fruit.name for fruit in fruits] \ No newline at end of file