diff --git a/endpoints/playlist.get.py b/endpoints/playlist.get.py index c164415..f083d5c 100644 --- a/endpoints/playlist.get.py +++ b/endpoints/playlist.get.py @@ -1,19 +1,23 @@ -# Entity: Playlist +# Entity: Meal from fastapi import APIRouter, Depends, HTTPException, status from sqlalchemy.orm import Session from typing import List from core.database import get_db -from models.playlist import Playlist -from schemas.playlist import PlaylistSchema -from helpers.playlist_helpers import get_playlist_songs +from models.meal import Meal +from schemas.meal import MealSchema +from helpers.meal_helpers import get_all_meals, get_meals_by_category router = APIRouter() -@router.get("/playlist", response_model=List[PlaylistSchema], status_code=200) -async def get_playlist_songs( +@router.get("/meals", status_code=200, response_model=List[MealSchema]) +async def get_meals( + category: str = None, db: Session = Depends(get_db) ): - """Get all songs from playlist""" - songs = get_playlist_songs(db) - return songs \ No newline at end of file + """Get all meals with optional category filter""" + if category: + meals = get_meals_by_category(db, category) + else: + meals = get_all_meals(db) + return meals \ No newline at end of file