Flask 中 MongoDB 的使用

1. Flask 应用的外部脚本:Flask-Script

Flask-Script 是一个可以在 flask 应用外部编写脚本的扩展。它有一些常用的功能如下:

  • 运行一个开发的服务器
  • Python shell 中操作数据库
  • 命令行任务

首先,我们需要安装 Flask-Script,在Linux系统下,我们可以通过 pip install flask-script 命令行来安装。

我们还是先来看一下,我们目录下面的文件:

192:test ming$ tree
.
├── app.py
├── static
└── templates

2 directories, 1 file

之后,我们来创建一个文件用来存放我们的 web 脚本,一般我们将它命名为 manage.py。然后在里面编写一个 hello world 的脚本,如下:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from flask_script import Manager
from app import app

# 实例化
manager = Manager(app)

# 修饰器
@manager.command
def hello():
    print 'hello world'

if __name__ =="__main__":
    manager.run()

然后再终端中运行这个脚本 python manage.py hello ,结果如下

192:test ming$ python manage.py hello
hello world

我们在运行命令中,也是可以从终端传入一些参数的,我们修改一下代码,实现参数的传入,如下:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from flask_script import Manager
from app import app

# 实例化
manager = Manager(app)

# 修饰器
@manager.command
def hello():
    print 'hello world'

@manager.option('-m', '--msg', dest='msg_val', default='world')
def hello_world(msg_val):
    print 'hello ' + msg_val

if __name__ =="__main__":
    manager.run()

现在,我们来运行一下这个程序。首先,我们不传入任何的参数来运行这个程序,运行结果如下:

192:test ming$ python manage.py hello_world
hello world

之后,我们从终端中传入参数 ming ,使其能打印 hello ming,运行结果如下:

192:test ming$ python manage.py hello_world -m ming
hello ming

其实,细心的你肯定发现了,我们的运行脚本是

python manage.py <方法名>

至此,我们已经学会了如何简单的去操作 Flask 应用的外部脚本。

2. MongoDB 的使用

如果你对 MongoDB 还不了解,那么可以先查看这篇博客,做一个入门学习。

python 与 MongoDB 之间的交互有很多种的方法,在这里我们选择 pymongo 来进行连接。

首先,我们来查看一下,我们创建的文件,如下:

192:test ming$ tree
.
├── app.py
├── manage.py
├── models.py
├── static
└── templates

2 directories, 3 files

我们开始编写文件,先来编写 models.py 中的模型代码,如下:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import pymongo

# 返回一个 collection
def get_coll():

    client = pymongo.MongoClient('127.0.0.1', 27017)
    db = client.test
    user = db.user_collection

    return user


class User(object):

    def __init__(self, name, email):
        self.name = name
        self.email = email

    def save(self):
        user = {"name": self.name, "email": self.email}
        coll = get_coll()
        id = coll.insert(user)
        print id

    @staticmethod
    def query_users():
        users = get_coll().find()
        return users

之后,编写编写脚本文件,如下:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from flask_script import Manager
from app import app
from models import User

# 实例化
manager = Manager(app)

# 修饰器
@manager.command
def save():
    user = User('ming', '[email protected]')
    user.save()

@manager.command
def query_users():
    users = User.query_users()
    for user in users:
        print user


if __name__ =="__main__":
    manager.run()

接下来,我们可以查看一下我们的程序,首先是保存脚本,结果如下:

192:test ming$ python manage.py save
58234f769f5a725d8dda373e

返回了一个 ID ,说明保存成功了。接下来,我们运行一下查询脚本,查询结果如下:

192:test ming$ python manage.py query_users
{u'_id': ObjectId('58234f769f5a725d8dda373e'), u'name': u'ming', u'email': u'[email protected]'}

接下来,我们来学习 MongoDB 中的ORM。MongoDB 的 ORM 有很多种,在本博客中我们 MongoEngine 来实现。

首先,我们需要修改 app.py 程序,将 MongoEngine 导入程序中,如下:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from flask import Flask 
from flask.ext.mongoengine import MongoEngine 

app = Flask(__name__)

# test 是链接的数据库
app.config['MONGODB_SETTINGS'] = {'db': 'test'}

# 实例化
db = MongoEngine(app)

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


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

之后,我们修改 models.py 程序,添加两个字段 nameemail。如下:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from app import db

# 类名定义 collection
class User(db.Document):
    # 字段
    name = db.StringField()
    email = db.StringField()

    def __str__(self):
        return "name:{} - email:{}".format(self.name, self.email)

最后,我们修改脚本文件 manage.py ,如下:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from flask_script import Manager
from app import app
from models import User

# 实例化
manager = Manager(app)

# 修饰器
@manager.command
def save():
    user = User(name = 'ming02', email = '[email protected]')
    user.save()

@manager.command
def query_users():
    users = User.objects.all()
    for user in users:
        print user


if __name__ =="__main__":
    manager.run()

现在,你可以运行 python manage.py savepython manage.py query_users 命令来测试程序了。

转载于:https://my.oschina.net/u/3579120/blog/1533496

你可能感兴趣的:(Flask 中 MongoDB 的使用)