Skip to content

Commit 8030534

Browse files
committed
patch: upsert function now returns rows
1 parent 43d6a6d commit 8030534

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

sqlmodel_crud_utils/a_sync.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -313,18 +313,18 @@ async def bulk_upsert_mappings(
313313
index_elements=[getattr(model, x) for x in pk_fields],
314314
set_={k: getattr(stmnt.excluded, k) for k in payload[0].keys()},
315315
)
316-
await session_inst.execute(stmnt)
316+
results = await session_inst.execute(stmnt)
317317

318318
await session_inst.commit()
319319

320-
return True
320+
return True, results.all()
321321

322322
except Exception as e:
323323
logger.error(
324324
f"Failed to upsert values to DB. Please see error: "
325325
f"{type(e), e, e.args}"
326326
)
327-
return False
327+
return False, []
328328

329329

330330
@logger.catch

sqlmodel_crud_utils/sync.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -310,18 +310,18 @@ def bulk_upsert_mappings(
310310
index_elements=[getattr(model, x) for x in pk_fields],
311311
set_={k: getattr(stmnt.excluded, k) for k in payload[0].keys()},
312312
)
313-
session_inst.execute(stmnt)
313+
results = session_inst.execute(stmnt)
314314

315315
session_inst.commit()
316316

317-
return True
317+
return True, results.all()
318318

319319
except Exception as e:
320320
logger.error(
321321
f"Failed to upsert values to DB. Please see error: "
322322
f"{type(e), e, e.args}"
323323
)
324-
return False
324+
return False, []
325325

326326

327327
@logger.catch

0 commit comments

Comments
 (0)