FastAPI 学习之路（三十二）创建数据库

在大型的web开发中，我们肯定会用到数据库操作，那么FastAPI也支持数据库的开发，你可以用PostgreSQL，MySQL，SQLite，Oracle等

本文使用Mysql为例，我们看下在fastapi中是如何操作设计数据库的

 安装依赖

pip install sqlalchemy

也有支持异步的，但是没有上面那个成熟

pip install fastapi-async-sqlalchemy

我们看下如何在项目中使用

 

from sqlalchemy import create_engine, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker, relationship
from sqlalchemy import Column, Integer, String, Boolean

mysql_url = "mysql+pymysql://{username}:{password}@{host}:{port}/{database}?charset=utf8".format(
    username="root", password="xxx", host="xxx", port=3306, database="sqlalchemy_db")
# SQLALCHEMY_DATABASE_URL = "sqlite:///.test.db"
engine = create_engine(
    mysql_url,
    # SQLALCHEMY_DATABASE_URL,
    # connect_args 参数只有sqlite才有
    # connect_args = {
    #     "check_same_thread": False
    # }
)

# 该类的每个实例都是一个数据库会话，该类本身还不是数据库会话，但是一旦我们创建了SessionLocal的实例，这个实例将是实际的数据库会话
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

# 创建数据库基类
Base = declarative_base()


# 创建两个数据库模型，继承自Base基类
class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    email = Column(String(10), unique=True, index=True)
    hashed_password = Column(String(10))
    is_active = Column(Boolean, default=True)
    items = relationship("Item", back_populates="owner")

class Item(Base):
    __tablename__ = "items"

    id = Column(Integer, primary_key=True, index=True)
    title = Column(String(10), index=True)
    description = Column(String(10), index=True)
    owner_id = Column(Integer, ForeignKey("users.id"))
    owner = relationship("User", back_populates="items")


# 那么我们如何将该模型对应到数据库表呢? 很简单
Base.metadata.create_all(bind=engine)

 可以看到表结构和我们设计的是一样的 ，这样我们就完成了创建数据库的操作，我们接下来就是看，如何去操作数据库