feat: Add new endpoints/books.post.py endpoint for Book 🚀

This commit is contained in:
Backend IM Bot 2025-04-26 08:19:32 +00:00
parent a273439de4
commit 5833164dd0
5 changed files with 136 additions and 0 deletions

View File

@ -0,0 +1,34 @@
"""create table for books
Revision ID: 2f7a3d8c2b62
Revises: 0001
Create Date: 2023-05-23 12:00:00
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.sql import func
# revision identifiers, used by Alembic.
revision = '2f7a3d8c2b62'
down_revision = '0001'
branch_labels = None
depends_on = None
def upgrade():
op.create_table(
'books',
sa.Column('id', sa.String(36), primary_key=True, server_default=sa.text("uuid_generate_v4()")),
sa.Column('title', sa.String(), nullable=False),
sa.Column('author', sa.String(), nullable=False),
sa.Column('description', sa.String(), nullable=True),
sa.Column('pages', sa.Integer(), nullable=False),
sa.Column('publisher_id', sa.String(36), nullable=False),
sa.Column('created_at', sa.DateTime(), server_default=func.now()),
sa.Column('updated_at', sa.DateTime(), server_default=func.now(), onupdate=func.now()),
sa.ForeignKeyConstraint(['publisher_id'], ['publishers.id'], ),
)
def downgrade():
op.drop_table('books')

View File

@ -0,0 +1,17 @@
from fastapi import APIRouter, status
from schemas.book import BookCreate, BookSchema
from helpers.book_helpers import create_book
from sqlalchemy.orm import Session
from fastapi import Depends
from core.database import get_db
router = APIRouter()
@router.post("/books", status_code=status.HTTP_201_CREATED, response_model=BookSchema)
async def create_new_book(
book: BookCreate,
db: Session = Depends(get_db)
):
"""Create a new book"""
new_book = create_book(db=db, book_data=book)
return new_book

35
helpers/book_helpers.py Normal file
View File

@ -0,0 +1,35 @@
from typing import Optional
from uuid import UUID
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 get_book_by_id(db: Session, book_id: UUID) -> Optional[Book]:
"""
Retrieves a single book by its ID.
Args:
db (Session): The database session.
book_id (UUID): The ID of the book to retrieve.
Returns:
Optional[Book]: The book object if found, otherwise None.
"""
return db.query(Book).filter(Book.id == book_id).first()

21
models/book.py Normal file
View File

@ -0,0 +1,21 @@
from sqlalchemy import Column, String, Integer, DateTime, ForeignKey
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func
from core.database import Base
import uuid
class Book(Base):
__tablename__ = "books"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
title = Column(String, nullable=False)
author = Column(String, nullable=False)
description = Column(String, nullable=True)
pages = Column(Integer, nullable=False)
publisher_id = Column(UUID(as_uuid=True), ForeignKey("publishers.id"), nullable=False)
publisher = relationship("Publisher", back_populates="books")
created_at = Column(DateTime, default=func.now())
updated_at = Column(DateTime, default=func.now(), onupdate=func.now())

29
schemas/book.py Normal file
View File

@ -0,0 +1,29 @@
from pydantic import BaseModel, Field
from typing import Optional
from datetime import datetime
from uuid import UUID
class BookBase(BaseModel):
title: str = Field(..., description="Book title")
author: str = Field(..., description="Book author")
description: Optional[str] = Field(None, description="Book description")
pages: int = Field(..., gt=0, description="Number of pages")
publisher_id: UUID = Field(..., description="Publisher ID")
class BookCreate(BookBase):
pass
class BookUpdate(BookBase):
title: Optional[str] = Field(None, description="Book title")
author: Optional[str] = Field(None, description="Book author")
description: Optional[str] = Field(None, description="Book description")
pages: Optional[int] = Field(None, gt=0, description="Number of pages")
publisher_id: Optional[UUID] = Field(None, description="Publisher ID")
class BookSchema(BookBase):
id: UUID
created_at: datetime
updated_at: datetime
class Config:
orm_mode = True