本文共 13043 字，大约阅读时间需要 43 分钟。

实验前提：

linux虚拟环境和连接pycharm操作以及mysql数据库安装部署已经完成

实验步骤：

1.安装sqlalchemy 模块

# 安装sqlalchemy模块(venv) [root@localhost flast]# pip install sqlalchemy -i https://pypi.tuna.tsinghua.edu.cn/simple# 测试是否安装成功(venv) [root@localhost flast]# pip list

2.创建数据库

(venv) [root@localhost flast]# mysql -uroot -pwestosMariaDB [(none)]> create database fastapi charset=utf8;

3.连接数据库

新建 连接数据库.py文件，并写入：

# 导入FastAPI模块from fastapi import FastAPI# 创建app实例app = FastAPI()## 连接数据库from sqlalchemy import create_enginefrom sqlalchemy.orm import sessionmaker# 配置数据库地址：数据库类型+数据库驱动名称://用户名:密码@机器地址:端口号/数据库名engine =create_engine("mysql+pymysql://root:westos@localhost:3306/fastapi",encoding='utf-8')# 把当前的引擎绑定给这个会话；# autocommit：是否自动提交 autoflush：是否自动刷新并加载数据库 bind：绑定数据库引擎Session = sessionmaker(autocommit=False, autoflush=False, bind=engine)# 实例化session = Session()# declarative_base类维持了一个从类到表的关系，通常一个应用使用一个Base实例，所有实体类都应该继承此类对象from sqlalchemy.ext.declarative import declarative_baseBase = declarative_base()from sqlalchemy import Column, String, Integer# 创建数据库模型（定义表结构:表名称，字段名称以及字段类型）class User(Base):    # 定义表名    __tablename__ = 'tbluser'    # 定义字段    # primary_key=True 设置为主键    userid = Column(Integer,primary_key=True)    username = Column(String(255))    # 构造函数    def __init__(self,userid,username):        self.userid = userid        self.username = username    # 打印形式    def __str__(self):        return "id:%s, name:%s" % (str(self.userid) ,self.username)# 在数据库中生成表Base.metadata.create_all(bind=engine)

测试：

(venv) [root@localhost flast]# uvicorn 连接数据库:app --host '192.168.56.20' --port 8080 --reload

报错：ModuleNotFoundError: No module named 'pymysql'

解决方法：

(1).安装pymysql

(venv) [root@localhost flast]# pip install pymysql -i https://pypi.tuna.tsinghua.edu.cn/simple

(2).导入pymysql

import pymysqlpymysql.install_as_MySQLdb()

完整代码如下：

# 导入FastAPI模块from fastapi import FastAPI# 创建app实例app = FastAPI()## 连接数据库# 连接mysql数据库需要导入pymysql模块import pymysqlpymysql.install_as_MySQLdb()from sqlalchemy import create_enginefrom sqlalchemy.orm import sessionmaker# 配置数据库地址：数据库类型+数据库驱动名称://用户名:密码@机器地址:端口号/数据库名engine =create_engine("mysql+pymysql://root:westos@localhost:3306/fastapi",encoding='utf-8')# 把当前的引擎绑定给这个会话；# autocommit：是否自动提交 autoflush：是否自动刷新并加载数据库 bind：绑定数据库引擎Session = sessionmaker(autocommit=False, autoflush=False, bind=engine)# 实例化session = Session()# declarative_base类维持了一个从类到表的关系，通常一个应用使用一个Base实例，所有实体类都应该继承此类对象from sqlalchemy.ext.declarative import declarative_baseBase = declarative_base()from sqlalchemy import Column, String, Integer# 创建数据库模型（定义表结构:表名称，字段名称以及字段类型）class User(Base):    # 定义表名    __tablename__ = 'tbluser'    # 定义字段    # primary_key=True 设置为主键    userid = Column(Integer,primary_key=True)    username = Column(String(255))    # 构造函数    def __init__(self,userid,username):        self.userid = userid        self.username = username    # 打印形式    def __str__(self):        return "id:%s, name:%s" % (str(self.userid) ,self.username)# 在数据库中生成表Base.metadata.create_all(bind=engine)

测试：

(venv) [root@localhost flast]# uvicorn 连接数据库:app --host '192.168.56.20' --port 8080 --reload

检查fastapi数据库中是否已经生成了tbluser表

4.添加数据

(1).添加单个

from pydantic import BaseModel# 定义数据模型class CreatUser(BaseModel):    userid: int    username: str    def __str__(self):        return "id：%s, name：%s" % (str(self.userid), self.username)## 添加单个@app.post("/user/addUser")async def InserUser(user: CreatUser):    try:        # 添加数据        dataUser = User(userid=user.userid, username=user.username)        session.add(dataUser)        session.commit()        session.close()    except ArithmeticError:        return {"code":"0002","message":"数据库异常"}    return {"code":"0000","message":"添加成功"}

在linux命令行启动后，Postman测试：

(2).添加多个

from typing import List## 添加多个@app.post("/user/addUserList")async def addUserList(*, user: List[CreatUser]):    try:        # user是一个列表，每个内部元素均为CreatUser类型        for u in user:            # 自定义的数据模型可以用.访问属性            dataUser = User(userid=u.userid, username=u.username)            session.add(dataUser)        session.commit()        session.close()    except ArithmeticError:        return {"code": "0002", "message": "数据库异常"}    return {"code": "0000", "message": "添加成功"}

在linux命令行启动后，Postman测试：

5.查询数据

one与first的区别：

   one：要求结果集中只有一个结果；如果数据库返回0或2个或更多结果，并且将引发异常，则为错误。    first：返回可能更大的结果集中的第一个，如果没有结果，则返回None。不会引发异常。filter_by与filter的区别：    filter_by接收的参数形式是关键字参数，而filter接收的参数是更加灵活的SQL表达式结构

## 按照user_id查询@app.get("/user/{user_id}")async def queryUserByUserId(user_id: int):    # 创建Query查询，filter是where条件，调用one返回唯一行，调用all则是返回所有行    try:        # user1 = session.query(User).filter_by(userid=user_id).first()        # user1 = session.query(User).filter(User.userid==user_id).one()        user1 = session.query(User).filter(User.userid==user_id).first()        session.close()        # 由于user1只有一个值，所以它直接是一个字典        if user1:            return {"code":"0000","message":"请求成功","data":user1}        else:            return {"code":"0001","message":"查询无结果"}    except ArithmeticError:            return {"code":"0002","message":"数据库异常"}

## 查询所有@app.get("/user/selectall/")async def queryUserByUserId():    # 创建Query查询，filter是where条件，调用one返回唯一行，调用all则是返回所有行    try:        user1 = session.query(User).all()        session.close()        # user1 是一个列表，内部元素为字典        return {"code": "0000", "message": "请求成功", "data": user1}    except ArithmeticError:            return {"code":"0002","message":"数据库异常"}

6.删除数据

(1).删除单个

@app.delete("/user/deleteUser/{user_id}")async def deleteUser(user_id: int):    try:        user1 = session.query(User).filter(User.userid == user_id).first()        if user1:            session.delete(user1)            session.commit()            session.close()            return {"code": "0000", "message": "删除成功"}        else:            return {"code": "0001", "message": "参数错误"}    except ArithmeticError:        return {"code": "0002", "message": "数据库错误"}

(2).删除多个

from typing import List# 删除多个@app.delete("/user/deleteUserList")async def deleteUser(user_ids: List[int]):    try:        for user_id in user_ids:            user1 = session.query(User).filter(User.userid == user_id).first()            if user1:                session.delete(user1)                session.commit()                session.close()        return {"code": "0000", "message": "删除成功"}    except ArithmeticError:        return {"code": "0002", "message": "数据库错误"}

7.修改数据

方法1：

## 根据user_id修改user_name@app.put("/user/updateUser/")# 定义查询参数user_id和user_nameasync def deleteUser(user_id: int, user_name: str):    try:        user1 = session.query(User).filter(User.userid == user_id).first()        if user1:            user1.username = user_name            session.commit()            session.close()            return {"code": "0000", "message": "修改成功"}        else:            return {"code": "0001", "message": "参数错误"}    except ArithmeticError:        return {"code": "0002", "message": "数据库错误"}

方法2：

class AlterUser(BaseModel):    userid: int    username: str@app.put("/user/updateUser01/")async def deleteUser(user: AlterUser):    try:        user1 = session.query(User).filter(User.userid == user.userid).first()        if user1:            user1.username = user.username            session.commit()            session.close()            return {"code": "0000", "message": "修改成功"}        else:            return {"code": "0001", "message": "参数错误"}    except ArithmeticError:        return {"code": "0002", "message": "数据库错误"}

完整代码如下：

# 导入FastAPI模块from fastapi import FastAPI# 创建app实例app = FastAPI()## 连接数据库# 连接mysql数据库需要导入pymysql模块import pymysqlpymysql.install_as_MySQLdb()from sqlalchemy import create_enginefrom sqlalchemy.orm import sessionmaker# 配置数据库地址：数据库类型+数据库驱动名称://用户名:密码@机器地址:端口号/数据库名engine =create_engine("mysql+pymysql://root:westos@localhost:3306/fastapi",encoding='utf-8')# 把当前的引擎绑定给这个会话；# autocommit：是否自动提交 autoflush：是否自动刷新并加载数据库 bind：绑定数据库引擎Session = sessionmaker(autocommit=False, autoflush=False, bind=engine)# 实例化session = Session()# declarative_base类维持了一个从类到表的关系，通常一个应用使用一个Base实例，所有实体类都应该继承此类对象from sqlalchemy.ext.declarative import declarative_baseBase = declarative_base()from sqlalchemy import Column, String, Integer# 创建数据库模型（定义表结构:表名称，字段名称以及字段类型）class User(Base):    # 定义表名    __tablename__ = 'tbluser'    # 定义字段    # primary_key=True 设置为主键    userid = Column(Integer,primary_key=True)    username = Column(String(255))    # 构造函数    def __init__(self,userid,username):        self.userid = userid        self.username = username    # 打印形式    def __str__(self):        return "id：%s, name：%s" % (str(self.userid) ,self.username)    # 定义返回结果    def to_dict(self):        return {            "userid": self.userid,            "username": self.username        }# 在数据库中生成表Base.metadata.create_all(bind=engine)### 添加数据from pydantic import BaseModel# 定义数据模型class CreatUser(BaseModel):    userid: int    username: str    def __str__(self):        return "id：%s, name：%s" % (str(self.userid), self.username)## 添加单个@app.post("/user/addUser")async def InserUser(user: CreatUser):    try:        # 添加数据        dataUser = User(userid=user.userid, username=user.username)        session.add(dataUser)        session.commit()        session.close()    except ArithmeticError:        return {"code":"0002","message":"数据库异常"}    return {"code":"0000","message":"添加成功"}from typing import List## 添加多个@app.post("/user/addUserList")async def addUserList(*, user: List[CreatUser]):    try:        # user是一个列表，每个内部元素均为CreatUser类型        for u in user:            # 自定义的数据模型可以用.访问属性            dataUser = User(userid=u.userid, username=u.username)            session.add(dataUser)        session.commit()        session.close()    except ArithmeticError:        return {"code": "0002", "message": "数据库异常"}    return {"code": "0000", "message": "添加成功"}### 查询## 按照user_id查询@app.get("/user/{user_id}")async def queryUserByUserId(user_id: int):    # 创建Query查询，filter是where条件，调用one返回唯一行，调用all则是返回所有行    try:        # one与first的区别：        # one：要求结果集中只有一个结果；如果数据库返回0或2个或更多结果，并且将引发异常，则为错误。        # first：返回可能更大的结果集中的第一个，如果没有结果，则返回None。不会引发异常。        # filter_by与filter的区别：        # filter_by接收的参数形式是关键字参数，而filter接收的参数是更加灵活的SQL表达式结构        # user1 = session.query(User).filter_by(userid=user_id).first()        user1 = session.query(User).filter(User.userid==user_id).first()        session.close()        # 由于user1只有一个值，所以它直接是一个字典        if user1:            return {"code":"0000","message":"请求成功","data":user1}        else:            return {"code":"0001","message":"查询无结果"}    except ArithmeticError:            return {"code":"0002","message":"数据库异常"}## 查询所有@app.get("/user/selectall/")async def queryUserByUserId():    # 创建Query查询，filter是where条件，调用one返回唯一行，调用all则是返回所有行    try:        user1 = session.query(User).all()        session.close()        # user1 是一个列表，内部元素为字典        return {"code": "0000", "message": "请求成功", "data": user1}    except ArithmeticError:            return {"code":"0002","message":"数据库异常"}###删除# 根据user_id删除单个@app.delete("/user/deleteUser/{user_id}")async def deleteUser(user_id: int):    try:        user1 = session.query(User).filter(User.userid == user_id).first()        if user1:            session.delete(user1)            session.commit()            session.close()            return {"code": "0000", "message": "删除成功"}        else:            return {"code": "0001", "message": "参数错误"}    except ArithmeticError:        return {"code": "0002", "message": "数据库错误"}from typing import List## 删除多个@app.delete("/user/deleteUserList")async def deleteUser(user_ids: List[int]):    try:        for user_id in user_ids:            user1 = session.query(User).filter(User.userid == user_id).first()            if user1:                session.delete(user1)                session.commit()                session.close()        return {"code": "0000", "message": "删除成功"}    except ArithmeticError:        return {"code": "0002", "message": "数据库错误"}###修改## 根据user_id修改user_name@app.put("/user/updateUser/")# 定义查询参数user_id和user_nameasync def updateUser(user_id: int, user_name: str):    try:        user1 = session.query(User).filter(User.userid == user_id).first()        print(user1)        if user1:            user1.username = user_name            session.commit()            session.close()            return {"code": "0000", "message": "修改成功"}        else:            return {"code": "0001", "message": "参数错误"}    except ArithmeticError:        return {"code": "0002", "message": "数据库错误"}# 方式2：class AlterUser(BaseModel):    userid: int    username: str@app.put("/user/updateUser01/")async def deleteUser(user: AlterUser):    try:        user1 = session.query(User).filter(User.userid == user.userid).first()        if user1:            user1.username = user.username            session.commit()            session.close()            return {"code": "0000", "message": "修改成功"}        else:            return {"code": "0001", "message": "参数错误"}    except ArithmeticError:        return {"code": "0002", "message": "数据库错误"}# if __name__ == '__main__':#     import uvicorn#     uvicorn.run(app=app, host="192.168.56.20", port=8000, debug=True)

 

转载地址：http://oildz.baihongyu.com/