Асинхронная миграция SQLAlchemy 2.0 с alembic, отношения между таблицами, не используя FastAPI – Видео 3

Асинхронная SQLAlchemy 2.0 является новой версией одной из самых популярных библиотек для работы с базами данных в Python. Она предлагает улучшенное асинхронное API для выполнения запросов к базам данных и работает с асинхронными фреймворками, такими как FastAPI.

В этом руководстве мы рассмотрим использование асинхронной SQLAlchemy 2.0 с alembic миграциями и отношениями между таблицами. Мы также обсудим, как создавать модели таблиц и выполнять запросы к базе данных, а также как настроить среду разработки для работы с этими инструментами.

1. Установка библиотек
   Для начала вам понадобится установить необходимые библиотеки. Для установки SQLAlchemy 2.0 и alembic выполните следующие команды:

pip install sqlalchemy sqlalchemy-async alembic

1. Настройка бд
   Далее вам необходимо настроить подключение к вашей базе данных. Для этого создайте файл config.py и добавьте в него следующий код:

from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
from sqlalchemy.orm import declarative_base, sessionmaker

DATABASE_URL = "postgresql+asyncpg://user:password@localhost/db_name"
engine = create_async_engine(DATABASE_URL, echo=True)

Base = declarative_base()
async_session = sessionmaker(bind=engine, class_=AsyncSession, expire_on_commit=False)

Замените значения user, password, localhost и db_name на соответствующие значения вашей базы данных.

1. Создание моделей таблиц
   Теперь создайте файл models.py и определите в нем модели таблиц. В примере ниже мы создадим модели для таблицы Users и Posts.

from sqlalchemy import Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship
from config import Base

class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String)
    email = Column(String, unique=True)

    posts = relationship("Post", back_populates="author")

class Post(Base):
    __tablename__ = "posts"

    id = Column(Integer, primary_key=True, index=True)
    title = Column(String)
    content = Column(String)
    author_id = Column(Integer, ForeignKey("users.id"))

    author = relationship("User", back_populates="posts")

1. Создание миграций
   Теперь давайте создадим миграции для наших моделей. Для этого выполните следующие команды:

alembic init alembic

Затем настройте файл alembic.ini, добавив путь к вашей базе данных в переменную sqlalchemy.url.

sqlalchemy.url = postgresql+asyncpg://user:password@localhost/db_name

Создайте миграцию:

alembic revision --autogenerate -m "initial migration"
alembic upgrade head

1. Выполнение запросов к базе данных
   Теперь мы можем выполнять запросы к базе данных с использованием асинхронной SQLAlchemy. Для этого создадим файл crud.py и добавим в него следующий код:

from sqlalchemy.ext.asyncio import AsyncSession
from models import User, Post
from config import async_session

async def create_user(user_data):
    async with async_session() as session:
        user = User(**user_data)
        session.add(user)
        await session.commit()
        return user

async def get_users():
    async with async_session() as session:
        return await session.execute(User.select()).scalars().all()

1. Настройка среды разработки
   Для того чтобы работать с асинхронной SQLAlchemy 2.0 в вашем проекте FastAPI, необходимо добавить пул подключений к базе данных в зависимости FastAPI.

from fastapi import FastAPI
from sqlalchemy.ext.asyncio import AsyncSession

app = FastAPI()

@app.middleware("http")
async def db_session_middleware(request, call_next):
    request.state.db = AsyncSession()
    response = await call_next(request)
    await request.state.db.close()
    return response

Теперь вы можете использовать асинхронную SQLAlchemy 2.0 в ваших FastAPI приложениях для работы с базами данных. Надеюсь, этот гайд был полезен для вас!

Видеоурок по данной теме можно посмотреть по ссылке: ссылка на видео