patch: bulk_upsert_mappings now take in multiple pk_fields arg for indexing elements

Author: fsecada01

sqlmodel_crud_utils/a_sync.py

@@ -303,12 +303,14 @@ async def bulk_upsert_mappings(
     payload: list,
     session_inst: AsyncSession,
     model: type[SQLModel],
-    pk_field: str = "id",
+    pk_fields: list[str] | None = None,
 ):
+    if not pk_fields:
+        pk_fields = ["id"]
     try:
         stmnt = upsert(model).values(payload)
         stmnt = stmnt.on_conflict_do_update(
-            index_elements=[getattr(model, pk_field)],
+            index_elements=[getattr(model, x) for x in pk_fields],
             set_={k: getattr(stmnt.excluded, k) for k in payload[0].keys()},
         )
         await session_inst.execute(stmnt)

sqlmodel_crud_utils/sync.py

@@ -300,12 +300,14 @@ def bulk_upsert_mappings(
     payload: list,
     session_inst: Session,
     model: type[SQLModel],
-    pk_field: str = "id",
+    pk_fields: list[str] | None = None,
 ):
+    if not pk_fields:
+        pk_fields = ["id"]
     try:
         stmnt = upsert(model).values(payload)
         stmnt = stmnt.on_conflict_do_update(
-            index_elements=[getattr(model, pk_field)],
+            index_elements=[getattr(model, x) for x in pk_fields],
             set_={k: getattr(stmnt.excluded, k) for k in payload[0].keys()},
         )
         session_inst.execute(stmnt)