博客

如何在Python中使用Flask SQLAlchemy操作数据库？

  • Post category:Python

如何在Python中使用Flask SQLAlchemy操作数据库？

Flask SQLAlchemy是一个基于Flask的Python ORM（对象关系映射）库，它提供了一种简单的方式来操作关系型数据库。使用Flask SQLAlchemy，我们可以使用Python代码来创建、读取、更新删除关系型数据库中的数据。以下是如何在Python中使用Flask SQLAlchemy操作数据库的完整使用攻略，包括连接数据库、创建表、插入数据、查询数据等步骤。同时，提供了两个示例以便更好理解如何在Python中使用Flask SQLAlchemy操作数据库。

步骤1：安装Flask SQLAlchemy

在Python中，我们需要安装Flask SQLAlchemy才能使用它来操作关系型数据库。以下是安装Flask SQLAlchemy的基本语法：

!pip install flask_sqlalchemy

在上面的语法中，我们使用pip命令安装Flask SQLAlchemy。

步骤2：连接数据库

在Python中，我们可以使用Flask SQLAlchemy连接关系型数据库。以下是连接的基本语法：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

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

在上面的语法中，我们首先导入Flask和SQLAlchemy类。然后，我们创建一个Flask应用程序对象，并设置数据库连接URI。最后，我们使用SQLAlchemy类创建一个数据库对象。

步骤3：创建表

在Flask SQLAlchemy中我们使用模型来表示关系型数据库中的表。以下是创建表的基本语法：

class Employee(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50), nullable=False)
    age = db.Column(db.Integer, nullable=False)

在上面的语法中，我们使用db.Model类创建一个模型对象。然后，我们使用db.Column类创建模型的字段。在字段中，我们可以设置字段的类型、长度、是否必须等属性。最后，我们使用primary_key=True设置主键。

步骤4：插入数据

在Python中，我们可以使用Flask SQLAlchemy插入数据到关系型数据库中。以下是插入数据的基本语法：

employee = Employee(name='John', age=30)
db.session.add(employee)
db.session.commit()

在上面的语法中，我们一个Employee对象，并设置name和age字段的值。然后，我们使用db.session.add方法将数据插入到关系型数据库中。最后，我们使用db.session.commit方法提交更改。

步骤5：查询数据

在Python中，我们可以使用Flask SQLAlchemy查询关系型数据库中的数据。以下是查询数据的基本语法：

employees = Employee.query.all()

for employee in employees:
    print(employee.name, employee.age)

在上面的语法中，我们使用query.all()方法查询Employee表中的所有数据。然后，我们使用for循环遍历所有数据，并打印出每个员工的姓名和年龄。

示例1

在这个示例中，我们使用Flask SQLAlchemy连接SQLite数据库，并创建一个Employee表。然后，我们插入一条数据到Employee表中。接着，我们查询所有数据。

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

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

class Employee(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50), nullable=False)
    age = db.Column(db.Integer, nullable=False)

employee = Employee(name='John', age=30)
db.session.add(employee)
db.session.commit()

employees = Employee.query.all()

for employee in employees:
    print(employee.name, employee.age)

在上面的代码中，我们首先创建一个Flask应用程序对象，并设置数据库连接URI。接着，我们使用db.Model类创建一个名为Employee的模型对象。在模型中，我们使用db.Column类创建id、name和age字段。最后，我们使用primary_key=True设置id字段为主键。

接着，我们使用db.session.add方法将数据插入到SQLite数据库中。

然后，我们使用query.all()方法查询Employee表中的所有数据。然后，我们使用for循环遍历所有数据，并打印出每个员工的姓名和年龄。

示例2

在这个示例中，我们使用Flask SQLAlchemy连接SQLite数据库，并创建一个Order表。然后，我们插入两条数据到Order表中。接着，我们查询所有数据，并更新一条数据和删除一条数据。

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

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

class Order(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    customer_name = db.Column(db.String(50), nullable=False)
    order_date = db.Column(db.Date, nullable=False)

order1 = Order(customer_name='John', order_date='2023-05-12')
db.session.add(order1)

order2 = Order(customer_name='Mary', order_date='2023-05-13')
db.session.add(order2)

db.session.commit()

orders = Order.query.all()

for order in orders:
    print(order.customer_name, order.order_date)

order = Order.query.filter_by(customer_name='John').first()
order.order_date = '2023-05-14'
db.session.commit()

order = Order.query.filter_by(customer_name='Mary').first()
db.session.delete(order)
db.session.commit()

在上面的代码中，我们首先创建一个Flask应用程序对象，并设置数据库连接URI。接着，我们使用db.Model类创建一个名为Order的模型对象。在模型中，我们使用db.Column类创建id、customer_name和order_date字段。最后，我们使用primary_key=True设置id字段为主键。

接着，我们使用db.session.add方法将数据插入到SQLite数据库中。

然后，我们使用query.all()方法查询Order表中的所有数据。然后，我们使用for循环遍历所有数据，并打印出每个订单客户姓名和订单日期。

然后，我们使用query.filter_by方法查询Order表中customer_name为John的数据。然后，更新该数据的order_date字段，并使用db.session.commit方法提交更改。

最后，我们使用query.filter_by方法查询Order表中customer_name为Mary的数据。然后，我们使用db.session.delete方法删除该数据，并使用db.session.commit方法提交更改。

Tags: ,