安裝
pip install Flask-PyMongo
初始化Pymongo實例
from flask import Flask from flask.ext.pymongo import PyMongo app = Flask(__name__) app.config.update( MONGO_HOST='localhost', MONGO_PORT=27017, MONGO_USERNAME='bjhee', MONGO_PASSWORD='111111', MONGO_DBNAME='flask' ) mongo = PyMongo(app)
在應用配置中，我們指定了MongoDB的服務器地址，端口，數據庫名，用戶名和密碼。對於上面的配置，我們也可以簡化為：
app.config.update( MONGO_URI='mongodb://localhost:27017/flask', MONGO_USERNAME='bjhee', MONGO_PASSWORD='111111' )
在同一應用中，我們還可以初始化兩個以上的Flask-PyMongo實例，分別基於不同的配置項：
app.config.update( MONGO_URI='mongodb://localhost:27017/flask', MONGO_USERNAME='bjhee', MONGO_PASSWORD='111111', MONGO_TEST_URI='mongodb://localhost:27017/test' ) mongo = PyMongo(app) mongo_test = PyMongo(app, config_prefix='MONGO_TEST')
當調用初始化方法”PyMongo()”時，傳入”config_prefix”參數，該PyMongo實例就會使用以”MONGO_TEST”為前綴的配置項，而不是默認的”MONGO”前綴，比如上例中的”MONGO_TEST_URI”。
添加數據
user = {'name':'Michael', 'age':18, 'scores':[{'course': 'Math', 'score': 76}]}
mongo.db.users.insert_one(user)
“mongo.db.users”用來獲取名為”users”集合對象，類型是”pymongo.collection.Collection”，該對象上的”insert_one()”方法用來創建一條記錄。相應的，集合對象上的”insert_many()”方法可以同時創建多條記錄，比如：
result = mongo.db.tests.insert_many([{'num': i} for i in range(3)])
查詢數據
集合對象提供了”find_one()”和”find()”方法分別用來獲取一條和多條文檔記錄，兩個方法都可以傳入查詢條件作為參數：
@app.route('/user') @app.route('/user/<string:name>') def user(name=None): if name is None: users = mongo.db.users.find() return render_template('users.html', users=users) else: user = mongo.db.users.find_one({'name': name}) if user is not None: return render_template('users.html', users=[user]) else: return 'No user found!'
上例中的模板文件”users.html”如下：
<!doctype html> <title>PyMongo Sample</title> <h1>Users:</h1> <ul> {% for user in users %} <li>{{ user.name }}, {{ user.age }}</li> <ul> {% for score in user.scores %} <li>{{ score.course }}, {{ score.score }}</li> {% endfor %} </ul> {% endfor %} </ul>
“find_one()”方法返回的就是一個字典，所以我們可以直接對其作操作。”find()”方法返回的其實是一個”pymongo.cursor.Cursor”對象，不過Cursor類實現了”__iter__()”和”next()”方法，因此可以用”for … in …”循環來遍歷它。
Cursor類還提供了很多功能接口來強化查詢功能
- “count()”方法, 獲取返回數據集的大小
users = mongo.db.users.find({'age':{'$lt':20}})
print users.count() # 打印年齡小於20的用戶個數
- “sort()”方法, 排序
from flask_pymongo import DESCENDING # 返回所有用戶，並按名字升序排序 users = mongo.db.users.find().sort('name') # 返回所有用戶，並按年齡降序排序 users = mongo.db.users.find().sort('age', DESCENDING)
- “limit()”和”skip()”方法, 分頁
# 最多只返回5條記錄，並且忽略開始的2條 # 即返回第三到第七（如果存在的話）條記錄 users = mongo.db.users.find().limit(5).skip(2)
- “distinct()”方法, 獲取某一字段的唯一值
ages = mongo.db.users.find().distinct('age') print (ages) # 打印 [18, 21, 17]
注意，”distinct()”方法需傳入字段名，它返回的是一個列表，而不是Cursor或文檔。上例列出了’age’字段所有的唯一值。
更新數據
“pymongo.collection.Collection”提供了兩種更新數據的方法，一種是update，可以更新指定文檔中某個字段的值，同關系型數據庫中的update類似。update有兩個函數，”update_one()”更新一條記錄，”update_many()”更新多條記錄：
# 找到名為Tom的第一條記錄，將其年齡加3 result = mongo.db.users.update_one({'name': 'Tom'}, {'$inc': {'age': 3}}) # 打印被改動過的記錄數 print '%d records modified' % result.modified_count # 找到所有年齡小於20的用戶記錄，將其年齡設為20 result = mongo.db.users.update_many({'age':{'$lt':20}}, {'$set': {'age': 20}}) # 打印被改動過的記錄數 print '%d records modified' % result.modified_count
另一種更新數據的方法是replace，它不是用來更新某一字段，而是把整條記錄替換掉。它就一個函數”replace_one()”：
user = {'name':'Lisa', 'age':23, 'scores':[{'course': 'Politics', 'score': 95}]}
# 找到名為Jane的第一條記錄，將其替換為上面的名為Lisa的記錄
result = mongo.db.users.replace_one({'name': 'Jane'}, user)
# 打印被改動過的記錄數
print '%d records modified' % result.modified_count
刪除數據
刪除數據可以使用集合對象上的delete方法，它也有兩個函數，”delete_one()”刪除一條記錄，”delete_many()”刪除多條記錄：
# 刪除名為Michael的第一條記錄 result = mongo.db.users.delete_one({'name': 'Michael'}) # 打印被刪除的記錄數 print '%d records deleted' % result.deleted_count # 找到所有年齡大於20的用戶記錄 result = mongo.db.users.delete_many({'age':{'$gt':20}}) # 打印被刪除的記錄數 print '%d records deleted' % result.deleted_count
如果你想將集合整個刪除，可以使用”drop()”方法：
mongo.db.users.drop()
練習：PyMongo結合Restful
from flask import Flask, request from flask_restful import Api, Resource from flask_pymongo import PyMongo app = Flask(__name__) app.config['MONGO_URI']='mongodb://localhost:27017/flask' api = Api(app) mongo = PyMongo(app) class User(Resource): def get(self, name): user = mongo.db.users.find_one({'name': name}) if user is not None: user.pop('_id') return dict(result='success', user=user) return dict(result='error', message='No record found') def delete(self, name): result = mongo.db.users.delete_one({'name': name}) count = result.deleted_count if count > 0: return dict(result='success', message='%d records deleted' % count) return dict(result='error', message='Failed to delete') def put(self, name): user = request.get_json() result = mongo.db.users.replace_one({'name': 'name'}, user) count = result.modified_count if count > 0: return dict(result='success', message='%d records modified' % count) return dict(result='error', message='Failed to modify') class UserList(Resource): def get(self): users = mongo.db.users.find() user_list = [] for user in users: user.pop('_id') user_list.append(user) return dict(result='success', userlist=user_list) def post(self): user = request.get_json() user_id = mongo.db.users.insert_one(user).inserted_id if user_id is not None: return dict(result='success', message='1 record added') return dict(result='error', message='Failed to insert') api.add_resource(UserList, '/users') api.add_resource(User, '/users/<name>') if __name__ == '__main__': app.run(host='0.0.0.0', debug=True)
這里我們使用name作為鍵值來查詢，因為MongoDB中的id太復雜。注意，我們在輸出時都會把”_id”字段去掉，因為它是”ObjectId”類型無法JSON序列化，同時如果你的數據中有日期時間類型，也要特別處理后才能被JSON序列化。