Flask學習之旅--數據庫

一、寫在前面

　　在Web開發中，數據庫操作是很重要的一部分，因為網站的很多重要信息都保存在數據庫之中。而Flask在默認情況下是沒有數據庫、表單驗證等功能的，但是可以用Flask-extension為Web應用添加這些功能。

 

二、Flask SQLite

　　SQLite是一款輕型的數據庫，是遵守ACID的關系型數據庫管理系統。由於Python對SQlite有內置的支持，因此在Flask應用程序中和SQLite進行交互是比較容易的。

　　首先需要創建一個SQLite數據庫“user.db”，並在其中創建一張用戶表。代碼如下：

import sqlite3 2 

conn = sqlite3.connect("user.db") 5 print("Connected!") 6 
conn.execute("CREATE TABLE USER(username TEXT, password TEXT, EMAIL TEXT)") 8 print("Table created successfully!") 9 conn.close()

　　這里就不貼HTML代碼了，就是一個注冊頁面，在注冊的時候會將用戶輸入的用戶名、密碼和郵箱傳到后台。在app.py中需要導入sqlite3模塊，然后連接前面創建的user.db，在建立連接之后創建一個游標對象，然后編寫SQL語句進行數據庫操作，整個過程都算是比較容易的。app.py中的代碼如下：

from flask import Flask, render_template, request  2 import sqlite3 as sql  3 
app = Flask(__name__)  5 

@app.route('/register', methods=['GET', 'POST'])  8 def register():  9     if request.method == 'GET': 10         return render_template('register.html') 11     else: 12         msg = ""
        try: 14             username = request.form["usr"] 15             password = request.form["pwd"] 16             email = request.form["email"] 17             print(username, password, email) 18             with sql.connect("user.db") as con: 19                 cur = con.cursor() 20                 cur.execute("INSERT INTO USER (username, password, email) VALUES (?,?,?)", 21  (username, password, email)) 22  con.commit() 23             msg = "注冊成功！"
        except: 25  con.rollback() 26             msg = "注冊失敗！請重試！"
        finally: 28  con.close() 29             return render_template('register.html', msg=msg) 30 

if __name__ == '__main__': 33     app.run()

 

三、Flask SQLAlchemy

　　Flask-SQLAlchemy是Flask擴展，它將對SQLAlchemy的支持添加到Flask應用程序中。在Flask Web應用程序中使用原始SQL對數據庫執行CRUD操作可能很繁瑣，不過SQLAlchemy 為應用程序開發人員提供了SQL的全部功能和靈活性。它是一個對數據庫的抽象，讓開發者不用這些編寫SQL語句，而是使用其提供的接口去操作數據庫，這其中涉及到一個非常重要的思想：ORM（Object Relation Mapping，對象關系映射），主要的功能是實現模型對象到關系型數據庫數據的映射。說白了就是使用通過對象去操作數據庫。

1.Flask-SQLAlchemy安裝

　　使用pip install flask-sqlalchemy進行安裝。不過在安裝的時候可能會出現如下錯誤：

pip._vendor.urllib3.exceptions.ReadTimeoutError: HTTPSConnectionPool(host='files.pythonhosted.org', port=443): Read timed out.

　　這是因為在下載python庫的時候，由於國內網絡原因，導致python包的下載速度非常慢，查看pip 文檔，只要在 pip的時候控制超時即可, 具體參數為 --default-timeout=100， 后面的時間可以自己指定。因此可以用如下命令進行下載安裝：

pip install --default-timeout=100 flask-sqlalchemy

2.Flask-SQLAlchemy配置

　　從此模塊導入SQLAlchemy類，創建一個Flask應用程序對象並為要使用的數據庫設置URI的代碼如下：

from flask_sqlalchemy import SQLAlchemy


app = Flask(__name__)

# URI的格式為：用戶名：密碼@ip地址：端口號（默認可以不寫）/數據庫名

app.config['SQLALCHEMY_DATABASE_URI'] = "mysql://root:qwer1234@localhost/flask"

3.配置的時候可能出現的問題及解決辦法

1）ModuleNotFoundError: No module named 'MySQLdb'

解決辦法：

　　使用pymysql模塊，URI改為:

app.config['SQLALCHEMY_DATABASE_URI'] = "mysql+pymysql://root:qwer1234@localhost/flask"

2）Warning: (1366, "Incorrect string value: '\\xD6\\xD0\\xB9\\xFA\\xB1\\xEA...' for column 'VARIABLE_VALUE' at row 481")

解決辦法：

　　改用mysql-connector，使用pip install mysql-connector下載安裝，URI改為:

app.config['SQLALCHEMY_DATABASE_URI'] = "mysql+mysqlconnector://root:qwer1234@localhost/flask"

3）sqlalchemy.exc.NotSupportedError: (mysql.connector.errors.NotSupportedError) Authentication plugin 'caching_sha2_password' is not supported

解決辦法：

　　出現這個錯誤是因為MySQL8.0采用了Use Strong Password Encryption for Authentication即強密碼加密，而mysql.connector的引擎不支持caching_sha2_password的加密格式，所以解決思路有如下幾種：

　　（1）重裝MySQL，在Authentication Method中選擇第二項（重裝比較麻煩，我就沒有嘗試）：

　　（2）在client端，將加密格式選擇成mysql_native_password，命令如下（個人嘗試后無效）：

ALTER user 'root'@'localhost' IDENTIFIED WITH mysql_native_password BY 'root'

　　（3）最后一種是我自己摸索出來的，就是在URI設置的時候加上一個參數auth_plugin並設置為mysql_native_password：

app.config['SQLALCHEMY_DATABASE_URI'] = "mysql+mysqlconnector://root:qwer1234@localhost/flask?auth_plugin=mysql_native_password"

4.Flask-SQLAlchemy的基本操作

1） 常用查詢過濾器：

　　過濾器得到的還只是一些對象，需要使用執行器來獲取真正的數據。

filter()： 把過濾器添加到原查詢上，返回一個新查詢，需要使用模型類名去獲取字段來進行比較；

filter_by()：把等值(只能使用=比較操作)過濾器添加到查詢上，返回一個新查詢；

order_by()：根據指定條件對查詢結果進行排序，返回一個新查詢；

group_by()：根據指定條件對原查詢結果進行分組，返回一個新查詢。

2.）常用查詢執行器

all()：以列表的形式返回查詢的所有結果；

first()：返回查詢的第一個結果；

first_or_404()：同first(), 只不過如果沒有找到的話，返回404錯誤；

get()：返回指定主鍵對應的行；

get_or_404()：返回指定主鍵對應的行，如不存在，返回404錯誤；

count()：返回查詢結果的數量；

paginate()：返回一個Paginate對象，包含指定范圍內的結果。

3.）查詢條件

startswith('xx')：查詢以xx開頭的所有數據；

endswith('xx')：查詢以xx結尾的所有數據；

not_()：取反；

and_()：返回and()條件滿足的所有數據；

or_()：返回or()條件滿足的所有數據。

5.使用示例

　　app.py中代碼如下：

from flask_sqlalchemy import SQLAlchemy
import mysql.connector


app.config['SQLALCHEMY_DATABASE_URI'] = "mysql+mysqlconnector://root:qwer1234@localhost/flask？auth_plugin=mysql_native_password"
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True


# 獲取SQLAlchemy實例對象
db = SQLAlchemy(app)

　　然后新建一個model.py，編寫如下代碼進行測試：

from app import db


# 創建模型對象
class User(db.Model):
    __tablename__ = "users"
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(16), unique=True)
    password = db.Column(db.String(16))
    email = db.Column(db.String(32), unique=True)

    def __repr__(self):
        return '<User %r>' % self.username


# 1.創建表
db.create_all()
print("Created Successfully!")

# 2.增加記錄
usr1 = User()
usr1.id = 1
usr1.username = "wang"
usr1.password = "wangwang"
usr1.email = "[e-mail]"
usr2 = User(id=2, username="yang", password="yang", email="[e-mail]")
db.session.add(usr1)
print("Add usr1")
db.session.add(usr2)
print("Add usr2")
db.session.commit()

# 3.查詢記錄,注意查詢返回對象，如果查詢不到返回None
users1 = User.query.all()  # 查詢所有
print(users1)
print("User Count：", len(users1))

# 4.刪除
user = User.query.get(1)
db.session.delete(user)
print("Delete usr1")
db.session.commit()

users2 = User.query.all()  # 查詢所有
print(users2)
print("User Count：", len(users2))

　　運行結果如下所示：

Created Successfully!
Add usr1
Add usr2
[<User 'wang'>, <User 'yang'>]
User Count： 2
Delete usr1
[<User 'yang'>]
User Count： 1