Flask连接数据库|极客笔记

Flask连接数据库

在Web开发中，数据库连接是非常重要的一环，可以将数据存储在数据库中，方便管理和查询。Flask作为一款轻量级的Web框架，提供了许多插件和扩展来方便我们连接数据库，其中最常用的数据库是SQLite和MySQL。在本文中，我们将详细介绍如何在Flask中连接数据库。

安装Flask和数据库插件

在开始之前，首先需要安装Flask和相应的数据库插件。可以通过pip来安装这些依赖。在命令行中执行以下命令：

pip install Flask
pip install flask_sqlalchemy

上面的命令会安装Flask和Flask-SQLAlchemy插件，Flask-SQLAlchemy是Flask框架中最常用的数据库插件之一，它提供了便利的ORM（对象关系映射）功能，可以让我们用对象的方式操作数据库。

连接SQLite数据库

SQLite是一种轻量级的嵌入式关系型数据库，非常适合用于开发和小型项目。下面我们将演示如何在Flask中连接SQLite数据库。

首先，创建一个Flask应用并配置数据库连接：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

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

上面的代码中，我们创建了一个Flask应用实例，然后配置了数据库连接URI，这里使用的是SQLite数据库，并指定了数据库文件为test.db。接着，创建了一个SQLAlchemy对象，并传入Flask应用实例，用于连接数据库。

接下来，定义一个模型类，用于创建数据库表：

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

上面的代码定义了一个User模型类，表示用户表，包含了id和username两个字段。id字段为主键，username字段为唯一且不可为空。

最后，创建数据库表和插入数据：

db.create_all()

user = User(username='admin')
db.session.add(user)
db.session.commit()

上面的代码中，调用create_all()方法创建数据库表，然后创建一个User对象并插入到数据库中。

连接MySQL数据库

除了SQLite外，MySQL是另一种常用的关系型数据库，适用于中小型和大型项目。接下来我们将演示如何在Flask中连接MySQL数据库。

首先需要安装MySQL数据库，并创建一个数据库和用户供Flask应用连接。假设我们已经安装了MySQL数据库并创建了名为test的数据库，在Flask应用中配置MySQL数据库连接：

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://username:password@localhost/test'
db = SQLAlchemy(app)

上面的代码中，我们修改了数据库URI为MySQL的格式，指定了用户名、密码、主机和数据库名。

接下来定义一个模型类，并创建数据库表和插入数据的过程与连接SQLite数据库类似，这里不再赘述。

总结

通过本文的介绍，我们学习了如何在Flask中连接SQLite和MySQL数据库。Flask提供了丰富的插件和扩展，使得数据库连接变得简单快捷。在实际开发中，根据项目的需要选择适合的数据库类型，合理设计模型类和数据库表结构，这样可以更好地管理和操作数据。

Flask连接数据库