23 lines
603 B
Python
23 lines
603 B
Python
# Entity: Book
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.orm import Session
|
|
from typing import List
|
|
from core.database import get_db
|
|
from core.models.book import Book
|
|
from core.schemas.book import BookSchema, BookCreate
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("/books", status_code=201, response_model=List[BookSchema])
|
|
async def generate_books(
|
|
book_data: BookCreate,
|
|
db: Session = Depends(get_db)
|
|
):
|
|
books = []
|
|
new_book = Book(**book_data.dict())
|
|
db.add(new_book)
|
|
db.commit()
|
|
db.refresh(new_book)
|
|
books.append(new_book)
|
|
return books |