FastAPI如何优雅的连接数据库？
FastAPI如何优雅的连接数据库？
大家好，我是Kuls。
这是《小白学FastAPI》系列的第四篇文章。
在前面一篇文章中，我们有去写一个简单的博客框架，对它的路径、查询参数及路径参数函数等进行了学习。
今天我们来学习FastAPI的数据库连接，当然我们不仅仅只是为了学习这个知识点而学习。
今天K哥也来给大家说说如何去优雅的写一个FastAPI项目。
这是我今天所要讲解所编写项目的文件格式。
大家可以看到我们会在里面新建一个blog库来存放我们编写博客相关的操作，而不是像前面我们所写的，直接写一个main.py就行了。
那样对于我们编写一个项目是不太好的。
我们编写一个项目可以先创建一个虚拟环境，如果你还不懂虚拟环境是什么，可以去看下我的这篇文章 虚拟环境真的太重要了，很多人还不知道！
$ 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，里面填写这个项目要使用的第三方库。
fastapi
uvicorn
sqlalchemy
pymysql执行
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文件吗？我们将会在这里面编写数据库相关的内容：
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
既然数据库相关的配置都弄好了，接下来我们编写一个模型。
其实这个模型在我们上一篇文章中就已经编写过了。
但是还是有一些不同
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中所编写的
from pydantic import BaseModel
from typing import Optional
class Blog(BaseModel):
title: str
content: str
published: Optional[bool]这部分内容搬到了一个新的文件当中，方便我们统一管理。
main.py
其实main.py跟我们上一篇文章的中的main.py差不多，只是多了几行代码：
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