stream-project-ycnpdj/endpoints/create-fruit.post.py

21 lines
804 B
Python

from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session
from core.database import get_db
from schemas.fruit import FruitCreate, FruitSchema
from helpers.fruit_helpers import create_fruit, validate_fruit_name, format_fruit_response
router = APIRouter()
@router.post("/create-fruit", status_code=status.HTTP_201_CREATED, response_model=FruitSchema)
async def create_new_fruit(
fruit: FruitCreate,
db: Session = Depends(get_db)
):
if not validate_fruit_name(fruit.name):
raise HTTPException(status_code=400, detail="Invalid fruit name")
new_fruit = create_fruit(db=db, fruit_data=fruit)
if not new_fruit:
raise HTTPException(status_code=400, detail="Could not create fruit")
return format_fruit_response(new_fruit)