AsyncDatabase_API.py

from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker, AsyncSession
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
from sqlalchemy import select

from typing import Annotated
from pydantic import BaseModel, Field, ConfigDict
from fastapi import FastAPI, Depends



app = FastAPI()

# создаем движок бд и сессию для работы с ней 
engine = create_async_engine("sqlite+aiosqlite:///books.db")

# грубо говоря, сессия = транзакция
new_session = async_sessionmaker(engine, expire_on_commit=False)

async def get_session():
    async with new_session() as session:
        yield session
        
SessionDep = Annotated[AsyncSession, Depends(get_session)]





# модель таблицы для бд
class Base(DeclarativeBase):
    pass

class BookModel(Base):
    __tablename__ = "books"
    
    id: Mapped[int] = mapped_column(primary_key = True)
    title: Mapped[str]
    author: Mapped[str]




# ручка запуска бд
@app.post("/setup_database") 

async def setup_database():
    async with engine.begin() as conn:
        await conn.run_sync(Base.metadata.drop_all) # drop очищает бд
        await conn.run_sync(Base.metadata.create_all) # в метданные записываем всю инфу о бд и ее содержимом
    return {'ok': True}






class BookAddSchema(BaseModel): # айди не добавляем в поля, тк он задается базой автоматически
    title: str 
    author: str
    
    model_config = ConfigDict(extra="forbid")
    
    
class BookSchema(BaseModel):
    id: int
    
    
    
    
    

@app.post("/books")
async def add_book(data: BookAddSchema, session: SessionDep):
    new_book = BookModel(
        title=data.title,
        author=data.author,
    )
    session.add(new_book)
    await session.commit()
    return {"ok": True}
    
    
@app.get("/books")
async def get_books(session: SessionDep):
    query = select(BookModel)
    result = await session.execute(query)
    return result.scalars().all()
    
    
    
#   Когда работаем с pydantic чаще встречается Schema в названии
#   А при работе с SQLalchemy - Model (Например BookSchema и BookModel)
#   sql alchemy видео: артем шумейко курс SQLAlchemy #1, #13