python - 是否可以在 mysqldb 中执行多次插入后获取所有 lastrowids？

Question

我不确定生成的id是否是连续生成的，如果不是，有没有其他方法可以获取它们？

class BaseDao(object):

    def __init__(self,pooldb):
        self.pooldb = pooldb

    def insertmany(self,sql,args):
        conn,cur = None,None
        try:
            conn = pooldb.dedicated_connection()
            cur = conn.cursor()
            num=cur.executemany(sql,args)
            if num <= 0:
                raise Exception("insert failure with num equals zero")
            lastrowid = int(cur.lastrowid)
            return [range(lastrowid - num + 1,lastrowid+1)]
        except:
            conn.rollback()
            traceback.print_exc()
            raise Exception("error happened when insert sql=%s args=%s " % (sql,str(args)))
        finally:
            if cur:
                cur.close()
            if conn:
                conn.close() 

Answer 1

每次 mysqldb 执行涉及自动增量列的查询时，最后的插入 ID 将丢失，除非您在下一个 execute() 之前读取它，而使用 executemany() 您将无法做到这一点。

您必须将上面的代码更改为:

num = 0
insert_ids = []

for arg in args:
    num += cur.execute(sql, arg)
    insert_ids.append(cur.lastrowid)

由于 Python 模块实际上是 MySQL API 之上的一个薄层，这里是对 mysql_insert_id() 的一些了解。功能有效。