MongoDB和Flask：如何存储和检索数据

1. 安装与导入

在开始使用MongoDB和Flask之前，需要确保MongoDB已经安装并运行在本地环境中。然后安装Flask扩展pymongo，可以使用pip命令进行安装：

pip install pymongo

在Flask应用程序的入口模块中，通过导入pymongo扩展来利用MongoDB进行数据存储和检索：

from flask import Flask
from pymongo import MongoClient

app = Flask(__name__)

client = MongoClient()
db = client.my_database
collection = db.my_collection

@app.route('/')
def index():
    return 'Hello World!'

2. 存储数据

在MongoDB中，数据存储在集合(collection)中，集合类似于关系数据库中的表，但是MongoDB不需要事先定义表结构，可以动态地添加字段。

为了将数据存储到MongoDB中，可以通过调用集合对象的insert_one()方法，将一个文档(document)插入到集合中。文档是MongoDB中的基本单位，类似于一条记录。

在Flask应用程序中，可以使用Flask的HTTP请求处理装饰器来定义路由和处理函数。在处理函数中，可以通过request对象的属性来获取请求中的参数和数据。

下面是一个将用户的姓名、年龄和地址存储到MongoDB中的示例：

from flask import request

@app.route('/add_user', methods=['POST'])
def add_user():
    name = request.form['name']
    age = request.form['age']
    address = request.form['address']
    result = collection.insert_one({'name': name, 'age': age, 'address': address})
    return 'Inserted user with id: %s' % result.inserted_id

在上面的示例中，使用了request对象的form属性来获取通过POST请求发送的表单数据。将用户的姓名、年龄和地址存储到一个字典中，然后调用insert_one()方法将字典插入到集合中，并返回插入的文档的_id属性。

3. 检索数据

MongoDB提供了强大的文档查询功能，可以在集合中快速地检索数据。可以使用pymongo扩展提供的查询方法来实现在Flask应用程序中检索数据。

在下面的示例中，使用了find_one()方法和find()方法来获取集合中的数据。可以使用字典参数指定查询条件和限制字段。在Flask应用程序中，可以通过HTTP请求的query string或者使用request对象的args属性来传递查询条件。

@app.route('/get_user', methods=['GET'])
def get_user():
    name = request.args.get('name')
    result = collection.find_one({'name': name}, {'_id': 0})
    if result:
        return str(result)
    else:
        return 'No result found'

@app.route('/get_users', methods=['GET'])
def get_users():
    age = request.args.get('age')
    results = collection.find({'age': age}, {'_id': 0, 'name': 1})
    return str(list(results))

在上面的示例中，使用了find_one()方法和find()方法来获取集合中的数据。使用find_one()方法可以获取一个满足条件的文档，使用find()方法可以获取多个满足条件的文档。使用字典参数指定查询条件和限制字段，使用0表示不包含该字段，使用1表示包含该字段。使用list()函数将查询结果转换为列表形式，最终返回一个字符串。

4. 总结

在Flask应用程序中使用MongoDB进行数据存储和检索，可以使用pymongo扩展提供的方法来实现。通过调用集合对象的insert_one()方法将文档插入到集合中，通过调用find_one()方法和find()方法来获取集合中的数据。在处理HTTP请求时，可以使用request对象的属性来获取请求中的参数和数据。