Flask 连接数据库

在本文中，我们将介绍如何在 Flask 中连接数据库，并探讨不同的方法。

阅读更多：Flask 教程

方法一：使用 Flask-SQLAlchemy

Flask-SQLAlchemy 是一个与 Flask 集成的 SQL 工具包，它提供了简单且强大的数据库操作功能。使用这个方法，我们可以通过定义模型类来创建数据库表，并使用 SQLAlchemy 提供的查询语言进行数据查询和操作。

首先，我们需要安装 Flask-SQLAlchemy 库：

pip install flask-sqlalchemy

然后，在 Flask 应用中引入库并配置数据库连接：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'

db = SQLAlchemy(app)

接下来，我们可以定义一个模型类来创建数据库表：

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50))
    email = db.Column(db.String(100))

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

数据库表的创建会在第一次访问应用时自动完成。我们可以使用以下代码来创建一个用户并保存到数据库中：

user = User(name='John', email='john@example.com')
db.session.add(user)
db.session.commit()

使用 Flask-SQLAlchemy，我们可以方便地进行数据库的增删改查等操作。

方法二：使用原生的 SQL

除了使用 Flask-SQLAlchemy，我们还可以使用原生的 SQL 语句来连接数据库。这种方法需要使用 Flask 提供的 flask_mysqldb 扩展。

首先，我们需要安装 flask_mysqldb 扩展及相应的数据库驱动：

pip install flask-mysqldb
pip install pymysql

然后，在 Flask 应用中引入扩展并配置数据库连接：

from flask import Flask
from flask_mysqldb import MySQL

app = Flask(__name__)
app.config['MYSQL_HOST'] = 'localhost'
app.config['MYSQL_USER'] = 'root'
app.config['MYSQL_PASSWORD'] = 'password'
app.config['MYSQL_DB'] = 'database'

mysql = MySQL(app)

接下来，我们可以使用原生的 SQL 语句来进行数据库的操作：

@app.route('/')
def index():
    cur = mysql.connection.cursor()
    cur.execute("SELECT * FROM users")
    users = cur.fetchall()
    cur.close()
    return str(users)

使用原生的 SQL，我们可以灵活地编写各种复杂的数据库查询和操作语句。

方法选择

使用 Flask-SQLAlchemy 可以更好地与 Flask 集成，并提供了更方便的数据库操作方式。通过定义模型类来创建数据库表，可以使代码更加清晰和易于维护。同时，SQLAlchemy 提供了丰富的查询语言，可以方便地进行各种查询操作。

然而，如果你更熟悉原生的 SQL 语句或需要进行复杂的数据库操作，使用原生的 SQL 语句可能更适合你。原生的 SQL 语句可以提供更高的灵活性和自由度，可以满足各种复杂的业务需求。

综上所述，选择使用哪种方法连接数据库取决于你的实际需求和个人偏好。无论选择哪种方法，Flask 都提供了灵活且强大的扩展机制，可以与各种数据库进行无缝集成。

总结

本文介绍了在 Flask 中连接数据库的两种方法：使用 Flask-SQLAlchemy 和使用原生的 SQL。使用 Flask-SQLAlchemy 可以提供更方便的操作方式，通过定义模型类来创建数据库表，并使用 SQLAlchemy 提供的查询语言进行数据查询和操作。而使用原生的 SQL 语句可以提供更高的灵活性和自由度，适用于复杂的数据库操作。

根据实际需求和个人偏好，可以选择适合自己的方法来连接数据库，并在 Flask 应用中进行数据的增删改查等操作。无论选择哪种方法，Flask 都提供了强大的扩展机制，可以与各种数据库进行无缝集成，为我们的应用提供稳定可靠的数据支持。