Python Pony ORM一次性插入多个值

Question

我试图使用Pony ORM将多个值插入到我的Postgres数据库中。我目前的方法非常低效：

from pony.orm import *

db = Database()

class Names(db.Entity):
    first_name = Optional(str)
    last_name = Optional(str)


family = [["Peter", "Mueller"], ["Paul", "Meyer"], ...] 


@db_session
def populate_names(name_list)
    for name in name_list:
        db.insert("Names", first_name=name[0], last_name=name[1])


if __name__ == "__main__":
    db.bind(provider='postgres', user='', password='', host='', database='')
    db.generate_mappings(create_tables=True)
    populate_names(family)

这只是一个简单的例子，但输入的结构都是一样的：一堆列表。 我从几个 XML 文件中提取数据，并逐个“文件”插入。

有没有人知道如何在 Pony ORM 中将多行数据放入一个插入查询中？

- Melvin

Answer 1

马没有为此提供特别的东西，您可以使用 psycopg2.extras 中的 execute_values。从 db 获取 connection 对象以使用它。

from psycopg2.extras import execute_values
...
names = [
    ('はると', '一温'),
    ('りく', '俐空'),
    ('はる', '和晴'),
    ('ひなた', '向日'),
    ('ゆうと', '佑篤')
]

@db_session
def populate_persons(names):
    sql = 'insert into Person(first_name, last_name) values %s'
    con = db.get_connection()
    cur = con.cursor()
    execute_values(cur, sql, names)

populate_persons(names)

execute_values 在Fast execution helpers 列表中，因此我认为这应该是最有效的方法。

- sashaaero

Answer 2

在PonyORM的问题跟踪器中，有一个开放问题，要求实现这个功能。

我建议投票支持它。

- mrh1997

Answer 3

我目前正在尝试使用PonyORM进行未来项目的开发，并得出了与您相同的结论。

唯一批量插入数据的方法是：

# assuming data has this structure:
# [['foo','bar','bazooka'],...]
@db_session
def insert_bulk_array(field1, field2, field3):
    MyClass(field1=field1, field2=field2, field3=field3)

# assuming the data is:
# {'field1':'foo','field2':'bar','field3':'bazooka'}
@db_session
def insert_bulk_dict(data)
    MyClass(**data)

但从我的角度来看，这仍然有些方便，特别是当您的数据以JSON格式呈现时。

- brillenheini