如何从Peewee的批量插入中获取id？

Question

如何从Peewee的大容量插入中获得ids？在

我需要返回插入的id来创建一个新的dict数组，如下所示：

a = [{"product_id": "inserted_id_1", "name": "name1"}, {"product_id": "inserted_id_2", "name": "name1"}]

然后我需要使用bulk插入如下内容：

^{pr2}$

反过来，最后一个查询应该返回新的id，以便进一步进行类似的插入。在

Answer 1

如果您使用的是Postgresql，它支持“INSERT…”形式的查询。。。返回“，您可以获得所有ID：

data = [{'product_id': 'foo', 'name': 'name1'}, {...}, ...]
id_list = SomeModel.insert_many(data).execute()

对于不支持RETURNING子句的SQLite或MySQL，您最好这样做：

^{pr2}$