The Way to Flask 项目教程：Flask 与 MongoDB 数据库集成指南

2025-06-19 02:56:53作者：鲍丁臣Ursa

前言

在现代 Web 开发中，数据库是不可或缺的组成部分。本文将详细介绍如何在 Flask 项目中集成 MongoDB 数据库，使用 Flask-MongoEngine 扩展来简化数据库操作。这是 The Way to Flask 项目系列教程中的重要章节，将帮助开发者从文件存储升级到专业的数据库存储方案。

为什么选择 MongoDB

MongoDB 是一种流行的 NoSQL 数据库，特别适合以下场景：

数据结构灵活多变，不需要预先定义严格的表结构
需要处理大量非结构化或半结构化数据
开发迭代速度快，需要频繁修改数据模型
需要高性能的读写操作

在 Flask 生态中，Flask-MongoEngine 是基于 MongoEngine 的扩展，提供了与 Flask 框架无缝集成的能力。

环境准备

在开始之前，请确保已安装以下组件：

MongoDB 3.2.6 或更高版本（运行在 localhost:27017）
Flask-MongoEngine 扩展
Python 3.x 环境

基础集成步骤

1. 配置 MongoDB 连接

首先需要在 Flask 应用中配置 MongoDB 的连接信息：

app = Flask(__name__)
app.config['MONGODB_SETTINGS'] = {
    'db': 'the_way_to_flask',  # 数据库名称
    'host': 'localhost',       # 主机地址
    'port': 27017              # 端口号
}

db = MongoEngine()
db.init_app(app)

2. 定义数据模型

创建一个简单的用户模型，包含 name 和 email 两个字段：

class User(db.Document):
    name = db.StringField()
    email = db.StringField()
    
    def to_json(self):
        return {
            "name": self.name,
            "email": self.email
        }

这里我们定义了：

StringField: 字符串类型字段
to_json(): 将模型对象转换为 JSON 格式的方法

数据库操作详解

1. 查询数据

MongoEngine 提供了灵活的查询接口：

# 查询单个用户
user = User.objects(name="zhangsan").first()

# 查询所有用户
users = User.objects.all()

# 复杂查询
users = User.objects(name__ne="zhangsan")  # 名字不等于zhangsan
users = User.objects(email__contains="gmail")  # 邮箱包含gmail

2. 新增数据

创建并保存新记录非常简单：

new_user = User(name="lisi", email="lisi@gmail.com")
new_user.save()  # 保存到数据库

3. 更新数据

更新现有记录的两种方式：

# 方式1：查询后更新
user = User.objects(name="zhangsan").first()
user.update(email="new_email@example.com")

# 方式2：直接批量更新
User.objects(name="zhangsan").update(email="new_email@example.com")

4. 删除数据

删除记录同样简单：

user = User.objects(name="zhangsan").first()
user.delete()  # 删除该记录

# 或者批量删除
User.objects(name="zhangsan").delete()

构建 RESTful API

将数据库操作集成到 REST 服务中：

@app.route('/', methods=['GET'])
def query_records():
    name = request.args.get('name')
    user = User.objects(name=name).first()
    return jsonify(user.to_json()) if user else jsonify({'error': 'Not found'})

@app.route('/', methods=['PUT'])
def create_record():
    data = request.get_json()
    user = User(name=data['name'], email=data['email']).save()
    return jsonify(user.to_json())

@app.route('/', methods=['POST'])
def update_record():
    data = request.get_json()
    user = User.objects(name=data['name']).first()
    if user:
        user.update(email=data['email'])
    return jsonify(user.to_json()) if user else jsonify({'error': 'Not found'})

@app.route('/', methods=['DELETE'])
def delete_record():
    data = request.get_json()
    user = User.objects(name=data['name']).first()
    if user:
        user.delete()
        return jsonify({'message': 'Deleted'})
    return jsonify({'error': 'Not found'})