feat: Updated endpoint endpoints/fruits.get.py via AI

This commit is contained in:
Backend IM Bot 2025-04-12 01:31:39 +00:00
parent c8f49faef2
commit 728d305a42

View File

@ -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
return [fruit.name for fruit in fruits]