Flask 如何从 `db.engine.connect().execute` 调用中获取插入的主键

在本文中，我们将介绍如何使用Flask从 db.engine.connect().execute() 调用中获取插入的主键。db.engine.connect().execute() 方法是Flask中用于执行原始SQL查询和命令的功能强大的功能。有时候在执行插入操作后，我们需要获取插入的主键值，以便于后续的操作，本文将详细介绍如何实现这一功能。

阅读更多：Flask 教程

什么是插入的主键

在关系型数据库中，每个表一般都会有一个主键列，来唯一标识该表的每一行数据。当往这个表插入一行数据时，数据库会为这一行分配一个唯一的主键值。在某些情况下，我们需要在插入数据后获取刚刚插入的行的主键值，以便于进行后续的操作，例如根据主键进行查询或更新。

使用Flask获取插入的主键

Flask提供了强大的数据库访问工具，其中包括SQLAlchemy扩展，它为我们提供了许多用于执行SQL查询和命令的方法。我们可以使用db.engine.connect().execute()方法来执行原始的SQL语句，包括插入操作。下面是一个使用Flask获取插入的主键的示例：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'your_database_uri'
db = SQLAlchemy(app)

@app.route('/')
def insert_data():
    # 执行插入操作
    db.engine.connect().execute("INSERT INTO users (name) VALUES ('John')")

    # 获取插入行的主键
    result = db.engine.connect().execute("SELECT LAST_INSERT_ID()")

    # 提取主键值
    inserted_primary_key = result.fetchone()[0]

    return f"插入的主键值为: {inserted_primary_key}"

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

在上面的示例中，我们首先定义了一个Flask应用，并配置了数据库的URI。然后，我们使用SQLAlchemy创建了一个数据库对象db。在insert_data()函数中，首先执行了插入操作，然后再执行了一个SELECT语句来获取刚刚插入的行的主键值。使用fetchone()方法可以获取查询结果的第一行数据，再使用索引[0]提取出主键值。最后，将获取到的主键值返回给客户端。

需要注意的是，上述示例中的users表是一个简单的示例表名，你需要根据你实际的数据表名进行修改。