Flask 連線 MongoDB

本文的參考專案 https://github.com/shinder/flask-practice

這裡直接用最基本的 PyMongo 連線方式，官方文件。

建立連線模組，app/modules/mongo_connection.py：

from pymongo import MongoClient

url = 'mongodb://localhost:27017'
client = None

def getDB(db_name):
    global client
    if not client:
        client = MongoClient(url)
    return (client[db_name], client)

這個 MongoDB 連線模組的想法和 mysql_connection.py 的想法相同，有使用才會連線。測試的 route 部份：

import modules.mongo_connection

@app.route('/try-mongo')
def try_mongo():
    (db, c) = modules.mongo_connection.getDB('test')
    one = db.inventory.find_one()
    one['_id'] = str(one['_id'])  # 將 ObjectId 轉換為字串顯示
    return one