python pymysql 没插入自增ID报错

原创

mob64ca12db7156 2023-12-22 07:50:52 ©著作权

文章标签 自增查询语句插入数据 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12db7156的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python pymysql 没插入自增ID报错解析

引言

在使用Python的pymysql库进行数据库操作时，有时候会遇到插入数据时没有正确返回自增ID的情况，导致程序报错。本文将介绍这个问题的原因，并提供相应的解决方案。

问题描述

当我们使用pymysql库插入一条新的记录时，通常会使用如下的SQL语句：

sql = "INSERT INTO table_name (column1, column2, ...) VALUES (%s, %s, ...)"
cursor.execute(sql, (value1, value2, ...))

然后通过cursor.lastrowid获取刚刚插入的记录的自增ID。然而，有时候这个操作会报错，提示AttributeError: 'Cursor' object has no attribute 'lastrowid'。

问题分析

首先，我们需要了解MySQL数据库的自增ID是如何生成的。MySQL使用一个名为AUTO_INCREMENT的属性来实现自增ID的功能。当我们向一个带有AUTO_INCREMENT属性的表插入一条新的记录时，MySQL会自动为这条记录分配一个唯一的自增ID。

在pymysql中，当我们执行插入操作后，需要通过cursor.lastrowid来获取刚插入记录的自增ID。然而，cursor.lastrowid这个属性是在执行cursor.execute()之后才会被设置的。

因此，如果我们在执行cursor.execute()之前就尝试获取cursor.lastrowid，就会报错。

解决方案

为了解决这个问题，我们需要调整我们的代码逻辑。我们可以在插入数据之前，先执行一条简单的SQL查询语句，然后再执行插入操作。

# 先执行一条查询语句
cursor.execute("SELECT 1")

# 执行插入操作
sql = "INSERT INTO table_name (column1, column2, ...) VALUES (%s, %s, ...)"
cursor.execute(sql, (value1, value2, ...))

# 获取自增ID
last_id = cursor.lastrowid

通过执行一条简单的查询语句，我们可以保证cursor.lastrowid属性已经被正确设置，然后我们再执行插入操作，并获取自增ID。

状态图

下面是一个用mermaid语法绘制的状态图，描述了插入数据时获取自增ID的过程。

stateDiagram
    [*] --> 初始状态
    初始状态 --> 执行查询语句
    执行查询语句 --> 执行插入操作
    执行插入操作 --> 获取自增ID
    获取自增ID --> 结束状态
    结束状态 --> [*]

甘特图

下面是一个用mermaid语法绘制的甘特图，描述了插入数据时的各个步骤及其耗时。

gantt
    dateFormat  YYYY-MM-DD
    title 插入数据时的步骤及耗时

    section 执行查询语句
    执行查询语句       :2022-01-01, 1d

    section 执行插入操作
    执行插入操作       :2022-01-02, 2d

    section 获取自增ID
    获取自增ID       :2022-01-04, 1d