FastAPI如何优雅的连接数据库?

2022-03-14 19:48:00 浏览数 (1)

大家好,我是Kuls。

这是《小白学FastAPI》系列的第四篇文章。

在前面一篇文章中,我们有去写一个简单的博客框架,对它的路径、查询参数及路径参数函数等进行了学习。

今天我们来学习FastAPI的数据库连接,当然我们不仅仅只是为了学习这个知识点而学习。

今天K哥也来给大家说说如何去优雅的写一个FastAPI项目。

这是我今天所要讲解所编写项目的文件格式。

大家可以看到我们会在里面新建一个blog库来存放我们编写博客相关的操作,而不是像前面我们所写的,直接写一个main.py就行了。

那样对于我们编写一个项目是不太好的。

我们编写一个项目可以先创建一个虚拟环境,如果你还不懂虚拟环境是什么,可以去看下我的这篇文章 虚拟环境真的太重要了,很多人还不知道!

代码语言:javascript复制
$ cd demo3-blog/
$ ls
__init__.py             blog                    requirements.txt

$ python -m venv blogenv
$ ls
__init__.py             blog                    blogenv                 requirements.txt

$ source blogenv/bin/activate

随后创建一个requeriment.txt,里面填写这个项目要使用的第三方库。

代码语言:javascript复制
fastapi
uvicorn
sqlalchemy
pymysql

执行

代码语言:javascript复制
pip install -r requeriment.txt

好了,上面其实算是题外话。

今天我们的主角其实是SQLAlchemy,可能之前学过Django或Flask的同学应该接触过SQLAlchemy。

我们看下官网的解释:SQLAlchemy is the Python SQL toolkit and Object Relational Mapper that gives application developers the full power and flexibility of SQL.

简答理解,它是一个sql工具箱,是一个ORM框架。

ORM:对象关系映射,你可以简单理解为 Python中的一个类映射一张数据表。

其实关于SQLAlchemy,里面有很多的知识,我也把它的详细的基础使用方法链接给大家。

官方文档:https://docs.sqlalchemy.org/en/14/orm/tutorial.html

本篇文章中我们也会学习SQLAlchemy。

database.py

还记得我们创建的database.py文件吗?我们将会在这里面编写数据库相关的内容:

代码语言:javascript复制
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

SQLALCHEMY_DATABASE_URL = 'mysql pymysql://user:password@localhost/fastapi'
engine = create_engine(SQLALCHEMY_DATABASE_URL)
SessionLocal = sessionmaker(bind=engine, autocommit=False, autoflush=False)
Base = declarative_base()

这里我们一行一行来进行分析

engine = create_engine(SQLALCHEMY_DATABASE_URL)'

这一行创建了我们的数据库引擎,我们就是通过这个引擎来创建表等各种操作的。其中的参数就是我们数据库的连接url,fastapi支持的数据库非常的多。

这里提醒一下,如果我们使用的是sqlite数据库,我们需要在create_engine的参数中新增connect_args={"check_same_thread": False}

下面也是官网给出的支持的数据库,这些我们都可以通过sqlalchemy来进行连接。

•PostgreSQL•MySQL•SQLite•Oracle•Microsoft SQL Server, etc.

关于具体数据库的url是啥,这里我给出官网,大家可以进行查阅

https://docs.sqlalchemy.org/en/14/core/engines.html

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

这里我们创建的是SessionLocal,后续我们用到会进行讲解。

Base = declarative_base()

通过declarative_base()方法生成的类,该类是我们后面要用到的ORM 模型所要继承的父类。

models.py

既然数据库相关的配置都弄好了,接下来我们编写一个模型。

其实这个模型在我们上一篇文章中就已经编写过了。

但是还是有一些不同

代码语言:javascript复制
from sqlalchemy import Column, Integer, String, Boolean
from .database import Base

class Blog(Base):
    __tablename__ = 'blog'

    id = Column(Integer, primary_key=True, index=True) # 设置主键和索引
    title = Column(String(32))
    content = Column(String(32))
    pulished = Column(Boolean)

大家可以发现,其实非常的直观和简单,无非就是创建一个类,里面有一些字段变量。

其中的__tablename__表示的是我们待会创建数据表的名称。

schemas.py

这个其实很简单,就是将我们在main.py中所编写的

代码语言:javascript复制
from pydantic import BaseModel
from typing import Optional

class Blog(BaseModel):
    title: str
    content: str
    published: Optional[bool]

这部分内容搬到了一个新的文件当中,方便我们统一管理。

main.py

其实main.py跟我们上一篇文章的中的main.py差不多,只是多了几行代码:

代码语言:javascript复制
from fastapi import FastAPI
from .database import engine
from . import schemas,models
from typing import Optional
app = FastAPI()

models.Base.metadata.create_all(engine)

# 博客首页
@app.get('/blog')
def index(limit: int = 10, published: bool = True, sort: Optional[str] = None):
    return {'data': f'我是博客首页,显示{limit}篇内容,并且发布状态为{published},排序顺序是根据{sort}字段'}

@app.get('/blog/unpublished')
def unpublished():
    return {'data': '这里是没有发布的博文列表'}

@app.get('/blog/{id}')
def showblog(id: int):
    return {'data': f'这是id为 {id} 的博文'}

@app.get('/blog/{id}/comments')
def comments(id: int):
    return {'data': f'这是id为{id}的博文评论内容'}

@app.post('/blog')
def new_blog(blog: schemas.Blog):
    return {'data': f'博文标题:{blog.title},博文内容:{blog.content},博文发表状态:{blog.published}'}

其中的create_all()方法就帮助我们创建了数据表

运行

到了这里基本就大功告成。

我们直接运行

查看数据库,发现我们的数据表已经成功创建。

总结

好了,今天主要讲了数据库相关的操作以及编写项目时的注意事项。

数据库具体的读写删操作,我们会在后面继续写,也会基于这个blog来写。

整个系列的代码我都放在了github中,大家可以访问下面链接:

https://github.com/hellokuls/fastapi/tree/master

代码语言:javascript复制

0 人点赞