FastAPI+SQLAlchemy数据库连接

FastAPI+SQLAlchemy数据库连接

配置数据库连接

# db.py
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

DATABASE_URL = "mysql+pymysql://root:7997@localhost/fastapidemo"

engine = create_engine(
    DATABASE_URL,
    pool_size=10,            # 设置连接池大小
    max_overflow=20,         # 设置连接池最大溢出连接数量
    pool_timeout=30,         # 设置连接超时时间
    pool_recycle=1800,       # 设置连接的回收时间
)

SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

创建表模型

from sqlalchemy import Column, Integer, String, ForeignKey, Table
from sqlalchemy.orm import DeclarativeBase, relationship


class Base(DeclarativeBase):
    __abstract__ = True  # 标记为抽象基类，防止直接创建表

    @classmethod
    def get_all(cls, session):
        return session.query(cls).all