Fixed Loader

gravy-jones-locker · gravy-jones-locker · commit ecbde0fcdb94 · 2023-07-06T13:47:48.000+01:00
diff --git a/src/strawberry_sqlalchemy_mapper/loader.py b/src/strawberry_sqlalchemy_mapper/loader.py
@@ -39,7 +39,11 @@ async def load_fn(keys: List[Tuple]) -> List[Any]:
                 def group_by_remote_key(row: Any) -> Tuple:
                     return tuple(
                         [
-                            getattr(row, remote.key)
+                            [
+                                getattr(row, k)
+                                for k, column in row.__mapper__.c.items()
+                                if remote.key == column.key
+                            ][0]
                             for _, remote in relationship.local_remote_pairs
                         ]
                     )

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,11 @@ async def load_fn(keys: List[Tuple]) -> List[Any]:`
`39`	`39`	`def group_by_remote_key(row: Any) -> Tuple:`
`40`	`40`	`return tuple(`
`41`	`41`	`[`
`42`		`- getattr(row, remote.key)`
	`42`	`+ [`
	`43`	`+ getattr(row, k)`
	`44`	`+ for k, column in row.__mapper__.c.items()`
	`45`	`+ if remote.key == column.key`
	`46`	`+ ][0]`
`43`	`47`	`for _, remote in relationship.local_remote_pairs`
`44`	`48`	`]`
`45`	`49`	`)`