Peewee 如何从批量插入中获取ID

在本文中，我们将介绍如何在Peewee中进行批量插入操作，并获取插入的记录对应的ID。

1. 插入数据的方式

在Peewee中，我们可以通过insert_many方法来实现批量插入数据。该方法接受一个包含多个字典的列表作为参数，每个字典代表一条记录。例如，我们有一个Person模型，有name和age两个字段，我们可以通过以下方式进行批量插入：

data = [
    {"name": "Alice", "age": 25},
    {"name": "Bob", "age": 30},
    {"name": "Charlie", "age": 35}
]

query = Person.insert_many(data)
query.execute()

2. 获取插入记录的ID

在进行批量插入后，我们可能需要获取插入记录对应的ID。Peewee提供了一种简单的方法来实现这个目标。我们可以使用returning方法来指定要返回的字段，默认返回插入的记录数量。例如，我们可以修改上面的插入代码，让它返回插入记录的ID：

query = Person.insert_many(data).returning(Person.id)
inserted_ids = query.execute()

print(inserted_ids)  # 输出插入记录的ID

上述代码中，我们使用returning方法指定要返回的字段为Person.id，然后通过execute方法执行插入操作，并将返回的ID存储在inserted_ids变量中。我们可以通过打印inserted_ids来检查插入记录的ID。

3. 处理自增主键

上述代码演示了如何在批量插入中获取记录的ID，但是它只适用于使用Peewee的自增主键模式。如果我们使用的是其他类型的主键，比如UUID或字符串，那么上述代码将无法获取到ID。

针对这种情况，Peewee提供了一个名为insert_returning的方法来处理。我们可以使用这个方法指定要返回的字段，并在执行插入操作时获取到返回的ID。以下是一个示例：

from peewee import SQL

data = [
    {"name": "Alice", "uuid": "123abc"},
    {"name": "Bob", "uuid": "456def"},
    {"name": "Charlie", "uuid": "789ghi"}
]

query = Person.insert_many(data).returning(Person.uuid.alias('id'))
inserted_ids = query.execute()

print(inserted_ids)  # 输出插入记录的ID

在这个示例中，我们在Person.uuid字段上使用了别名id，这样我们可以通过inserted_ids获取插入记录的ID。

4. 总结

在本文中，我们介绍了如何在Peewee中进行批量插入，并获取插入记录的ID。我们可以使用insert_many方法来实现批量插入操作，使用returning方法来指定要返回的字段，并通过execute方法执行插入操作。对于非自增主键的情况，我们可以使用insert_returning方法来处理。通过应用这些方法，我们可以方便地插入大量数据，并获取插入记录的ID。

如果你对Peewee的使用还有更深入的了解，可以查阅Peewee官方文档进行更详细的学习和探索。