FastAPI 快速开发 Web API 项目: 连接 MySQL 数据库

2023-06-1017:46:00后端程序开发Comments1,235 views字数 4201阅读模式

介绍

FastAPI 不要求我们强制使用 SQL（关系）数据库。因此，我们可以根据需求使用任何关系数据库，例如：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

PostgreSQL文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
MySQL文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
SQLite文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
Oracle文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
Microsoft SQL Server文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
...文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

这篇文章以 MySQL 为例，首先需要为 MySQL 安装 Python 库，FastAPI 需要使用 Python 的 MySQL 客户端库来连接到 MySQL 数据库，这些驱动包括 mysql-connector-python 和 pymysql。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

安装这些库：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

pip install mysql-connector-python pymysql

安装成功如图所示：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

用于定义 FastAPI 依赖项的 Python 文件名没有特定的命名约定。通常，最好使用与应用程序的功能和角色匹配的名称命名，包含定义 FastAPI 依赖项的函数的 Python 文件。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

在终端中，运行以下命令以在 MySQL 数据库中创建数据库 example_db：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

//Login to MySQL
mysql -u root -p

//Create database named example_db
CREATE DATABASE example_db;

创建数据库表 users：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

CREATE TABLE
  `users` (
    `id` int unsigned NOT NULL AUTO_INCREMENT,
    `name` varchar(255) DEFAULT NULL,
    `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
    PRIMARY KEY (`id`)
  ) ENGINE = InnoDB DEFAULT CHARSET = utf8mb3

往数据库表中插入一条数据：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

INSERT INTO users (name) VALUES ("Atom");

db.py 文件

例如，通常命名一个包含依赖项函数的 Python 文件，该函数定义与上面示例中所示的 MySQL 数据库的连接，以指示它具有与数据库相关的功能，例如 db.py 或 database.py。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

此外，如果您有多个定义 FastAPI 依赖项的函数，或者如果您为不同功能定义了依赖项，则可以通过为每个功能指定不同的名称来提高代码的可读性。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

例如，请考虑以下文件名：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

db.py：包含定义与 MySQL 数据库连接的依赖项的文件。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
auth.py：定义与身份验证相关的依赖项的文件。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
validation.py：定义验证输入数据的依赖项的文件。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

根据项目的大小和结构，您还可以设计目录和模块的层次结构。通过将包含定义依赖项的函数的 Python 文件放置在适当的目录中并将它们组织为模块，可以改进代码管理和可维护性。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

将 MySQL 客户端库添加到 FastAPI 依赖项：使用 FastAPI 依赖项将 MySQL 客户端库添加到您的应用程序。例如，如果使用 mysql-connector-python，请按如下方式添加依赖项：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

$ touch routers/db.py

然后在 db.py 中写入如下代码：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

from fastapi import Depends
import mysql.connector


def get_db_connection():
    connection = mysql.connector.connect(
        host='localhost',
        port=3306,
        user="root",
        password="123456",
        database="example_db"
    )
    return connection

def get_db():
    connection = get_db_connection()
    db = connection.cursor()

    try:
        yield db
    finally:
        db.close()
        connection.close()

db_router 文件

在路由器上使用 MySQL 数据库：要将 MySQL 数据库与 FastAPI 的路由器一起使用，请使用上面定义的 get_db() 函数作为依赖项。下面是在 FastAPI 的路由器中使用 MySQL 数据库的示例。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

在将 MySQL 数据库与 FastAPI 路由器一起使用的示例 Python 文件名中，通常最好根据应用程序的功能和角色对其进行命名。你可以想到这样的文件名：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

main.py：包含示例代码的文件，该示例是应用程序的主要入口点，定义 FastAPI 路由器并使用 MySQL 数据库。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
router.py：定义 FastAPI 路由器并包含使用 MySQL 数据库的示例的代码的文件。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
db.py：包含用于连接和查询 MySQL 数据库的函数的文件。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

根据项目的大小和结构，您还可以设计目录和模块的层次结构。例如，您可以通过创建名为路由器的目录，在其中创建定义 FastAPI 路由器的模块，并在其中包含使用 MySQL 数据库的示例代码来提高代码组织和可读性。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

例如，可以考虑以下文件名和目录结构：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

main.py：作为应用程序主入口点的文件，导入并使用路由器目录中的路由器模块。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
routers/db_router.py：定义使用 MySQL 数据库的示例路由器的模块。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html
routers/db.py：定义用于连接和查询MySQL数据库的函数的模块。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

db_router.py 文件写入如下内容：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

from fastapi import FastAPI, Depends
from mysql.connector import cursor
from db import get_db
import json


app = FastAPI()

# def get_db(db: cursor.MySQLCursor = Depends(get_db)):
#     return db

@app.get("/users/")
async def get_users(db: cursor.MySQLCursor = Depends(get_db)):
    
    query = "SELECT * FROM users"
    db.execute(query)
    result = db.fetchall()
    if result:
        return {"users": result}
    else:
        return {"error": "User not found"}
    
@app.get("/users/{user_id}")
async def get_user(user_id: int,
                   db: cursor.MySQLCursor = Depends(get_db)):
    query = "SELECT * FROM users WHERE id = %s"
    db.execute(query, (user_id,))
    result = db.fetchall()
    if result:
        return {"user_id": result[0][0], "username": result[0][1]}
    else:
        return {"error": "User not found"}
    
@app.get("/user_name/{user_name}")
async def insert_user(user_name: str,
                      db: cursor.MySQLCursor = Depends(get_db)):
    query = "INSERT INTO users (name) VALUES (%s)"
    db.execute(query, (user_name,))
    result = db.fetchone()
    db.execute("COMMIT")
    return {"user_name": user_name}

启动 FastAPI 服务

uvicorn db_router:app --reload

INFO:     Will watch for changes in these directories: ['/Users/sharland/dev/fastapi/MySQL/routers']
INFO:     Uvicorn running on http://127.0.0.1:8000 (Press CTRL+C to quit)
INFO:     Started reloader process [60501] using WatchFiles
INFO:     Started server process [60503]
INFO:     Waiting for application startup.
INFO:     Application startup complete.

当您在浏览器中访问 http://127.0.0.1:8000/docs 时，将显示一个屏幕，允许您使用 GUI 操作快速 API 服务。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

访问 http://127.0.0.1:8000/users/，可以获取当前数据库中的所有用户数据：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

"users":[[1,"Atom","2023-04-20T13:14:00"],[2,"Nuoza","2023-04-20T13:14:20"]]}

当访问 http://127.0.0.1:8000/users/1 时，结果：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

{"user_id":1,"username":"Atom"}

新增一个用户，http://127.0.0.1:8000/user_name/Dole。操作如图：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html

总结

本文总结了通过 MySQL 驱动连接 Python，实现了一个 FastAPI 连接 MySQL 数据库的示例，并结合代码结构设计了目录结构和实现了获取用户和增加用户的功能，但是本文只是简单使用了原生的 MySQL，其实 FastAPI 支持结合 SQLAlchemy 更方面创建模型的数据库语句，今后会继续沿用这个示例写用户的登录和注册功能，敬请期待。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/46835.html