diff --git a/endpoints/books.post.py b/endpoints/books.post.py index 88070fb..b93c2ab 100644 --- a/endpoints/books.post.py +++ b/endpoints/books.post.py @@ -1,6 +1,6 @@ # Entity: Pen -from fastapi import APIRouter, Depends, HTTPException, status +from fastapi import APIRouter, Depends, status from sqlalchemy.orm import Session from core.database import get_db from models.pen import Pen @@ -9,10 +9,9 @@ from helpers.pen_helpers import create_pen router = APIRouter() -@router.post("/pens", status_code=201, response_model=PenSchema) +@router.post("/pens", status_code=status.HTTP_201_CREATED, response_model=PenSchema) async def create_pen_endpoint( pen_data: PenCreate, db: Session = Depends(get_db) ): - pen = create_pen(db, pen_data) - return pen \ No newline at end of file + return create_pen(db, pen_data) \ No newline at end of file