Flask-中-MongoDB-的使用
作者：chen_h
微信号 & QQ：862251340
微信公众号：coderpai
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 程序，添加两个字段 name 和 email。如下：
#!/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 save 和 python manage.py query_users 命令来测试程序了。