Flask如何ORM修改数据|极客笔记

Flask如何ORM修改数据

什么是ORM

ORM（Object-Relational Mapping）是一种编程技术，用于将关系数据库的数据映射到对象模型。ORM库可以帮助开发者通过面向对象的方式操作数据库，而不需要编写SQL语句。

在Python中，常用的ORM库有SQLAlchemy、Django ORM等。在本文中，我们将以Flask框架结合SQLAlchemy为例，演示如何使用ORM修改数据。

准备工作

在开始之前，你需要确保已经安装了Flask和SQLAlchemy库。你可以使用以下命令来安装这两个库：

pip install Flask
pip install SQLAlchemy

创建Flask应用

首先，我们需要创建一个简单的Flask应用，并配置好SQLAlchemy连接。下面是一个简单的Flask应用示例：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

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

class User(db.Model):
    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)

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

@app.route('/')
def hello_world():
    return 'Hello, World!'

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

在上述代码中，我们创建了一个名为User的模型类，表示数据库中的用户表。然后，我们在hello_world路由中返回一个简单的字符串。接下来，我们需要初始化数据库，并创建用户数据。

初始化数据库

在Flask应用中，我们可以使用Flask-Migrate来管理数据库迁移，以确保数据库结构和代码的一致性。下面是一些常用的Flask-Migrate命令：

初始化数据库：flask db init
创建迁移脚本：flask db migrate
应用迁移：flask db upgrade

现在，我们需要执行以下命令初始化数据库：

flask db init
flask db migrate
flask db upgrade

修改数据

一旦数据库初始化完成，我们就可以开始使用ORM修改数据了。下面是一个简单的示例，演示如何使用ORM更新一个用户的邮箱信息：

from app import User, db

# 查询要更新的用户
user = User.query.filter_by(username='Alice').first()

if user:
    # 修改邮箱信息
    user.email = 'alice@example.com'

    # 提交数据库修改
    db.session.commit()
    print('User email updated successfully!')
else:
    print('User not found!')

在上述代码中，我们首先通过filter_by方法查询用户名为Alice的用户信息，然后修改用户的邮箱地址为alice@example.com，最后使用commit方法提交数据库修改。如果用户存在，程序将输出User email updated successfully!，否则输出User not found!。