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