summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/dialects/postgresql
diff options
context:
space:
mode:
authormike bayer <mike_mp@zzzcomputing.com>2022-09-26 01:17:44 +0000
committerGerrit Code Review <gerrit@ci3.zzzcomputing.com>2022-09-26 01:17:44 +0000
commit6201b4d88666983b883b96d22a159aa2594de94b (patch)
tree4036c155ca7c274ea4bd12c059fd8fcd277fc026 /lib/sqlalchemy/dialects/postgresql
parentf81fdd9a9008a6517f89f2115765b7db9a32721b (diff)
parenta8029f5a7e3e376ec57f1614ab0294b717d53c05 (diff)
downloadsqlalchemy-6201b4d88666983b883b96d22a159aa2594de94b.tar.gz
Merge "ORM bulk insert via execute" into main
Diffstat (limited to 'lib/sqlalchemy/dialects/postgresql')
-rw-r--r--lib/sqlalchemy/dialects/postgresql/provision.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/lib/sqlalchemy/dialects/postgresql/provision.py b/lib/sqlalchemy/dialects/postgresql/provision.py
index 8dd8a4995..4609701a2 100644
--- a/lib/sqlalchemy/dialects/postgresql/provision.py
+++ b/lib/sqlalchemy/dialects/postgresql/provision.py
@@ -134,9 +134,11 @@ def _upsert(cfg, table, returning, set_lambda=None):
stmt = insert(table)
+ table_pk = inspect(table).selectable
+
if set_lambda:
stmt = stmt.on_conflict_do_update(
- index_elements=table.primary_key, set_=set_lambda(stmt.excluded)
+ index_elements=table_pk.primary_key, set_=set_lambda(stmt.excluded)
)
else:
stmt = stmt.on_conflict_do_nothing()