前言：

數(shù)據(jù)庫(kù)是大多數(shù) Web 應(yīng)用的基礎(chǔ)設(shè)施，只要想把數(shù)據(jù)存儲(chǔ)下來(lái)，就離不開(kāi)數(shù)據(jù)庫(kù)，下面將一起學(xué)習(xí)一下如何給 Flask 應(yīng)用添加數(shù)據(jù)庫(kù)支持。

1、使用Flask-SQLAlchemy管理數(shù)據(jù)庫(kù)

Flask-SQLAlchemy 集成了 SQLAlchemy，它簡(jiǎn)化了連接數(shù)據(jù)庫(kù)服務(wù)器、管理數(shù)據(jù)庫(kù)及操作會(huì)話等各類工作，讓 Flask 更方便的進(jìn)行數(shù)據(jù)存儲(chǔ)及處理，我們不必過(guò)多關(guān)心原生 SQL 語(yǔ)句，只需要使用 Python 類就可以輕松的完成對(duì)數(shù)據(jù)庫(kù)表的增刪改查操作，并且該插件還支持多種數(shù)據(jù)庫(kù)類型，如MySQL、PostgreSQL、和SQLite等。

我們可以使用pip install flask-sqlalchemy進(jìn)行安裝。

1.1 配置Flask_SQLAlchemy

下面以 SQLite 數(shù)據(jù)庫(kù)為例，Flask-SQLAlchemy 數(shù)據(jù)庫(kù)的 url 通過(guò)配置變量SQLALCHEMY_DATABASE_URI指定，通過(guò) Flask-SQLAlchemy 提供的 SQLAlchemy 類傳入 Flask 的實(shí)例 app，創(chuàng)建 db 實(shí)例，表示程序使用的數(shù)據(jù)庫(kù)，這個(gè) db 對(duì)象能夠使用 Flask-SQLAlchemy 的所有功能。

from flask import Flask

from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

basedir = os.path.abspath(app.root_path)

app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join(basedir, 'data.db')

app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

db = SQLAlchemy(app)

其中，SQLALCHEMY_TRACK_ MODIFICATIONS配置變量表示是否追蹤對(duì)象的修改，在此設(shè)為False。

1.2 定義數(shù)據(jù)庫(kù)模型

所謂數(shù)據(jù)模型，就是用來(lái)映射數(shù)據(jù)庫(kù)表的 Python 類，一個(gè)數(shù)據(jù)模型類對(duì)應(yīng)數(shù)據(jù)庫(kù)中的一個(gè)表，類的屬性代表數(shù)據(jù)庫(kù)表的字段。所有的模型類都需要繼承 Flask-SQLAlchemy 提供db.Model基類。

新建model.py文件，定義User類如下：

from app import db

class User(db.Model):

    id = db.Column(db.Integer, primary_key=True)

    user_name = db.Column(db.String)

    password = db.Column(db.String)

1.3 創(chuàng)建數(shù)據(jù)庫(kù)和表

運(yùn)行 Flask 應(yīng)用，然后在終端中輸入flask shell命令，進(jìn)入 Python 交互環(huán)境

如下所示：

>>> from app import db  # 從app.py中導(dǎo)入db實(shí)例

>>> from model import *  # 從model.py中導(dǎo)入所有模型類

>>> db.create_all()  # 用db.create_all()創(chuàng)建數(shù)據(jù)庫(kù)表 

執(zhí)行完之后，會(huì)在項(xiàng)目根目錄生成一個(gè)data.db庫(kù)文件。 注意： 數(shù)據(jù)庫(kù)和表一旦創(chuàng)建后，之后對(duì)模型類的改動(dòng)不會(huì)自動(dòng)作用到實(shí)際的表中，比如，在模型類中添加或刪除字段，修改字段的名稱和類型，再次調(diào)用db.create_all()不會(huì)重新創(chuàng)建表或是更新表，只有通過(guò)db.drop_all()刪除數(shù)據(jù)庫(kù)中所有的表之后再調(diào)用db.create_all()才能重新創(chuàng)建表，

那么就會(huì)出現(xiàn)這么一個(gè)問(wèn)題：

這樣操作的話，數(shù)據(jù)庫(kù)表被刪除重建了，那表中原有的數(shù)據(jù)也都沒(méi)有了，這肯定是不行的，這時(shí)就出現(xiàn)了數(shù)據(jù)庫(kù)遷移的概念，先留個(gè)坑，下篇文章介紹。

1.4 數(shù)據(jù)庫(kù)操作

現(xiàn)在我們創(chuàng)建了模型，也生成了數(shù)據(jù)庫(kù)和表，接下來(lái)就來(lái)學(xué)習(xí)一下常用的數(shù)據(jù)庫(kù)操作。數(shù)據(jù)庫(kù)操作主要是CRUD（Create 創(chuàng)建、Read 讀取/查詢、 Update 更新、Delete 刪除）。

1.4.1 Create

添加一條記錄到數(shù)據(jù)庫(kù)中，主要分為以下三步：

• 使用 Python 模型類創(chuàng)建對(duì)象作為一條記錄
• 添加新創(chuàng)建的對(duì)象到數(shù)據(jù)庫(kù)會(huì)話中
• 提交數(shù)據(jù)庫(kù)會(huì)話

如下，在上面的交互環(huán)境下，創(chuàng)建一個(gè)新用戶：

>>> from app import db  # 從app.py中導(dǎo)入db實(shí)例

>>> from model import User   # 導(dǎo)入模型類User

>>> user1=User(user_name='tigeriaf', password='123456')  # 創(chuàng)建用戶1

>>> user2=User(user_name='admin', password='123456')  # 創(chuàng)建用戶2

# 添加新創(chuàng)建的對(duì)象到數(shù)據(jù)庫(kù)會(huì)話中

>>> db.session.add(user1)

>>> db.session.add(user2)

# 將數(shù)據(jù)庫(kù)會(huì)話提交，數(shù)據(jù)寫(xiě)入data.db文件

>>> db.session.commit()

# 測(cè)試

>>> print(user1.id)

1

>>> print(user2.id)

2

另外，除了依次調(diào)用add()方法添加記錄，也可以使用add_all()一次添加包含多個(gè)模型類對(duì)象的列表。

1.4.2 Read

使用模型類

提供的 query 屬性 通過(guò)模型類提供的 query 屬性附加調(diào)用各種過(guò)濾方法可以查詢數(shù)據(jù)庫(kù)表的數(shù)據(jù)，

查詢模式如下：

<模型類>.query.<過(guò)濾方法>.<查詢方法>

從某個(gè)模型類出發(fā)，通過(guò)在 query 屬性對(duì)應(yīng)的 Query 對(duì)象上附加的過(guò)濾方法和查詢函數(shù)對(duì)模型類對(duì)應(yīng)的表中的記 進(jìn)行各種篩選等，最終返回包含對(duì)應(yīng)數(shù)據(jù)庫(kù)記錄數(shù)據(jù)的模型類實(shí)例，對(duì)返回的實(shí)例調(diào)用屬性即可獲取對(duì)應(yīng)的字段數(shù)據(jù)。

# 查詢?nèi)?

>>> User.query.all()

[<User 1>, <User 2>]

# 查詢指定id的記錄

>>> user1=User.query.get(1)

>>> user1.user_name

'tigeriaf'

# 查詢條數(shù)

>>> User.query.count()

2

# 查詢user_name為admin的用戶

>>> User.query.filter_by(user_name='admin').all()

[<User 2>]

SQLAlcherny 提供了很多過(guò)濾方法，使用這些過(guò)濾方法可以獲取更精確的查詢，這里就不展開(kāi)了。 完整的查詢、過(guò)濾方法可以查看：The Query Object。

1.4.3 Update

更新一條記錄非常簡(jiǎn)單，直接給模型類的屬性附上新的值，然后調(diào)用commit()方法提交會(huì)話即可。 如下，修改 id 為 2 的用戶的用戶名 user_name。

>>> user2=User.query.get(2)

>>> user2.user_name

'admin'

>>> user2.user_name='張三'

>>> db.session.commit()

>>> user2.user_name

'張三'

1.4.4 Delete

使用delete()進(jìn)行數(shù)據(jù)記錄的刪除，如下：

>>> db.session.delete(user2)

>>> db.session.commit()

>>> user2=User.query.get(2)

>>> print(user2)

None

1.4.5 在視圖函數(shù)里操作數(shù)據(jù)庫(kù)

在視圖函數(shù)里操作數(shù)據(jù)庫(kù)的方式在 Python Shell 交互環(huán)境下大致是相同的，無(wú)非就是多了從請(qǐng)求對(duì)象獲取數(shù)據(jù)及驗(yàn)證數(shù)據(jù)的步驟，

如下一個(gè)案例：

@app.route('/user', methods=['GET', 'POST'])

def user():

    if request.method == 'POST':

        user_name = request.form['user_name']

        password = request.form['password']

        user = User(user_name=user_name, password=password)

        db.session.add(user)

        db.session.commit()

        return 'user:{} add success! id:{}'.format(user_name, user.id)

    else:

        user_id = request.args.get('user_id')

        user = User.query.get(user_id)

        if user:

            return 'Hello user:{}!'.format(user.user_name)

        else:

            return 'failed'

上述代碼中，視圖函數(shù) user 接受兩種方式的請(qǐng)求，分別完成添加用戶、查詢用戶的功能，將接收數(shù)據(jù)存儲(chǔ)在數(shù)據(jù)庫(kù)中。

發(fā)送請(qǐng)求測(cè)試如下：

到此這篇關(guān)于 Flask 數(shù)據(jù)庫(kù)集成的介紹的文章就介紹到這了,更多相關(guān) Flask 數(shù)據(jù)庫(kù)集成內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論