d3vyce/fastapi-toolsets
1
0
Fork 0
mirror of https://github.com/d3vyce/fastapi-toolsets.git synced 2026-08-04 15:44:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: batch_merge crashes with AnnotatedJoin has no attribute _autoincrement_column on joined-table inheritance (#345)

Browse Source
This commit is contained in:
d3vyce
2026-07-07 18:22:13 +02:00
committed by GitHub
parent 040c023b8a
commit c256e58d64
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/fastapi_toolsets/fixtures/utils.py
+4 -3
View File
@@ -139,12 +139,13 @@ async def _batch_merge(
) -> None: ) -> None:
"""UPSERT: insert new rows, update existing ones with the provided values.""" """UPSERT: insert new rows, update existing ones with the provided values."""
for cls, group_dicts, _ in _grouped_table_dicts(model_cls, instances): for cls, group_dicts, _ in _grouped_table_dicts(model_cls, instances):
pk_names = [col.name for col in cls.__table__.primary_key] table = cast(Table, cls.__table__)
pk_names = [col.name for col in table.primary_key]
pk_names_set = set(pk_names) pk_names_set = set(pk_names)
own_col_keys = {col.key for col in cls.__table__.columns} own_col_keys = {col.key for col in table.columns}
non_pk_cols = [k for k in own_col_keys if k not in pk_names_set] non_pk_cols = [k for k in own_col_keys if k not in pk_names_set]
stmt = pg_insert(cls).values(group_dicts) stmt = pg_insert(table).values(group_dicts)
inserted_keys = set(group_dicts[0]) inserted_keys = set(group_dicts[0])
update_cols = [col for col in non_pk_cols if col in inserted_keys] update_cols = [col for col in non_pk_cols if col in inserted_keys]