Python的Flask框架与数据库连接的教程

下面我将详细讲解如何用Python的Flask框架连接数据库的完整攻略。这个过程中会包含两个示例，分别是连接MySQL和连接SQLite。

1. 安装Flask和数据库连接驱动

在开始之前，需要确保已经安装了Python和pip，并且通过pip安装了Flask和要连接的数据库的相应驱动。我们以连接MySQL和SQLite为例，分别需要安装pymysql和sqlite3两个库。

pip install Flask pymysql sqlite3

2. 创建Flask应用程序

Flask是一个轻量级的Web框架，我们可以在应用程序的根目录下创建一个名为app.py的Python文件，并在其中创建一个Flask实例。

from flask import Flask

app = Flask(__name__)

3. 配置数据库连接

Flask使用__name__参数指定应用程序根目录，因此我们可以在根目录下创建一个config.py文件，并在其中指定数据库连接相关参数。

3.1 MySQL连接配置

下面是配置MySQL数据库连接的示例，我们需要提供数据库的host、port、user、password、db等参数。

SQLALCHEMY_DATABASE_URI = "mysql+pymysql://user:password@host:port/db"

3.2 SQLite连接配置

下面是配置SQLite数据库连接的示例，我们需要指定数据库文件的路径。

SQLALCHEMY_DATABASE_URI = "sqlite:///data.db"

注意：在配置连接参数时，务必要注意安全性问题，在生产环境中不要将数据库密码等敏感信息直接硬编码在代码中。

4. 初始化数据库

Flask提供了一个名为Flask-SQLAlchemy的扩展，它简化了与SQLAlchemy的集成。我们可以使用db.create_all()方法来创建数据库和定义的表格。

from flask_sqlalchemy import SQLAlchemy

app.config.from_pyfile('config.py')
db = SQLAlchemy(app)

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

db.create_all()

在创建表格之前，我们需要从Flask-SQLAlchemy导入SQLAlchemy类，并创建一个db对象。在模型类中定义表格的名称和结构。

5. 使用数据库

接下来，我们可以在视图函数中通过db.session对象来对数据库进行操作。下面是一个查询数据库中全部用户信息的示例。

from flask import jsonify

@app.route('/users')
def users():
    users = User.query.all()
    return jsonify([{
        'id': user.id,
        'username': user.username,
        'email': user.email
    } for user in users])

6. 示例一：连接MySQL数据库

下面是一个连接MySQL数据库的完整示例。

from flask import Flask, jsonify
from flask_sqlalchemy import SQLAlchemy

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

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

@app.route('/users')
def users():
    users = User.query.all()
    return jsonify([{
        'id': user.id,
        'username': user.username,
        'email': user.email
    } for user in users])

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

在上面的代码中，我们首先导入需要的模块和库，创建Flask实例和SQLAlchemy对象。然后创建一个名为User的模型类，定义表格的名称和结构。

最后，在视图函数中使用User.query查询所有用户信息并返回。

7. 示例二：连接SQLite数据库

下面是一个连接SQLite数据库的完整示例。相对于连接MySQL的示例，这个示例更加简洁，因为SQLite是一个轻量级的嵌入式数据库，不需要像MySQL一样配置连接参数。

from flask import Flask, jsonify
from flask_sqlalchemy import SQLAlchemy

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

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

@app.route('/users')
def users():
    users = User.query.all()
    return jsonify([{
        'id': user.id,
        'username': user.username,
        'email': user.email
    } for user in users])

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

在这个示例中，我们同样导入需要的模块和库，并创建Flask实例和SQLAlchemy对象。然后创建一个名为User的模型类，定义表格的名称和结构。

最后，在视图函数中使用User.query查询所有用户信息并返回。因为我们在config.py中已经指定了数据库文件的路径，所以不需要再指定连接参数。

8. 总结

上面是如何使用Flask框架连接数据库的完整攻略，其中包含了连接MySQL和SQLite数据库的两个示例。在实际开发中，我们可以根据需要进行配置和扩展，例如增加数据的增、删、改等操作。希望这篇攻略对大家有所帮助。