Flask Flask-Sqlalchemy 设置引擎配置

在本文中，我们将介绍如何使用Flask和Flask-Sqlalchemy来设置数据库引擎的配置。

阅读更多：Flask 教程

1. 什么是Flask-Sqlalchemy？

Flask-Sqlalchemy是一个用于在Flask应用程序中进行数据库操作的扩展。它提供了一个简单的方式来定义数据模型，并且支持多种数据库引擎，如MySQL、PostgreSQL和SQLite等。

2. 安装Flask-Sqlalchemy

在开始之前，我们需要先安装Flask-Sqlalchemy扩展。通过以下命令可以使用pip进行安装：

pip install Flask-Sqlalchemy

3. 配置数据库引擎

在Flask-Sqlalchemy中，我们可以通过配置文件或者直接在代码中设置数据库引擎的配置。下面我们将分别介绍这两种方式。

3.1 使用配置文件设置引擎配置

在Flask应用程序的根目录下创建一个名为config.py的文件，用于存放数据库引擎的配置。在config.py中，我们可以定义各种配置选项，如数据库的URL、连接池大小等。

下面是一个示例的config.py文件：

# config.py

SQLALCHEMY_DATABASE_URI = 'mysql://username:password@localhost/dbname'
SQLALCHEMY_POOL_SIZE = 10

在Flask应用程序的入口文件中，通过以下代码读取config.py中的配置：

# main.py

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config.from_pyfile('config.py')

db = SQLAlchemy(app)

3.2 直接在代码中设置引擎配置

如果你不想使用配置文件，也可以直接在代码中设置数据库引擎的配置。下面是一个示例：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://username:password@localhost/dbname'
app.config['SQLALCHEMY_POOL_SIZE'] = 10

db = SQLAlchemy(app)

4. 数据库模型定义

在设置好数据库引擎配置之后，我们就可以定义数据模型了。Flask-Sqlalchemy使用类来表示数据模型，并且支持ORM（对象关系映射）操作。下面是一个示例的数据模型定义：

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(50), unique=True, nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)

5. 创建数据库表格

在定义好数据模型之后，我们需要使用Flask-Migrate对数据库进行迁移。首先，我们需要安装Flask-Migrate扩展：

pip install Flask-Migrate

然后，在Flask应用程序的入口文件中添加以下代码：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate

app = Flask(__name__)
app.config.from_pyfile('config.py')

db = SQLAlchemy(app)
migrate = Migrate(app, db)

运行以下命令来创建数据库迁移文件和迁移数据库表格：

flask db init
flask db migrate
flask db upgrade

6. 使用数据库

在Flask-Sqlalchemy中，我们可以使用ORM操作来对数据库进行增删改查操作。下面是一些常用的示例：

# 添加数据
user = User(username='John', email='john@example.com')
db.session.add(user)
db.session.commit()

# 查询数据
user = User.query.get(1)
users = User.query.filter_by(username='John').all()

# 更新数据
user = User.query.get(1)
user.email = 'new_email@example.com'
db.session.commit()

# 删除数据
user = User.query.get(1)
db.session.delete(user)
db.session.commit()