feat: add POST endpoint to create books by title and author
This commit is contained in:
parent
c042992398
commit
233c84fd09
34
alembic/versions/20250429_190900_ba53f86c_update_book.py
Normal file
34
alembic/versions/20250429_190900_ba53f86c_update_book.py
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
"""Initial creation of books table
|
||||||
|
Revision ID: 1a2b3c4d5e6f
|
||||||
|
Revises: 0001
|
||||||
|
Create Date: 2023-07-21 10:00:00.000000
|
||||||
|
"""
|
||||||
|
from alembic import op
|
||||||
|
import sqlalchemy as sa
|
||||||
|
|
||||||
|
|
||||||
|
# revision identifiers, used by Alembic.
|
||||||
|
revision = '1a2b3c4d5e6f'
|
||||||
|
down_revision = '0001'
|
||||||
|
branch_labels = None
|
||||||
|
depends_on = None
|
||||||
|
|
||||||
|
|
||||||
|
def upgrade():
|
||||||
|
op.create_table(
|
||||||
|
'books',
|
||||||
|
sa.Column('id', sa.String(36), primary_key=True),
|
||||||
|
sa.Column('title', sa.String(), nullable=False),
|
||||||
|
sa.Column('author', sa.String(), nullable=False),
|
||||||
|
sa.Column('description', sa.Text(), nullable=True),
|
||||||
|
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
||||||
|
sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now())
|
||||||
|
)
|
||||||
|
op.create_index(op.f('ix_books_title'), 'books', ['title'], unique=False)
|
||||||
|
op.create_index(op.f('ix_books_author'), 'books', ['author'], unique=False)
|
||||||
|
|
||||||
|
|
||||||
|
def downgrade():
|
||||||
|
op.drop_index(op.f('ix_books_author'), table_name='books')
|
||||||
|
op.drop_index(op.f('ix_books_title'), table_name='books')
|
||||||
|
op.drop_table('books')
|
@ -0,0 +1,26 @@
|
|||||||
|
from fastapi import APIRouter, Depends, HTTPException, status
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
from core.database import get_db
|
||||||
|
from schemas.book import BookCreate, BookSchema
|
||||||
|
from helpers.book_helpers import create_book, book_exists
|
||||||
|
|
||||||
|
router = APIRouter()
|
||||||
|
|
||||||
|
@router.post("/books", status_code=status.HTTP_201_CREATED, response_model=BookSchema)
|
||||||
|
async def add_book(
|
||||||
|
book_data: BookCreate,
|
||||||
|
db: Session = Depends(get_db)
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
Add a new book to the database using title and author
|
||||||
|
"""
|
||||||
|
# Check if book already exists
|
||||||
|
if book_exists(db, book_data.title, book_data.author):
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=status.HTTP_400_BAD_REQUEST,
|
||||||
|
detail="Book with this title and author already exists"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Create the book
|
||||||
|
new_book = create_book(db, book_data)
|
||||||
|
return new_book
|
50
helpers/book_helpers.py
Normal file
50
helpers/book_helpers.py
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
from typing import Optional
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
from models.book import Book
|
||||||
|
from schemas.book import BookCreate
|
||||||
|
|
||||||
|
def create_book(db: Session, book_data: BookCreate) -> Book:
|
||||||
|
"""
|
||||||
|
Creates a new book in the database.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
db (Session): The database session.
|
||||||
|
book_data (BookCreate): The data for the book to create.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Book: The newly created book object.
|
||||||
|
"""
|
||||||
|
db_book = Book(**book_data.dict())
|
||||||
|
db.add(db_book)
|
||||||
|
db.commit()
|
||||||
|
db.refresh(db_book)
|
||||||
|
return db_book
|
||||||
|
|
||||||
|
def find_book_by_title_and_author(db: Session, title: str, author: str) -> Optional[Book]:
|
||||||
|
"""
|
||||||
|
Finds a book by its title and author.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
db (Session): The database session.
|
||||||
|
title (str): The title of the book.
|
||||||
|
author (str): The author of the book.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[Book]: The book if found, otherwise None.
|
||||||
|
"""
|
||||||
|
return db.query(Book).filter(Book.title == title, Book.author == author).first()
|
||||||
|
|
||||||
|
def book_exists(db: Session, title: str, author: str) -> bool:
|
||||||
|
"""
|
||||||
|
Checks if a book with the given title and author already exists.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
db (Session): The database session.
|
||||||
|
title (str): The title of the book.
|
||||||
|
author (str): The author of the book.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: True if the book exists, False otherwise.
|
||||||
|
"""
|
||||||
|
book = find_book_by_title_and_author(db, title, author)
|
||||||
|
return book is not None
|
20
models/book.py
Normal file
20
models/book.py
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
from sqlalchemy import Column, String, DateTime, Text
|
||||||
|
from sqlalchemy.dialects.postgresql import UUID
|
||||||
|
from sqlalchemy.sql import func
|
||||||
|
from core.database import Base
|
||||||
|
import uuid
|
||||||
|
|
||||||
|
class Book(Base):
|
||||||
|
__tablename__ = "books"
|
||||||
|
|
||||||
|
# Primary key using UUID
|
||||||
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||||||
|
|
||||||
|
# Basic fields
|
||||||
|
title = Column(String, nullable=False, index=True)
|
||||||
|
author = Column(String, nullable=False, index=True)
|
||||||
|
description = Column(Text, nullable=True)
|
||||||
|
|
||||||
|
# Timestamps
|
||||||
|
created_at = Column(DateTime, default=func.now())
|
||||||
|
updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
|
@ -7,3 +7,6 @@ sqlalchemy>=1.4.0
|
|||||||
python-dotenv>=0.19.0
|
python-dotenv>=0.19.0
|
||||||
bcrypt>=3.2.0
|
bcrypt>=3.2.0
|
||||||
alembic>=1.13.1
|
alembic>=1.13.1
|
||||||
|
jose
|
||||||
|
passlib
|
||||||
|
pydantic
|
||||||
|
39
schemas/book.py
Normal file
39
schemas/book.py
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
from pydantic import BaseModel, Field
|
||||||
|
from typing import Optional
|
||||||
|
from datetime import datetime
|
||||||
|
from uuid import UUID
|
||||||
|
|
||||||
|
# Base schema for Book, used for inheritance
|
||||||
|
class BookBase(BaseModel):
|
||||||
|
title: str = Field(..., description="Book title")
|
||||||
|
author: str = Field(..., description="Book author")
|
||||||
|
description: Optional[str] = Field(None, description="Book description")
|
||||||
|
|
||||||
|
# Schema for creating a new Book
|
||||||
|
class BookCreate(BookBase):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Schema for updating an existing Book (all fields optional)
|
||||||
|
class BookUpdate(BaseModel):
|
||||||
|
title: Optional[str] = Field(None, description="Book title")
|
||||||
|
author: Optional[str] = Field(None, description="Book author")
|
||||||
|
description: Optional[str] = Field(None, description="Book description")
|
||||||
|
|
||||||
|
# Schema for representing a Book in responses
|
||||||
|
class BookSchema(BookBase):
|
||||||
|
id: UUID
|
||||||
|
created_at: datetime
|
||||||
|
updated_at: datetime
|
||||||
|
|
||||||
|
class Config:
|
||||||
|
orm_mode = True
|
||||||
|
schema_extra = {
|
||||||
|
"example": {
|
||||||
|
"id": "f47ac10b-58cc-4372-a567-0e02b2c3d479",
|
||||||
|
"title": "The Great Gatsby",
|
||||||
|
"author": "F. Scott Fitzgerald",
|
||||||
|
"description": "A novel about the American Dream",
|
||||||
|
"created_at": "2023-01-01T12:00:00",
|
||||||
|
"updated_at": "2023-01-01T12:00:00"
|
||||||
|
}
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user