当前位置：网站首页>Flask框架——MongoEngine使用MongoDB数据库

from flask import Flask
from flask_mongoengine import MongoEngine

app = Flask(__name__)

app.config['MONGODB_SETTINGS']={    #通过MONGODB_SETTINGS配置MongoEngine
    'db':'myproject'
}
db=MongoEngine(app)          #初始化MongoEngine配置

if __name__ == '__main__':
    app.run()

MONGODB_SETTINGS配置中，我们只添加了db参数——将要访问的数据库名称，除了这个参数，我们还可以添加以下参数：

alias：建立的数据库连接的别名，默认为：default；
db：访问的数据库名，默认为test；
host：MongoDB服务器地址，默认为localhost；
port：MongoDB服务器端口号，默认为27017；
username：用户名；
password：用户密码；
authentication_source：认证源，创建该用户的数据库；
is_mock：是否使用 mongomock；
connect：是否直接连接服务器，如果为false，则直到第一次操作时才会连接服务器；
tz_aware：是否自动识别时区，如果为false，则直接使用本地时间，忽略datetime的时区配置。

除了通过MONGODB_SETTINGS对MongoEngine配置，我们还可以使用"MONGODB_"来单独提供连接配置，示例代码如下：

app.config['MONGODB_DB'] = 'test'      #数据库名称
app.config['MONGODB_HOST'] = 'localhost'    #服务器地址
app.config['MONGODB_PORT'] = 27017      #服务器端口号
app.config['MONGODB_USERNAME'] = 'root'     #用户名
app.config['MONGODB_PASSWORD'] = '123456'    #用户密码

除了上面的单独连接配置，可以添加以下配置：

MONGODB_ALIAS：建立的数据库连接的别名，默认为：default；
MONGODB_IS_MOCK：是否使用 mongomock；
MONGODB_CONNECT：是否直接连接服务器，如果为false，则直到第一次操作时才会连接服务器；
MONGODB_TZ_AWARE：是否自动识别时区，如果为false，则直接使用本地时间，忽略datetime的时区配置。

注意：当我们使用了MONGODB_SETTINGS配置时，其他单独连接配置无效。

映射文档

首先创建一个名为Student学生模型类，该模型类需要继承Document的子类，再根据需要添加字段，示例代码如下：

import mongoengine as me
class Student(me.Document):    #创建Student学生模型类并继承Document类
    S_name=me.StringField()    #学生名字段，字段类型为StringField
    S_password=me.StringField()   #密码字段，字段类型为StringField

这里创建了名为Student学生模型类并继承Document类，在类中我们创建了学生名字段和密码字段，他们的字段类型都为StringField，除了StringField字段，MongoEngine还支持以下字段：

StringField：字符串字段；
BinaryField：二进制字段；
URLField：网址字段；
EmailField：电子邮件字段；
IntField：整型字段；
FloatField：浮点型字段；
DecimalField：十进制字段；
BooleanField：布尔型字段；
DateTimeField：日期时间字段
ListField ：列表字段；
SortedListField ：重复性列表字段；
EmbeddedDocumentField ：使用 wtforms.fields.FormField 并生成内联表单；
ReferenceField：使用 wtforms.fields.SelectFieldBase 和从 QuerySet 或 Document 加载的选项；
DictField：字典字段。

当然我们还可以在字段中添加参数，例如required=True，表示必填项。

当我们需要在Student模型类中嵌套其他字段时，那么可以在类中继承EmbeddedDocument来定义嵌套模型类，并在父模型类中使用EmbeddedDocumentField声明相应的字段。

例如：我们为Student模型类嵌套联系人和手机号字段，首先创建一个名为Contacts的模型类并继承EmbeddedDocument子类，实例代码如下：

class Contacts(me.EmbeddedDocument):
    contacts = me.StringField()
    phone = me.IntField()

创建Contacts模型类之后，还需要在父模型类中使用EmbeddedDocumentField声明相应的字段，示例代码如下：

class Student(me.Document):
 ...
 S_contacts = me.EmbeddedDocumentField(Contacts)

添加数据

添加数据我们通过在上面的配置MongoEngine的Flask程序中添加视图函数进行演示，查看MongoDB数据库我们是通过Navicat Premium 15数据库可视化软件进行查看。

添加单条数据

添加单条数据，创建实例化Student模型类对象中传入数据即可，示例代码如下：

@app.route('/')
def add_data():
    student = Student(S_name='小明',S_password='12345')  #添加数据
    student.S_contacts=Contacts(       #添加嵌套数据
        contacts='小芳',
        phone=13700000000
    )
    student.save()           #保存添加
    return '添加成功'

启动Flask程序，访问http://127.0.0.1:5000/后即可添加数据，如下图所示：

点击上图红框的展开按钮，即可看到添加的嵌套字段数据，如下图所示：

添加多条数据

添加多条数据需要利用模型类中的objects属性，示例代码如下所示：

@app.route('/adds')
def add_datas():
    student=[Student(S_name='张三',S_password='12345',S_contacts=Contacts(contacts='小芳',phone=13700000000)),
             Student(S_name='李四',S_password='12345',S_contacts=Contacts(contacts='小芳',phone=13700000000)),
     Student(S_name='小明',S_password='12345',S_contacts=Contacts(contacts='小米',phone=13700000000))]     #创建数据列表
    Student.objects.insert(student)   #添加数据
    return '添加成功'

首先创建一个名为student的里面，在列表中创建实例化Student类对象并传入数据，最后通过模型类中objects属性的insert方法插入数据，启动Flask程序，访问http://127.0.0.1:5000/adds，这样就成功添加了数据，如下图所示：

查询数据

查询全部数据

使用模型类中objects属性的all()，语法格式为：

模型类.objects().all()       #返回所有数据
模型类.objects(模型类字段='值').all()   #返回满足条件的所有数据

示例代码如下：

@app.route('/find_all')
def find_all():
    student=Student.objects().all()    #查找Student类中的全部数据返回给student
    return render_template('index.html',student=student)  #渲染index.html，并传递student

首先使用Student模型类中objects属性的all()方法获取Student中的所有数据并传递变量给student，再使用render_template()方法渲染index.html并传递变量student。

index.html示例代码如下：

{% for stu in student %}
    学生名：{
   { stu.S_name }}<br>
    密码：{
   { stu.S_password }}<br>
    联系人：{
   { stu.S_contacts.contacts }}<br>
    手机号：{
   { stu.S_contacts.phone }}<br>
    <br>
{% endfor %}

这里使用了for循环来遍历接收传递student变量，并通过.模型类字段的方式获取对应的数据。

启动Flask程序，并访问http://127.0.0.1:5000/find_all，如下图所示：

查询单条数据

查询单条数据使用模型类中objects属性的first()，语法格式为：

模型类.objects().first()      #返回第一条数据
模型类.objects(模型类字段='值').first()   #返回满足条件的第一条数据

示例代码如下：

@app.route('/find_first')
def find_first():
    student=Student.objects(S_name='小明').first()   #返回学生名为小明的第一条数据返回给student
    return render_template('index_first.html',student=student)  #渲染index_first.html，并传递student

使用Student模型类中objects属性的first()方法获取Student中的第一条数据并传递给变量student，再使用render_template()方法渲染index_first.html并传递变量student。

index_first.html示例代码如下：

    学生名：{
   { student.S_name }}<br>
    密码：{
   { student.S_password }}<br>
    联系人：{
   { student.S_contacts.contacts }}<br>
    手机号：{
   { student.S_contacts.phone }}<br>
    <br>

启动Flask程序，访问http://127.0.0.1:5000/find_all，如下图所示：

修改数据

修改数据首先要找到要修改的数据，再使用update()方法来修改，update()语法格式为：

update(模型类字段='修改值')

示例代码如下：

@app.route('/update')
def update():
    student=Student.objects(S_name='小明').first()   #查询学生名为小明的第一条数据
    if not student:            #数据不存在时，返回数据不存在
        return '数据不存在'
    student.update(S_name='王五')       #修改数据
    return '修改成功'

启动Flask程序，访问http://127.0.0.1:5000/update后，查看MongoDB数据库，如下图所示：

这样就成功修改了数据，当我们需要批量修改数据时，只需要把查询数据的first()方法改为all()即可

删除数据

删除数据首先要找到要删除的数据，再使用delete()方法来删除，示例代码如下：

@app.route('/delete')
def delete():
    student=Student.objects(S_name='王五').first()  #查询名为王五的第一条数据
    if not student:           #当数据不存在时返回数据不存在
        return '数据不存在' 
    student.delete()         #删除数据
    return '删除成功'

启动Flask程序，访问http://127.0.0.1:5000/delete，查看MongoDB数据库，如下图所示：