FastAPI 学习之路(三十三)创建数据库

2021-09-17 11:12:50 浏览数 (1)

前言

我们之前分享分享使用FastAPI 学习之路(三十二)CORS(跨域资源共享),这次我们来看下创建数据库。

正文

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

  • PostgreSQL
  • MySQL
  • SQLite
  • Oracle

本文用SQLite为例。我们看下在fastapi是如何操作设计数据库的

代码语言:javascript复制
#这个安装依赖也可以
pip install sqlalchemy
#我在看到有支持异步的,我用的这个,但是这个没有上面的成熟
pip install fastapi-async-sqlalchemy

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

代码语言:javascript复制
SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db" #固定格式
engine = create_engine(
    SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
)
#connect_args 这个只有sqlite才用
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
#SessionLocal 类的每个实例都是一个数据库会话。 该类本身还不是数据库会话。
# 但是一旦我们创建了 SessionLocal 类的实例,这个实例将是实际的数据库会话。
Base = declarative_base()

我们用Base作为基础的,通过继承来创建每个数据库模型或类。我们看下

代码语言:javascript复制
class User(Base):
    __tablename__ = "users"
    id = Column(Integer, primary_key=True, index=True)
    email = Column(String, unique=True, index=True)
    hashed_password = Column(String)
    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, index=True)
    description = Column(String, index=True)
    owner_id = Column(Integer, ForeignKey("users.id"))
    owner = relationship("User", back_populates="items")

我们去创建了两个类,一个人,一个每一项,然后有对应的关联关系,这个表格怎么同步到数据库呢,其实很简单。

代码语言:javascript复制
Base.metadata.create_all(bind=engine)
#数据库初始化,如果没有库或者表,会自动创建

我们去启动下看看,是否可以创建

我们运行后,可以看到我们的数据库创建完毕了。

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

后记

发现问题,解决问题。遇到问题,慢慢解决问题即可。

欢迎关注雷子说测试开发,后续将会持续为大家分享更多的技术知识

0 人点赞